tj*B(ddlmZddlZddlZddlZddlmZmZddlm Z m Z ddl m Z ddl mZmZmZddlmZmZmZmZmZddlmZdd lmZdd lmZdd lmZdd lm Z dd l!m"Z"ddl#m$Z$e rddl%m&Z&m'Z'ej(e)Z*GddZ+dS)) annotationsN) AsyncIteratorIterator) TYPE_CHECKINGAny)UUID) AgentAction AgentFinish AgentStep)AsyncCallbackManagerAsyncCallbackManagerForChainRunCallbackManagerCallbackManagerForChainRun Callbacks)dumpd)RunInfo) AddableDict)BaseTool)get_color_mapping)RUN_KEY)asyncio_timeout) AgentExecutorNextStepOutputc|eZdZUdZ d;dddddddddZed?dZ e jd@dZ edAd Z ed=d!Z dBd"Z dBd#Z dCd)ZdDd,ZdEd.ZdFd2ZdGd4ZdHd5ZdId6ZdJd9ZdKd:ZdS)LAgentExecutorIteratorzIterator for AgentExecutor.NF)tagsmetadatarun_namerun_idinclude_run_info yield_actionsagent_executorrinputsr callbacksrrlist[str] | Nonerdict[str, Any] | Noner str | Noner UUID | Noner boolr!c||_||_||_||_||_||_||_||_| |_| dS)aInitialize the `AgentExecutorIterator`. Initialize the `AgentExecutorIterator` with the given `AgentExecutor`, inputs, and optional callbacks. Args: agent_executor: The `AgentExecutor` to iterate over. inputs: The inputs to the `AgentExecutor`. callbacks: The callbacks to use during iteration. tags: The tags to use during iteration. metadata: The metadata to use during iteration. run_name: The name of the run. run_id: The ID of the run. include_run_info: Whether to include run info in the output. yield_actions: Whether to yield actions as they are generated. N) _agent_executorr#r$rrrrr r!reset) selfr"r#r$rrrrr r!s e/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langchain_classic/agents/agent_iterator.py__init__zAgentExecutorIterator.__init__+sW: . "      0* dict[str, str]_inputsreturnc|jS)z"The inputs to the `AgentExecutor`.)r2r-s r.r#zAgentExecutorIterator.inputs\s |r0NonecD|j||_dSN)r" prep_inputsr2)r-r#s r.r#zAgentExecutorIterator.inputsas*66v>> r0c|jS)z$The `AgentExecutor` to iterate over.)r+r5s r.r"z$AgentExecutorIterator.agent_executores ##r0c,||_|j|_dSr8)r+r#)r-r"s r.r"z$AgentExecutorIterator.agent_executorjs-k r0dict[str, BaseTool]c.d|jjDS)z!A mapping of tool names to tools.ci|] }|j| Sname.0tools r. z:AgentExecutorIterator.name_to_tool_map..ssFFFD 4FFFr0)r"toolsr5s r.name_to_tool_mapz&AgentExecutorIterator.name_to_tool_mappsGFD,?,EFFFFr0cPtd|jjDddgS)z"A mapping of tool names to colors.cg|] }|j Sr?r@rBs r. z7AgentExecutorIterator.color_mapping..ys = = =4TY = = =r0greenred)excluded_colors)rr"rFr5s r. color_mappingz#AgentExecutorIterator.color_mappingus:! = =4#6#< = = =$e,    r0ctdg|_d|_d|_t j|_dS)zReset the iterator to its initial state. Reset the iterator to its initial state, clearing intermediate steps, iterations, and time elapsed. z0(Re)setting AgentExecutorIterator to fresh statergN)loggerdebugintermediate_steps iterations time_elapsedtime start_timer5s r.r,zAgentExecutorIterator.reset}s?  GHHHAC)++r0c|xjdz c_tj|jz |_td|j|jdS)z?Increment the number of iterations and update the time elapsed.z$Agent Iterations: %s (%.2fs elapsed)N)rSrUrVrTrPrQr5s r.update_iterationsz'AgentExecutorIterator.update_iterationssU 1 IKK$/9 2 O       r0outputsdict[str, Any] run_manager"AI,=GCI, I>3&I,I>, I66I>9I6:I>>4K5 K>KKnext_step_output+AgentFinish | list[tuple[AgentAction, str]]rctdt|tr1td|||S|j|tdt|dkr;|d}|j |}||||St|S) zProcess the output of the next step. Process the output of the next step, handling AgentFinish and tool return cases. z$Processing output of Agent loop stepzBHit AgentFinish: _return -> on_chain_end -> run final output logicr\+Updated intermediate_steps with step outputrXrNrj) rPrQrqr _returnrRextendlenr"_get_tool_returnrr-rr\next_step_action tool_returns r.rsz/AgentExecutorIterator._process_next_step_output>s  ;<<< & 4 4 K LLT   << 0k<JJ J &&'7888 BCCC  A % %/2 ->>?OPPK&||K[|III-=>>>>r0r cKtdt|tr7td|||d{VS|j|tdt|dkrA|d}|j |}||||d{VSt|S) zProcess the output of the next async step. Process the output of the next async step, handling AgentFinish and tool return cases. z*Processing output of async Agent loop stepzCHit AgentFinish: _areturn -> on_chain_end -> run final output logicrNrrXrr) rPrQrqr _areturnrRrrr"rrrs r.rz0AgentExecutorIterator._aprocess_next_step_output[s  ABBB & 4 4 R LLU   '7[QQQQQQQQ Q &&'7888 BCCC  A % %/2 ->>?OPPK&!]];K]PPPPPPPPP-=>>>>r0ctd|jjj|jj|jfi|j}|||S)zuStop the iterator. Stop the iterator and raise a StopIteration exception with the stopped response. ;Stopping agent prematurely due to triggering stop conditionr) rPwarningr" _action_agentreturn_stopped_responseearly_stopping_methodrRr#rr-r\r|s r.rvzAgentExecutorIterator._stopxsi TUUUJ$2J   5  #  k   ||F |<<&>&>&>&@O-O-O-O-b????:????: = = = = D D D D E E E E E E E E E Er0r), __future__rrloggingrUcollections.abcrrtypingrruuidrlangchain_core.agentsr r r langchain_core.callbacksr r rrrlangchain_core.load.dumprlangchain_core.outputsrlangchain_core.runnables.utilsrlangchain_core.toolsrlangchain_core.utils.inputrlangchain_classic.schemar#langchain_classic.utilities.asynciorlangchain_classic.agents.agentrr getLoggerrrPrr?r0r.rs"""""" 33333333 +***********666666))))))888888,,,,,,??????MLLLLLLLL  8 $ $HEHEHEHEHEHEHEHEHEHEr0