From 8870a34a06392d88812837fb0937b9f074c73d26 Mon Sep 17 00:00:00 2001 From: Henry Zhang Date: Sun, 6 Jul 2025 18:57:45 -0700 Subject: [PATCH] fix memory error --- tradingagents/agents/utils/memory.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tradingagents/agents/utils/memory.py b/tradingagents/agents/utils/memory.py index 69b8ab8c..65c5473c 100644 --- a/tradingagents/agents/utils/memory.py +++ b/tradingagents/agents/utils/memory.py @@ -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"""