rag

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

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)