Add support for vllm
This commit is contained in:
parent
d9540c3bba
commit
ec266cb5c2
|
|
@ -99,17 +99,12 @@ class TradingAgentsGraph:
|
||||||
]
|
]
|
||||||
),
|
),
|
||||||
).ask()
|
).ask()
|
||||||
self.bull_memory = FinancialSituationMemory("bull_memory", self.config)
|
|
||||||
self.bear_memory = FinancialSituationMemory("bear_memory", self.config)
|
self.bull_memory = FinancialSituationMemory("bull_memory", self.config)
|
||||||
self.trader_memory = FinancialSituationMemory("trader_memory", self.config)
|
self.bear_memory = FinancialSituationMemory("bear_memory", self.config)
|
||||||
self.invest_judge_memory = FinancialSituationMemory("invest_judge_memory", self.config)
|
self.trader_memory = FinancialSituationMemory("trader_memory", self.config)
|
||||||
self.risk_manager_memory = FinancialSituationMemory("risk_manager_memory", self.config)
|
self.invest_judge_memory = FinancialSituationMemory("invest_judge_memory", self.config)
|
||||||
else:
|
self.risk_manager_memory = FinancialSituationMemory("risk_manager_memory", self.config)
|
||||||
self.bull_memory = FinancialSituationMemory("bull_memory", self.config)
|
|
||||||
self.bear_memory = FinancialSituationMemory("bear_memory", self.config)
|
|
||||||
self.trader_memory = FinancialSituationMemory("trader_memory", self.config)
|
|
||||||
self.invest_judge_memory = FinancialSituationMemory("invest_judge_memory", self.config)
|
|
||||||
self.risk_manager_memory = FinancialSituationMemory("risk_manager_memory", self.config)
|
|
||||||
|
|
||||||
# Create tool nodes
|
# Create tool nodes
|
||||||
self.tool_nodes = self._create_tool_nodes()
|
self.tool_nodes = self._create_tool_nodes()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue