fix: Error in RootListenersTracer.on_chain_end callback: ValueError()

src/codeinterpreterapi/agents/structured_chat/agent.py

@@ -170,9 +170,9 @@ def create_structured_chat_agent(
         | create_complement_input(prompt)
         | prompt
         | llm_with_stop
-        | output_parser
     )
     agent = assign_runnable_history(agent, runnable_config)
+    agent = agent | output_parser
     return agent
 
 

src/codeinterpreterapi/agents/structured_chat/agent_executor.py

@@ -17,7 +17,7 @@ def load_structured_chat_agent_executor(
     Load an agent executor(general purpose).
     """
     prompt = create_structured_chat_agent_prompt(ci_params.is_ja)
-    if agent_def.agent_role is not None:
+    if agent_def and agent_def.agent_role is not None:
         prompt = prompt.partial(agent_role=agent_def.agent_role)
     input_variables = prompt.input_variables
     if ci_params.verbose_prompt:
@@ -31,10 +31,12 @@ def load_structured_chat_agent_executor(
         runnable_config=ci_params.runnable_config,
         # stop_sequence=["Observation:", "最終回答", "Final Answer"],
     )
-    agent_def.agent = agent
+    if agent_def:
+        agent_def.agent = agent
 
     agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=ci_params.tools, verbose=ci_params.verbose)
-    agent_def.agent_executor = agent_executor
+    if agent_def:
+        agent_def.agent_executor = agent_executor
     return agent_executor
 
 

src/codeinterpreterapi/agents/tool_calling/agent.py

@@ -105,9 +105,9 @@ def magic_function(input: int) -> int:
         | create_complement_input(prompt)
         | prompt
         | llm_with_tools
-        | output_parser
     )
     agent = assign_runnable_history(agent, runnable_config)
+    agent = agent | output_parser
     return agent
 
 

src/codeinterpreterapi/agents/tool_calling/agent_executor.py

@@ -17,7 +17,7 @@ def load_tool_calling_agent_executor(
     Load an agent executor(general purpose).
     """
     prompt = create_tool_calling_agent_prompt(ci_params.is_ja)
-    if agent_def.message_prompt_template is not None:
+    if agent_def and agent_def.message_prompt_template is not None:
         prompt = prompt.partial(agent_role=agent_def.agent_role)
     if ci_params.verbose_prompt:
         input_variables = prompt.input_variables
@@ -30,10 +30,12 @@ def load_tool_calling_agent_executor(
         prompt=prompt,
         runnable_config=ci_params.runnable_config,
     )
-    agent_def.agent = agent
+    if agent_def:
+        agent_def.agent = agent
 
     agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=ci_params.tools, verbose=ci_params.verbose)
-    agent_def.agent_executor = agent_executor
+    if agent_def:
+        agent_def.agent_executor = agent_executor
 
     return agent_executor