NgddlZddlmZddlmZmZddlmZmZmZddl m Z ddl m Z m Z mZddlmZmZddlmZGd d eZd e d eeeeffd ZGddeZdS)N)JSONDecodeError)ListUnion) AgentActionAgentActionMessageLog AgentFinish)OutputParserException) AIMessage BaseMessageToolCall)ChatGeneration Generation)MultiActionAgentOutputParserceZdZUeed<dS)ToolAgentAction tool_call_idN)__name__ __module__ __qualname__str__annotations__a/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/agents/output_parsers/tools.pyrrs77rrmessagereturnc t|tstdt|g}|jr|j}n|jds*td|jit|jSg}|jdD]{}|d}|d} tj |dpd}| t|||d  [#t$rtd |d wxYw|D]o}|d}|d }d|vr |d}n|}|jr d|jdnd} d|d|d| d} | t!||| |g|d p|S)z6Parse an AI message potentially containing tool_calls.zExpected an AI message got tool_callsoutput) return_valueslogfunctionname argumentsz{}id)r#argsr%zCould not parse tool input: z+ because the `arguments` is not valid JSON.r&__arg1z responded:  z Invoking: `z` with `z` )tool tool_inputr! message_logr) isinstancer TypeErrortyperadditional_kwargsgetrcontentrjsonloadsappendr rr r) ractionsr tool_callr" function_namer& _tool_inputr* content_msgr!s rparse_ai_message_to_tool_actionr:s( gy ) )GEd7mmEEFFFG' (,,\:: '9s7??S?S   2<@  I ,H$V,M z(;"7"?4@@!!-dyOOO#   +:8:::     "&) ' { " "$X.JJ$J;B?T7GO7777PT SmSSZSSKSSS "%$I&t_         Ns ,AC44Dc eZdZdZedefdZdddeede de ee e ffdZ d ede ee e ffd Zd S) ToolsAgentOutputParserzParses a message into agent actions/finish. 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)Nztools-agent-output-parserr)selfs r_typezToolsAgentOutputParser._typeYs**rF)partialresultr@ct|dtstd|dj}t |S)Nrz6This output parser only works on ChatGeneration output)r,r ValueErrorrr:)r>rAr@rs r parse_resultz#ToolsAgentOutputParser.parse_result]sD&)^44 WUVV V)#.w777rtextc td)NzCan only parse messages)rC)r>rEs rparsezToolsAgentOutputParser.parsees2333rN)rrr__doc__propertyrr?rrboolrrrrDrGrrrr<r<Ps+s+++X+rQsZ QQQQQQQQQQ;;;;;; >=======??????88888+888 7 7 4 k )*7777t44444944444r