From 44a5cb133a1ecadca2cdffffde3eea04c3fb1623 Mon Sep 17 00:00:00 2001 From: Jeffrey Chu Date: Thu, 23 Oct 2025 19:07:27 +0800 Subject: [PATCH] update --- tradingagents/agents/utils/memory.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tradingagents/agents/utils/memory.py b/tradingagents/agents/utils/memory.py index 4ace3290..02a2733c 100644 --- a/tradingagents/agents/utils/memory.py +++ b/tradingagents/agents/utils/memory.py @@ -11,15 +11,15 @@ class FinancialSituationMemory: self.embedding = "nomic-embed-text" self.embedding_client = OpenAI(base_url="http://localhost:11434/v1") self.use_local_embedding = False - elif config["llm_provider"] == "xai": - # Grok - use local Hugging Face model - self.embedding_model = SentenceTransformer('all-MiniLM-L6-v2') - self.use_local_embedding = True - else: + elif config["llm_provider"] == "openai": self.client = OpenAI(base_url=config["backend_url"], api_key=config.get("api_key")) self.embedding = "text-embedding-3-small" self.embedding_client = self.client self.use_local_embedding = False + else: + # use local Hugging Face model + self.embedding_model = SentenceTransformer('all-MiniLM-L6-v2') + self.use_local_embedding = True self.chroma_client = chromadb.Client(Settings(allow_reset=True)) self.situation_collection = self.chroma_client.create_collection(name=name)