a bg$@s,dZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddlmZmZddlmZmZmZddlmZmZmZddlmZdd lmZmZerdd l m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+d+ddddddddddddddd d!d"d#d$d%d&d'd(d)d*Z,dS),z SQL agent.) annotations) TYPE_CHECKINGAnyDictListLiteralOptionalSequenceUnioncast) AIMessage SystemMessage)BasePromptTemplatePromptTemplate)ChatPromptTemplateHumanMessagePromptTemplateMessagesPlaceholder)SQL_FUNCTIONS_SUFFIX SQL_PREFIX SQL_SUFFIX)SQLDatabaseToolkit)InfoSQLDatabaseToolListSQLDatabaseTool) AgentExecutor AgentType)BaseCallbackManager)BaseLanguageModel)BaseTool) SQLDatabaseN forceF)dbpromptrzOptional[SQLDatabaseToolkit]zCOptional[Union[AgentType, Literal['openai-tools', 'tool-calling']]]zOptional[BaseCallbackManager]z Optional[str]zOptional[List[str]]intz Optional[int]zOptional[float]strboolzOptional[Dict[str, Any]]zSequence[BaseTool]zOptional[SQLDatabase]zOptional[BasePromptTemplate]rr)llmtoolkit agent_typecallback_managerprefixsuffixformat_instructionsinput_variablestop_kmax_iterationsmax_execution_timeearly_stopping_methodverboseagent_executor_kwargs extra_toolsr$r%kwargsreturnc! sddlm}m}m}m}ddlm}m}m}ddl m }|durP|durPt d|r`|r`t d|pnt ||d}|px|j }|t|}|durt}dur|j|j|d }nd jvrʈjt|d d jvrj|jd tfdddDrV|}djvr,j|dddd|D}djvrVj|dddd|D}||j kr̈durddlm}|p|j}d|d||ptg}t||f|||dgdgd|}n||jkr@durt t!t|dt"dt#|pt$dt%d d!g}t&'||f|||dgdgd|}n|d"vrΈdurt t!t|dt"dt#|ptt$dt%d d!g}t&'||d#kr|||} n |||} |f| dgdgd|}nt d$|d%|fd&|||| | | | d'| piS)(a8 Construct a SQL agent from an LLM and toolkit or database. Args: llm: Language model to use for the agent. If agent_type is "tool-calling" then llm is expected to support tool calling. toolkit: SQLDatabaseToolkit for the agent to use. Must provide exactly one of 'toolkit' or 'db'. Specify 'toolkit' if you want to use a different model for the agent and the toolkit. agent_type: One of "tool-calling", "openai-tools", "openai-functions", or "zero-shot-react-description". Defaults to "zero-shot-react-description". "tool-calling" is recommended over the legacy "openai-tools" and "openai-functions" types. callback_manager: DEPRECATED. Pass "callbacks" key into 'agent_executor_kwargs' instead to pass constructor callbacks to AgentExecutor. prefix: Prompt prefix string. Must contain variables "top_k" and "dialect". suffix: Prompt suffix string. Default depends on agent type. format_instructions: Formatting instructions to pass to ZeroShotAgent.create_prompt() when 'agent_type' is "zero-shot-react-description". Otherwise ignored. input_variables: DEPRECATED. top_k: Number of rows to query for by default. max_iterations: Passed to AgentExecutor init. max_execution_time: Passed to AgentExecutor init. early_stopping_method: Passed to AgentExecutor init. verbose: AgentExecutor verbosity. agent_executor_kwargs: Arbitrary additional AgentExecutor args. extra_tools: Additional tools to give to agent on top of the ones that come with SQLDatabaseToolkit. db: SQLDatabase from which to create a SQLDatabaseToolkit. Toolkit is created using 'db' and 'llm'. Must provide exactly one of 'db' or 'toolkit'. prompt: Complete agent prompt. prompt and {prefix, suffix, format_instructions, input_variables} are mutually exclusive. **kwargs: Arbitrary additional Agent args. Returns: An AgentExecutor with the specified agent_type agent. Example: .. code-block:: python from langchain_openai import ChatOpenAI from langchain_community.agent_toolkits import create_sql_agent from langchain_community.utilities import SQLDatabase db = SQLDatabase.from_uri("sqlite:///Chinook.db") llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0) agent_executor = create_sql_agent(llm, db=db, agent_type="tool-calling", verbose=True) r)create_openai_functions_agentcreate_openai_tools_agentcreate_react_agentcreate_tool_calling_agent)r RunnableAgentRunnableMultiActionAgentrNz@Must provide exactly one of 'toolkit' or 'db'. Received neither.z=Must provide exactly one of 'toolkit' or 'db'. Received both.)r)r$)dialectr1r1)r1r@)r@c3s|]}|jvVqdS)N)r0).0keyr%r#y/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/agent_toolkits/sql/base.py z#create_sql_agent..) table_info table_namesrG)rGcSsg|]}t|ts|qSr#) isinstancerrAZtoolr#r#rD sz$create_sql_agent..rH)rHcSsg|]}t|ts|qSr#)rIrrJr#r#rDrKsrCz z{tools}inputoutput)runnableZinput_keys_argZreturn_keys_arg)contentz{input}Zagent_scratchpad)Z variable_name) openai-toolsz tool-callingrPz Agent type z not supported at the moment. Must be one of 'tool-calling', 'openai-tools', 'openai-functions', or 'zero-shot-react-description'.zSQL Agent Executor)nameagenttoolsr,r5r2r3r4)(Zlangchain.agentsr:r;r<r=langchain.agents.agentrr>r?langchain.agents.agent_typesr ValueErrorrZZERO_SHOT_REACT_DESCRIPTIONZ get_toolslistrformatr@r0partialr'anyZ get_contextZlangchain.agents.mrklr%ZFORMAT_INSTRUCTIONSjoinrrZ from_templateZOPENAI_FUNCTIONSr r rr rrrZ from_messages)!r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r$r%r8r:r;r<r=rr>r?rrSZ db_contextZ react_prompttemplaterRmessagesrNr#rCrDcreate_sql_agent.sI                        r^)NNNNNNNr r!Nr"FNr#)-__doc__ __future__rtypingrrrrrrr r r Zlangchain_core.messagesr r Zlangchain_core.promptsrrZlangchain_core.prompts.chatrrrZ-langchain_community.agent_toolkits.sql.promptrrrZ.langchain_community.agent_toolkits.sql.toolkitrZ+langchain_community.tools.sql_database.toolrrrTrrUrZlangchain_core.callbacksrZlangchain_core.language_modelsrZlangchain_core.toolsrZ*langchain_community.utilities.sql_databaserr^r#r#r#rDsB ,