Ng ddlmZmZddlmZmZddlmZddlm Z m Z ddl m Z ddl mZmZeZdedeeeeffd ZGd d e Zd S) )ListUnion) AgentAction AgentFinish) BaseMessage)ChatGeneration Generation)MultiActionAgentOutputParser)ToolAgentActionparse_ai_message_to_tool_actionmessagereturnc 6t|}t|tr|Sg}|D]m}t|trA|t |j|j|j|j |j X||n|S)z6Parse an AI message potentially containing tool_calls.)tool tool_inputlog message_log tool_call_id) r isinstancerr appendOpenAIToolAgentActionrrrrr)r tool_actions final_actionsactions h/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/agents/output_parsers/openai_tools.py&parse_ai_message_to_openai_tool_actionrs37;;L, ,,')M ) ) fo . . )  %%0 & 2!'!4        ( ( ( ( c eZdZdZedefdZdddeede de ee e ffdZ d ede ee e ffd Zd S) OpenAIToolsAgentOutputParseratParses a message into agent actions/finish. Is meant to be used with OpenAI models, as it relies on the specific tool_calls parameter from OpenAI to convey what tools to use. If a tool_calls parameter is passed, then that is used to get the tool names and tool inputs. If one is not passed, then the AIMessage is assumed to be the final output. rcdS)Nz openai-tools-agent-output-parser)selfs r_typez"OpenAIToolsAgentOutputParser._type4s11rF)partialresultr$ct|dtstd|dj}t |S)Nrz6This output parser only works on ChatGeneration output)rr ValueErrorr r)r"r%r$r s r parse_resultz)OpenAIToolsAgentOutputParser.parse_result8sD&)^44 WUVV V)#5g>>>rtextc td)NzCan only parse messages)r')r"r)s rparsez"OpenAIToolsAgentOutputParser.parse@s2333rN)__name__ __module__ __qualname____doc__propertystrr#rr boolrrrr(r+r!rrrr(s  2s222X2r9s :::::::://////========?????? (  4 k )*044444#?44444r