diff --git a/rag/init.py b/rag/init.py index 56bf3eb..3dcfdee 100644 --- a/rag/init.py +++ b/rag/init.py @@ -3,7 +3,6 @@ from ahserver.serverenv import ServerEnv import aiohttp from aiohttp import ClientSession, ClientTimeout import json -from .file import file_uploaded, file_deleted from .folderinfo import RagFileMgr from .ragprogram import set_program, get_rag_programs from .ragllm_utils import get_ragllms_by_catelog diff --git a/rag/ragapi.py b/rag/ragapi.py index 9174b00..355c20f 100644 --- a/rag/ragapi.py +++ b/rag/ragapi.py @@ -8,6 +8,7 @@ import math import uuid from rag.service_opts import get_service_params, sor_get_service_params from rag.rag_operations import RagOperations +from langchain_core.documents import Document helptext = """kyrag API: @@ -254,7 +255,7 @@ async def textinsert(request, params_kw, *params): result = { "status": "error", "userid": orgid, - "collection_name": "ragdb_{dbtype}", + "collection_name": f"ragdb_{db_type}", "message": "", "status_code": 400 } @@ -277,10 +278,10 @@ async def textinsert(request, params_kw, *params): # 插入 Milvus fields = { "text": text, - "fiid": fiid, - "orgid": orgid, + "knowledge_base_id": fiid, + "userid": orgid, "vector": embedding, - "id": id + "document_id": id } chunks_data = await rag_ops.insert_to_vector_text(request, db_type, fields, service_params, userid, timings)