fix: modify rag database

2026-06-07 22:36:02 +00:00 · 2025-07-05 17:53:11 +08:00
parent 34fe8b324d
commit 552fee9bac
2 changed files with 5 additions and 4 deletions
--- a/pkg/rag/knowledge/services/database.py
+++ b/pkg/rag/knowledge/services/database.py
@@ -11,7 +11,8 @@ class KnowledgeBase(Base):
    name = Column(String, index=True)
    description = Column(Text)
    created_at = Column(DateTime, default=datetime.utcnow)
-
+    embedding_model = Column(String, default="")  # 默认嵌入模型
+    top_k = Column(Integer, default=5)  # 默认返回的top_k数量
    files = relationship("File", back_populates="knowledge_base")

 class File(Base):