xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-15 23:10:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: bug

Browse Source
This commit is contained in:
Junyan Qin
2025-07-11 16:49:55 +08:00
parent 9ba1ad5bd3
commit 7d5503dab2
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/rag/knowledge/mgr.py
+4 -2
View File
@@ -26,7 +26,9 @@ class RAGManager:
"""Initializes the RAG system by creating database tables.""" """Initializes the RAG system by creating database tables."""
await asyncio.to_thread(create_db_and_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. Creates a new knowledge base if it doesn't already exist.
""" """
@@ -43,7 +45,7 @@ class RAGManager:
new_kb = KnowledgeBase( new_kb = KnowledgeBase(
name=kb_name, name=kb_name,
description=kb_description, description=kb_description,
embedding_model=embedding_model, embedding_model_uuid=embedding_model_uuid,
top_k=top_k, top_k=top_k,
id=id, id=id,
) )