From 7d5503dab201a44112b120e2bde6b47e3117f610 Mon Sep 17 00:00:00 2001 From: Junyan Qin Date: Fri, 11 Jul 2025 16:49:55 +0800 Subject: [PATCH] fix: bug --- pkg/rag/knowledge/mgr.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/pkg/rag/knowledge/mgr.py b/pkg/rag/knowledge/mgr.py index 5d4eece9..4da10a09 100644 --- a/pkg/rag/knowledge/mgr.py +++ b/pkg/rag/knowledge/mgr.py @@ -26,7 +26,9 @@ class RAGManager: """Initializes the RAG system by creating database tables.""" await asyncio.to_thread(create_db_and_tables) - async def create_knowledge_base(self, kb_name: str, kb_description: str, embedding_model: str = '', top_k: int = 5): + async def create_knowledge_base( + self, kb_name: str, kb_description: str, embedding_model_uuid: str = '', top_k: int = 5 + ): """ Creates a new knowledge base if it doesn't already exist. """ @@ -43,7 +45,7 @@ class RAGManager: new_kb = KnowledgeBase( name=kb_name, description=kb_description, - embedding_model=embedding_model, + embedding_model_uuid=embedding_model_uuid, top_k=top_k, id=id, )