yumoqing/rag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rag

Browse Source
This commit is contained in:
wangmeihua 2025-10-24 09:31:11 +08:00
parent 35d0eb2234
commit 05e4ecb190
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
rag/init.py
View File

@ -3,7 +3,6 @@ from ahserver.serverenv import ServerEnv
import aiohttp import aiohttp
from aiohttp import ClientSession, ClientTimeout from aiohttp import ClientSession, ClientTimeout
import json import json
from .file import file_uploaded, file_deleted
from .folderinfo import RagFileMgr from .folderinfo import RagFileMgr
from .ragprogram import set_program, get_rag_programs from .ragprogram import set_program, get_rag_programs
from .ragllm_utils import get_ragllms_by_catelog from .ragllm_utils import get_ragllms_by_catelog

9
rag/ragapi.py
View File

@ -8,6 +8,7 @@ import math
import uuid import uuid
from rag.service_opts import get_service_params, sor_get_service_params from rag.service_opts import get_service_params, sor_get_service_params
from rag.rag_operations import RagOperations from rag.rag_operations import RagOperations
from langchain_core.documents import Document
helptext = """kyrag API: helptext = """kyrag API:
@ -254,7 +255,7 @@ async def textinsert(request, params_kw, *params):
result = { result = {
"status": "error", "status": "error",
"userid": orgid, "userid": orgid,
"collection_name": "ragdb_{dbtype}", "collection_name": f"ragdb_{db_type}",
"message": "", "message": "",
"status_code": 400 "status_code": 400
} }
@ -277,10 +278,10 @@ async def textinsert(request, params_kw, *params):
# 插入 Milvus # 插入 Milvus
fields = { fields = {
"text": text, "text": text,
"fiid": fiid, "knowledge_base_id": fiid,
"orgid": orgid, "userid": orgid,
"vector": embedding, "vector": embedding,
"id": id "document_id": id
} }
chunks_data = await rag_ops.insert_to_vector_text(request, db_type, fields, service_params, userid, timings) chunks_data = await rag_ops.insert_to_vector_text(request, db_type, fields, service_params, userid, timings)