o Zh$@s dZddlmZddlmZmZmZmZmZm Z m Z m Z m Z ddl mZmZddlmZmZddlmZmZmZddlmZmZmZddlmZdd lmZmZervdd l m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'ddl(m)Z)ddl*m+Z+              d=dddd>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)dbpromptllmrtoolkitOptional[SQLDatabaseToolkit] agent_typeCOptional[Union[AgentType, Literal['openai-tools', 'tool-calling']]]callback_managerOptional[BaseCallbackManager]prefix Optional[str]suffixformat_instructionsinput_variablesOptional[List[str]]top_kintmax_iterations Optional[int]max_execution_timeOptional[float]early_stopping_methodstrverboseboolagent_executor_kwargsOptional[Dict[str, Any]] extra_toolsSequence[BaseTool]r$Optional[SQLDatabase]r%Optional[BasePromptTemplate]kwargsrreturnrc! sddlm}m}m}m}ddlm}m}m}ddl m }|dur(|dur(t d|r0|r0t d|p7t ||d}|p<|j }|t|}|durKt}durX|j|j|d }nPd jvrejt|d d jvrqj|jd tfdddDr|}djvrj|dddd|D}djvrj|dddd|D}||j krވdurddlm}|p|j}d|d||ptg}t||d(|||dgdgd|}n||jkrdurt t!t|dt"dt#|pt$dt%d d!g}t&'||d(|||dgdgd|}nO|d"vr^dur>t t!t|dt"dt#|p1t$dt%d d!g}t&'||d#krJ|||} n|||} |d(| dgdgd|}nt d$|d%|d(d&|||| | | | d'| pviS))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$)dialectr3r3)r3rK)rKc3s|]}|jvVqdS)N)r1).0keyr%r#b/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/agent_toolkits/sql/base.py sz#create_sql_agent..) table_info table_namesrQ)rQcSg|] }t|ts|qSr#) isinstancerrLZtoolr#r#rO  z$create_sql_agent..rR)rRcSrSr#)rTrrUr#r#rOrVrWrNz z{tools}inputoutput)runnableZinput_keys_argZreturn_keys_arg)contentz{input}Zagent_scratchpad)Z variable_name) openai-toolsz tool-callingr\z 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+r;r5r7r9r#)(Zlangchain.agentsrErFrGrHlangchain.agents.agentrrIrJlangchain.agents.agent_typesr ValueErrorrZZERO_SHOT_REACT_DESCRIPTIONZ get_toolslistrformatrKr1partialr:anyZ get_contextZlangchain.agents.mrklr%ZFORMAT_INSTRUCTIONSjoinrrZ from_templateZOPENAI_FUNCTIONSr r rr rrrZ from_messages)!r&r'r)r+r-r/r0r1r3r5r7r9r;r=r?r$r%rCrErFrGrHrrIrJrr_Z db_contextZ react_prompttemplater^messagesrZr#rNrOcreate_sql_agent.sI                      rj)NNNNNNNr r!Nr"FNr#)&r&rr'r(r)r*r+r,r-r.r/r.r0r.r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@r$rAr%rBrCrrDr)-__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.toolrrr`rrarZlangchain_core.callbacksrZlangchain_core.language_modelsrZlangchain_core.toolsrZ*langchain_community.utilities.sql_databaserrjr#r#r#rOsD ,