feat: add llm embedding model config

This commit is contained in:
delenzhang 2025-07-03 22:15:03 +08:00
parent 718df34932
commit 5c3011b429
2 changed files with 2 additions and 0 deletions

View File

@ -9,6 +9,7 @@ class FinancialSituationMemory:
self.embedding = "nomic-embed-text"
else:
self.embedding = "text-embedding-3-small"
self.embedding = config.get("embedding") or self.embedding
self.client = OpenAI(base_url=config["backend_url"])
self.chroma_client = chromadb.Client(Settings(allow_reset=True))
self.situation_collection = self.chroma_client.create_collection(name=name)

View File

@ -13,6 +13,7 @@ DEFAULT_CONFIG = {
"deep_think_llm": "o4-mini",
"quick_think_llm": "gpt-4o-mini",
"backend_url": "https://api.openai.com/v1",
"embedding": "",
# Debate and discussion settings
"max_debate_rounds": 1,
"max_risk_discuss_rounds": 1,