Add LLM configuration and pipeline execution

This commit is contained in:
2026-05-31 14:13:58 -07:00
parent 2858c7e235
commit 71ecdb3468
4 changed files with 80 additions and 14 deletions
+3 -2
View File
@@ -12,8 +12,9 @@ from src.llm.processor import LLMProcessor
class RAGManager:
def __init__(self, persist_dir: str = "data/rag_index"):
def __init__(self, persist_dir: str = "data/rag_index", llm_config: Optional[dict] = None):
self.persist_dir = persist_dir
self.llm_config = llm_config or {}
self.db = chromadb.PersistentClient(path=self.persist_dir)
self.collection_name = "phb_collection"
@@ -110,7 +111,7 @@ class RAGManager:
if not nodes:
return []
processor = LLMProcessor()
processor = LLMProcessor(**self.llm_config)
# Construct the context from retrieved nodes
context_text = "\n\n".join(