From 5c3011b429f71bb00717bab47fca1723c28896ad Mon Sep 17 00:00:00 2001 From: delenzhang Date: Thu, 3 Jul 2025 22:15:03 +0800 Subject: [PATCH] feat: add llm embedding model config --- tradingagents/agents/utils/memory.py | 1 + tradingagents/default_config.py | 1 + 2 files changed, 2 insertions(+) diff --git a/tradingagents/agents/utils/memory.py b/tradingagents/agents/utils/memory.py index 69b8ab8c..62a2e4fe 100644 --- a/tradingagents/agents/utils/memory.py +++ b/tradingagents/agents/utils/memory.py @@ -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) diff --git a/tradingagents/default_config.py b/tradingagents/default_config.py index 089e9c24..a8030168 100644 --- a/tradingagents/default_config.py +++ b/tradingagents/default_config.py @@ -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,