xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-27 07:54:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: stash

Browse Source
This commit is contained in:
Junyan Qin
2025-07-16 11:31:55 +08:00
parent 67bc065ccd
commit f731115805
10 changed files with 123 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/rag/knowledge/mgr.py
+6 -8
View File
@@ -13,10 +13,9 @@ from pkg.rag.knowledge.services.database import (
from pkg.core import app
from pkg.rag.knowledge.services.embedder import Embedder
from pkg.rag.knowledge.services.retriever import Retriever
from pkg.rag.knowledge.services.chroma_manager import ChromaIndexManager
from pkg.core import taskmgr
from ...entity.persistence import rag as persistence_rag
import sqlalchemy
from ...entity.persistence import rag as persistence_rag
from pkg.core import taskmgr
class RuntimeKnowledgeBase:
@@ -24,8 +23,6 @@ class RuntimeKnowledgeBase:
knowledge_base_entity: persistence_rag.KnowledgeBase
chroma_manager: ChromaIndexManager
parser: FileParser
chunker: Chunker
@@ -37,11 +34,12 @@ class RuntimeKnowledgeBase:
def __init__(self, ap: app.Application, knowledge_base_entity: persistence_rag.KnowledgeBase):
self.ap = ap
self.knowledge_base_entity = knowledge_base_entity
self.chroma_manager = ChromaIndexManager(ap=self.ap)
self.parser = FileParser(ap=self.ap)
self.chunker = Chunker(ap=self.ap)
self.embedder = Embedder(ap=self.ap, chroma_manager=self.chroma_manager)
self.retriever = Retriever(ap=self.ap, chroma_manager=self.chroma_manager)
self.embedder = Embedder(ap=self.ap)
self.retriever = Retriever(ap=self.ap)
# 传递kb_id给retriever
self.retriever.kb_id = knowledge_base_entity.uuid
async def initialize(self):
pass