perf(rag): ui and related apis

pkg/api/http/controller/groups/knowledge/base.py

@@ -20,7 +20,7 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
 
         @self.route(
             '/<knowledge_base_uuid>',
-            methods=['GET', 'DELETE'],
+            methods=['GET', 'DELETE', 'PUT'],
         )
         async def handle_specific_knowledge_base(knowledge_base_uuid: str) -> quart.Response:
             if quart.request.method == 'GET':
@@ -34,6 +34,12 @@ class KnowledgeBaseRouterGroup(group.RouterGroup):
                         'base': knowledge_base,
                     }
                 )
+
+            elif quart.request.method == 'PUT':
+                json_data = await quart.request.json
+                await self.ap.knowledge_service.update_knowledge_base(knowledge_base_uuid, json_data)
+                return self.success({})
+
             elif quart.request.method == 'DELETE':
                 await self.ap.knowledge_service.delete_knowledge_base(knowledge_base_uuid)
                 return self.success({})

pkg/api/http/service/knowledge.py

@@ -47,12 +47,18 @@ class KnowledgeService:
 
     async def update_knowledge_base(self, kb_uuid: str, kb_data: dict) -> None:
         """更新知识库"""
+        if 'uuid' in kb_data:
+            del kb_data['uuid']
+
+        if 'embedding_model_uuid' in kb_data:
+            del kb_data['embedding_model_uuid']
+
         await self.ap.persistence_mgr.execute_async(
             sqlalchemy.update(persistence_rag.KnowledgeBase)
             .values(kb_data)
             .where(persistence_rag.KnowledgeBase.uuid == kb_uuid)
         )
-        await self.ap.rag_mgr.remove_knowledge_base(kb_uuid)
+        await self.ap.rag_mgr.remove_knowledge_base_from_runtime(kb_uuid)
 
         kb = await self.get_knowledge_base(kb_uuid)
 
@@ -91,7 +97,7 @@ class KnowledgeService:
 
     async def delete_knowledge_base(self, kb_uuid: str) -> None:
         """删除知识库"""
-        await self.ap.rag_mgr.remove_knowledge_base(kb_uuid)
+        await self.ap.rag_mgr.delete_knowledge_base(kb_uuid)
 
         await self.ap.persistence_mgr.execute_async(
             sqlalchemy.delete(persistence_rag.KnowledgeBase).where(persistence_rag.KnowledgeBase.uuid == kb_uuid)

pkg/provider/runners/localagent.py

@@ -33,6 +33,9 @@ class LocalAgentRunner(runner.RequestRunner):
 
         kb_uuid = query.pipeline_config['ai']['local-agent']['knowledge-base']
 
+        if kb_uuid == '__none__':
+            kb_uuid = None
+
         user_message = copy.deepcopy(query.user_message)
 
         user_message_text = ''

pkg/rag/knowledge/kbmgr.py

@@ -198,7 +198,13 @@ class RAGManager:
                 return kb
         return None
 
-    async def remove_knowledge_base(self, kb_uuid: str):
+    async def remove_knowledge_base_from_runtime(self, kb_uuid: str):
+        for kb in self.knowledge_bases:
+            if kb.knowledge_base_entity.uuid == kb_uuid:
+                self.knowledge_bases.remove(kb)
+                return
+
+    async def delete_knowledge_base(self, kb_uuid: str):
         for kb in self.knowledge_bases:
             if kb.knowledge_base_entity.uuid == kb_uuid:
                 await kb.dispose()

pkg/vector/vdbs/chroma.py

@@ -5,6 +5,7 @@ from chromadb import PersistentClient
 from pkg.vector.vdb import VectorDatabase
 from pkg.core import app
 import chromadb
+import chromadb.errors
 
 
 class ChromaVectorDatabase(VectorDatabase):
@@ -51,5 +52,10 @@ class ChromaVectorDatabase(VectorDatabase):
     async def delete_collection(self, collection: str):
         if collection in self._collections:
             del self._collections[collection]
-        await asyncio.to_thread(self.client.delete_collection, name=collection)
+
+        try:
+            await asyncio.to_thread(self.client.delete_collection, name=collection)
+        except chromadb.errors.NotFoundError:
+            self.ap.logger.warning(f"Chroma collection '{collection}' not found.")
+            return
         self.ap.logger.info(f"Chroma collection '{collection}' deleted.")