Fix main local test of tradingagents/agents/utils/memory.py

This commit is contained in:
Taylor Mao 2025-07-21 20:34:58 -04:00
parent 873fe41f1d
commit 8df223b2b6
1 changed files with 1 additions and 2 deletions

View File

@ -2,7 +2,6 @@ import chromadb
from chromadb.config import Settings from chromadb.config import Settings
from openai import OpenAI from openai import OpenAI
class FinancialSituationMemory: class FinancialSituationMemory:
def __init__(self, name, config): def __init__(self, name, config):
if config["backend_url"] == "http://localhost:11434/v1": if config["backend_url"] == "http://localhost:11434/v1":
@ -69,7 +68,7 @@ class FinancialSituationMemory:
if __name__ == "__main__": if __name__ == "__main__":
# Example usage # Example usage
matcher = FinancialSituationMemory() matcher = FinancialSituationMemory( name="test", config={"backend_url": "https://api.openai.com/v1",})
# Example data # Example data
example_data = [ example_data = [