diff --git a/tradingagents/agents/trader/trader.py b/tradingagents/agents/trader/trader.py index 282a8411..1b05c35d 100644 --- a/tradingagents/agents/trader/trader.py +++ b/tradingagents/agents/trader/trader.py @@ -16,8 +16,11 @@ def create_trader(llm, memory): past_memories = memory.get_memories(curr_situation, n_matches=2) past_memory_str = "" - for i, rec in enumerate(past_memories, 1): - past_memory_str += rec["recommendation"] + "\n\n" + if past_memories: + for i, rec in enumerate(past_memories, 1): + past_memory_str += rec["recommendation"] + "\n\n" + else: + past_memory_str = "No past memories found." context = { "role": "user", diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py index eb06cf43..80a29e53 100644 --- a/tradingagents/graph/trading_graph.py +++ b/tradingagents/graph/trading_graph.py @@ -226,7 +226,7 @@ class TradingAgentsGraph: directory.mkdir(parents=True, exist_ok=True) with open( - f"eval_results/{self.ticker}/TradingAgentsStrategy_logs/full_states_log.json", + f"eval_results/{self.ticker}/TradingAgentsStrategy_logs/full_states_log_{trade_date}.json", "w", ) as f: json.dump(self.log_states_dict, f, indent=4)