bug fix for xai support

This commit is contained in:
Jeffrey Chu 2025-10-22 19:11:18 +08:00
parent 108bab8402
commit 08f9cd552d
2 changed files with 1 additions and 41 deletions

1
.gitignore vendored
View File

@ -9,3 +9,4 @@ eval_results/
eval_data/
*.egg-info/
.env
.history

View File

@ -62,47 +62,6 @@ class TradingAgentsGraph:
"""
self.debug = debug
self.config = config or DEFAULT_CONFIG
# get the language
self.output_language = self.config.get("output_language", "en")
self.language_instruction = self.config.get("language_system_prompts", {}).get(
self.output_language, ""
)
self._setup_agents()
self._setup_graph()
def _setup_agents(self):
"""初始化所有 agents並注入語言指示"""
# 為每個 agent 加入語言指示
if self.language_instruction:
self._inject_language_to_agents()
def _inject_language_to_agents(self):
"""將語言指示注入所有 agents"""
# 這個方法會在每個 agent 的 system message 前加入語言指示
self.language_system_message = SystemMessage(content=self.language_instruction)
def _create_agent_with_language(self, agent_name, agent_prompt, llm):
"""創建帶有語言指示的 agent"""
from langchain.agents import AgentExecutor, create_openai_functions_agent
from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
# 構建包含語言指示的 prompt
if self.language_instruction:
system_prompt = f"{agent_prompt}{self.language_instruction}"
else:
system_prompt = agent_prompt
prompt = ChatPromptTemplate.from_messages([
("system", system_prompt),
MessagesPlaceholder(variable_name="chat_history", optional=True),
("human", "{input}"),
MessagesPlaceholder(variable_name="agent_scratchpad"),
])
agent = create_openai_functions_agent(llm, tools=[], prompt=prompt)
return AgentExecutor(agent=agent, tools=[], verbose=self.debug)
# Update the interface's config
set_config(self.config)