
    Ng6                        d dl mZ d dlZd dlmZ d dlmZmZ d dlm	Z	 d dl
mZ d dlmZ  ej        e          Z G d d	e          ZdS )
    )annotationsN)Union)AgentActionAgentFinish)OutputParserException)parse_json_markdown)AgentOutputParserc                  2    e Zd ZdZd	dZed
d            ZdS )JSONAgentOutputParseraH  Parses tool invocations and final answers in JSON format.

    Expects output to be in one of two formats.

    If the output signals that an action should be taken,
    should be in the below format. This will result in an AgentAction
    being returned.

    ```
    {
      "action": "search",
      "action_input": "2+2"
    }
    ```

    If the output signals that a final answer should be given,
    should be in the below format. This will result in an AgentFinish
    being returned.

    ```
    {
      "action": "Final Answer",
      "action_input": "4"
    }
    ```
    textstrreturnUnion[AgentAction, AgentFinish]c                z   	 t          |          }t          |t                    r#t                              d|           |d         }|d         dk    rt          d|d         i|          S t          |d         |                    di           |          S # t          $ r}t          d|           |d }~ww xY w)Nz!Got multiple action responses: %sr   actionzFinal Answeroutputaction_inputzCould not parse LLM output: )
r   
isinstancelistloggerwarningr   r   get	Exceptionr   )selfr   responsees       `/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/agents/output_parsers/json.pyparsezJSONAgentOutputParser.parse+   s    	V*400H(D)) 'BHMMM#A;!^33"Hh~.F#GNNN"X&^R(H(H$    	V 	V 	V'(Mt(M(MNNTUU	Vs   A*B -*B 
B:"B55B:c                    dS )Nz
json-agent )r   s    r   _typezJSONAgentOutputParser._type;   s    |    N)r   r   r   r   )r   r   )__name__
__module____qualname____doc__r   propertyr!   r    r"   r   r   r      sV         6V V V V     X  r"   r   )
__future__r   loggingtypingr   langchain_core.agentsr   r   langchain_core.exceptionsr   "langchain_core.output_parsers.jsonr   langchain.agents.agentr	   	getLoggerr#   r   r   r    r"   r   <module>r0      s    " " " " " "        : : : : : : : : ; ; ; ; ; ; B B B B B B 4 4 4 4 4 4		8	$	$. . . . .- . . . . .r"   