蘺hx ddlmZmZmZmZddlmZddlmZddl m Z ddl m Z ddl mZmZddlmZmZded efd ZGd d eZGd deZ ddededee ded eeeefeff dZdS))ListOptional TypedDictUnion) SQLDatabase)BaseLanguageModel)StrOutputParser)BasePromptTemplate)RunnableRunnableParallel)PROMPT SQL_PROMPTStextreturnc*|SN)strip)rs k/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/langchain/chains/sql_database/query.py_stripr s ::<<ceZdZUdZeed<dS)SQLInputInput for a SQL Chain.questionN)__name__ __module__ __qualname____doc__str__annotations__rrrrs  MMMMMrrc4eZdZUdZeed<eeed<dS)SQLInputWithTablesrrtable_names_to_useN)rrrrrr rr!rrr#r#s1  MMMS !!!!!rr#Nllmdbpromptkc"||n(jtvrtjntdfdfdd}djvr fd|d<t |z|dg zt ztzS) aDCreate a chain that generates SQL queries. *Security Note*: This chain generates SQL queries for the given database. The SQLDatabase class provides a get_table_info method that can be used to get column information as well as sample data from the table. To mitigate risk of leaking sensitive data, limit permissions to read and scope to the tables that are needed. Optionally, use the SQLInputWithTables input type to specify which tables are allowed to be accessed. Control access to who can submit requests to this chain. See https://python.langchain.com/docs/security for more information. Args: llm: The language model to use db: The SQLDatabase to generate the query for prompt: The prompt to use. If none is provided, will choose one based on dialect. Defaults to None. k: The number of results per select statement to return. Defaults to 5. Returns: A chain that takes in a question and generates a SQL query that answers that question. Nc|ddzS)Nrz SQLQuery: r!)xs rz(create_sql_query_chain..Fs1Z=>9rcSrr!)_r)s rr-z(create_sql_query_chain..Gs1rcV|dS)Nr$) table_names)get_table_infoget)r,r's rr-z(create_sql_query_chain..Hs, 1 1233!2! ! r)inputtop_k table_infodialectcjfSr)r7)r/r' prompt_to_uses rr-z(create_sql_query_chain..Msrz=&Arz SQLResult:)stop)r7rr input_variablesr bindr r)r&r'r(r)inputsr9s ` ` @rcreate_sql_query_chainr>sD { " "#BJ/  99    FM111AAAAAy    (((( ) ) *       r)Nr%)typingrrrr*langchain_community.utilities.sql_databaserlangchain_core.language_modelsrlangchain_core.output_parsersr langchain_core.promptsr langchain_core.runnablesr r $langchain.chains.sql_database.promptr rrrrr#intr>r!rrrGs333333333333BBBBBB<<<<<<999999555555????????DDDDDDDDy """""""",0  77 77 ' (7 7 eH001367 777777r