charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix memory error

This commit is contained in:
Henry Zhang 2025-07-06 18:57:45 -07:00 committed by GitHub
parent a438acdbbd
commit 8870a34a06
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tradingagents/agents/utils/memory.py
View File

@ -11,7 +11,13 @@ class FinancialSituationMemory:
self.embedding = "text-embedding-3-small"
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)
# Try to get existing collection, create if it doesn't exist
try:
self.situation_collection = self.chroma_client.get_collection(name=name)
except Exception:
# Collection doesn't exist, create it
self.situation_collection = self.chroma_client.create_collection(name=name)
def get_embedding(self, text):
"""Get OpenAI embedding for a text"""