o }!g@sdZddlmZmZmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZdZ Gddde Z!dS)zOpenAI Agent. Simple wrapper around AgentRunner + OpenAIAgentWorker. For the legacy implementation see: ```python from llama_index.agent.legacy.openai.base import OpenAIAgent ``` )AnyDictListCallableOptionalType)OpenAIAgentWorker) AgentRunner)CallbackManager) ChatMessage)LLM)ChatMemoryBuffer) BaseMemory)ObjectRetriever)Settings)BaseTool)OpenAI)OpenAIToolCallc s&eZdZdZdeddddfdeedededee d e d e d e d e ed e eede eegefddffdd Zedddddededddddf de eed e eede ede ee de edeed e d e d e d e ede e de ee de eegefdeddfddZZS) OpenAIAgentzOpenAI agent. Subclasses AgentRunner with a OpenAIAgentWorker. For the legacy implementation see: ```python from llama_index.agent.legacy.openai.base import OpenAIAgent ``` FautoNtoolsllmmemoryprefix_messagesverbosemax_function_callsdefault_tool_choicecallback_managertool_retrievertool_call_parserreturnc s>|p|j}tj|| |||||| d} tj| ||||ddS)z Init params.)rrrrrrrr )rrrrN)rr from_toolssuper__init__) selfrrrrrrrrrr step_engine __class___/mnt/skqttb/ctump_chatbot/chatbot/lib/python3.10/site-packages/llama_index/agent/openai/base.pyr$0s$   zOpenAIAgent.__init__ chat_history memory_cls system_promptkwargsc Ks|pg}|pg}|p tj}t|tstd| dur| |_|p%|j||d}|jjs3td|j d| durF| dur?tdt | ddg} | pIg} |||||| ||| | | d S) zCreate an OpenAIAgent from a list of tools. Similar to `from_defaults` in other classes, this method will infer defaults for a variety of parameters, including the LLM, if they are not specified. zllm must be a OpenAI instanceN)rz Model name z( does not support function calling API. z5Cannot specify both system_prompt and prefix_messagessystem)contentrole) rrrrrrrrrr ) rr isinstancer ValueErrorr from_defaultsmetadatais_function_calling_modelmodelr )clsrrrr+rr,rrrrr-rr r.r)r)r*r"Qs>   zOpenAIAgent.from_tools)__name__ __module__ __qualname____doc__DEFAULT_MAX_FUNCTION_CALLSrrrrr boolintstrrr rrrrr$ classmethodr r rrr" __classcell__r)r)r'r*r$s    !        rN)"r<typingrrrrrrllama_index.agent.openai.stepr"llama_index.core.agent.runner.baser llama_index.core.callbacksr llama_index.core.base.llms.typesr llama_index.core.llms.llmr *llama_index.core.memory.chat_memory_bufferr llama_index.core.memory.typesrllama_index.core.objects.baserllama_index.core.settingsrllama_index.core.toolsrllama_index.llms.openairllama_index.llms.openai.utilsrr=rr)r)r)r*s