dh$aSrSSKJr SSKrSSKJrJrJrJrJ r J r J r J r SSK r SSKJr SSKJr SSK JrJrJrJrJrJr SSKJrJrJr SS KJrJr SS KJ r SS K!J"r" SS K#J$r$ SS jr%SS.SSjjr&SSjr'"SS5r(g)%SQLAlchemy wrapper around a database.) annotationsN)AnyDictIterableListLiteralOptionalSequenceUnion) deprecated) get_from_env)MetaDataTable create_engineinspectselecttext)URLEngineResult)ProgrammingErrorSQLAlchemyError) CreateTable) Executable)NullTypec <SUSSUSS[US53$)NzName: namez , Unique: uniquez , Columns: column_namesstr)indexs b/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/utilities/sql_database.py _format_indexr%s7 vz%/):;~./0 2z...)suffixc[U[5(aUS::aU$[U5U::aU$USU[U5- RSS5SU-$)zQ Truncate a string to a certain number of words, based on the max string length. rN ) isinstancer"lenrsplit)contentlengthr's r$ truncate_wordr0!s[ gs # #v{ 7|v )Vc&k) * 1 1#q 9! # UHnUT;dM UTRU4v M g7fN)_custom_table_info).0table intersectionselfs r$ 'SQLDatabase.__init__..zs2+4EL(8//674s ((viewsbindonlyr6N)_engine_schemar4r _inspectorsetlistget_table_namesget_view_names _all_tables_include_tables_ignore_tablesget_usable_table_names_usable_tablesr+int TypeError_sample_rows_in_table_info_indexes_in_table_infor?dictrB_max_string_length _view_supportr _metadatareflect)rCenginer6metadata ignore_tablesinclude_tablessample_rows_in_table_infoindexes_in_table_infocustom_table_info view_supportmax_string_lengthlazy_table_reflectionmissing_tables usable_tablesrBs` @r$__init__SQLDatabase.__init__=sK  mST T!$,,/ 000? @@Lt--V-2#%   !11D4D4DDN %n%55KL5Bc-0su   !0043C3CCN $^$44JK335 4Ac-0tGWGW3S99JK K*C'&;#"3  " "d55t<<3 t667DDTEUEUVL&*+!44+'D # #4)!/XZ$ NN " ""\\$--.|| # %r&c >U=(d 0nU"[U40UD640UD6$)z'Construct a SQLAlchemy engine from URI.)r)cls database_uri engine_argskwargs _engine_argss r$from_uriSQLDatabase.from_uris(#(b =>>I&IIr&z0.3.18zFor performing structured retrieval using Databricks SQL, see the latest best practices and recommended APIs at https://docs.unitycatalog.io/ai/integrations/langchain/ insteadz1.0)messageremovalc SSKJn Sn SSKJn U "5n U R n Uc [SSU 5nU (a U ROSn Uc [SS U 5nUc"UcU (a U RnO [S 5eU(aU(a [S 5eU(aS U3nOS U3nSUSUSUSUSU3 nUR"SXS.UD6$![a [S5ef=f![[ 4a Sn Nf=f)a Class method to create an SQLDatabase instance from a Databricks connection. This method requires the 'databricks-sql-connector' package. If not installed, it can be added using `pip install databricks-sql-connector`. Args: catalog (str): The catalog name in the Databricks database. schema (str): The schema name in the catalog. host (Optional[str]): The Databricks workspace hostname, excluding 'https://' part. If not provided, it attempts to fetch from the environment variable 'DATABRICKS_HOST'. If still unavailable and if running in a Databricks notebook, it defaults to the current workspace hostname. Defaults to None. api_token (Optional[str]): The Databricks personal access token for accessing the Databricks SQL warehouse or the cluster. If not provided, it attempts to fetch from 'DATABRICKS_TOKEN'. If still unavailable and running in a Databricks notebook, a temporary token for the current user is generated. Defaults to None. warehouse_id (Optional[str]): The warehouse ID in the Databricks SQL. If provided, the method configures the connection to use this warehouse. Cannot be used with 'cluster_id'. Defaults to None. cluster_id (Optional[str]): The cluster ID in the Databricks Runtime. If provided, the method configures the connection to use this cluster. Cannot be used with 'warehouse_id'. If running in a Databricks notebook and both 'warehouse_id' and 'cluster_id' are None, it uses the ID of the cluster the notebook is attached to. Defaults to None. engine_args (Optional[dict]): The arguments to be used when connecting Databricks. Defaults to None. **kwargs (Any): Additional keyword arguments for the `from_uri` method. Returns: SQLDatabase: An instance of SQLDatabase configured with the provided Databricks connection details. Raises: ValueError: If 'databricks-sql-connector' is not found, or if both 'warehouse_id' and 'cluster_id' are provided, or if neither 'warehouse_id' nor 'cluster_id' are provided and it's not executing inside a Databricks notebook. r)sqlzfdatabricks-sql-connector package not found, please install with `pip install databricks-sql-connector`N) get_contexthostDATABRICKS_HOST api_tokenDATABRICKS_TOKENz6Need to provide either 'warehouse_id' or 'cluster_id'.z/Can't have both 'warehouse_id' or 'cluster_id'.z/sql/1.0/warehouses/z/sql/protocolv1/o/0/zdatabricks://token:@z ?http_path=z &catalog=z&schema=)rorp) databricksrx ImportError!dbruntime.databricks_repl_contextrybrowserHostNameAttributeErrorrapiToken clusterIdr4rs)rncatalogr6rzr| warehouse_id cluster_idrprqrxcontextry default_hostdefault_api_token http_pathuris r$from_databricksSQLDatabase.from_databrickssFx  &  E!mG"22L <(9<HD07G,,T  $[2DFWXI  J$6$..  L JNO O .|n=I.zl;I")AdV4" 9WIXfX G ||PPPPS :  ^, L sCC'C$'C<;C<ctSSKJn U"XX4U5nURUS9$![a [S5ef=f)a Class method to create an SQLDatabase instance from a CnosDB connection. This method requires the 'cnos-connector' package. If not installed, it can be added using `pip install cnos-connector`. Args: url (str): The HTTP connection host name and port number of the CnosDB service, excluding "http://" or "https://", with a default value of "127.0.0.1:8902". user (str): The username used to connect to the CnosDB service, with a default value of "root". password (str): The password of the user connecting to the CnosDB service, with a default value of "". tenant (str): The name of the tenant used to connect to the CnosDB service, with a default value of "cnosdb". database (str): The name of the database in the CnosDB tenant. Returns: SQLDatabase: An instance of SQLDatabase configured with the provided CnosDB connection details. r)make_cnosdb_langchain_uri)rozRcnos-connector package not found, please install with `pip install cnos-connector`)cnosdb_connectorrrsr)rnurluserpasswordtenantdatabaserrs r$ from_cnosdbSQLDatabase.from_cnosdbsJ<  B+CxRC<up[%U R$5[&LdM#U R(R+U 5 M@ [1[3U 5R5UR55n U R75nU(aR0nU R H/nUR8(dMUR8UUR 'M1 U(a USUS3- nUR<=(d UR>nU(aUS - nUR<(aUS URAU 5S 3- nUR>(aUS URCU 5S 3- nU(aUS - nU RU5 GM U RE5 S RGU 5nU$s snfs snf![,ai [/U R 5R#5H>up[%U R$5[&LdM#U R(R+U 5 M@ GNf=f![:a [S5ef=f) 6Get information about specified tables. Follows best practices as specified in: Rajkumar et al, 2022 (https://arxiv.org/abs/2204.00498) If `sample_rows_in_table_info`, the specified number of sample rows will be appended to each table description. This can increase performance as demonstrated in the paper. z table_names r<rFsqlitesqlite_z /* Column Comments: z */z:Column comments are available on PostgreSQL, MySQL, Oraclez /* z*/z )$rTrM differencer4r] sorted_tablesrr^r\rJrNrKr startswithr?appendcolumnsitemstyper_columnsremoverrZr"rcompilerstripcomment ExceptionrYrX_get_table_indexes_get_sample_rowssortjoin)rC table_namesget_col_commentsall_table_namesritblmetadata_table_names to_reflect meta_tablestablesrAkv create_tablercolumn_comments_dictcolumnhas_extra_info final_strs r$rSQLDatabase.get_table_info?s_557  " -88IN </??U!VWW)O48NN4P4PQ4PS4PQ)C0D,EE  NN " "((\\*%|| # ~~33 3xx3// \\X-#((2E2Ei2P 3   E&&5::9P9P+P d55ejjAB 1!MM//1DAAFF|x/--a02{5199$,,GHL(//12J  +-("'--!>>>@F0=#0,"78L7MTR ++Nt/N/N h& **4#:#:5#A"B"EE ..4#8#8#?"@CC d" MM* %c!d  KK' M R &" 1 /557DAAFF|x/--a08 1,!$TsIM-M .M?M%9M$"M$ O+-O$A O4OOO0cURRUR5nSR[ [ U55nSU3$)NrzTable Indexes: )rL get_indexesrrmapr%)rCrAindexesindexes_formatteds r$rSQLDatabase._get_table_indexess?//--ejj9 IIc-&AB!"3!455r&cD[U5RUR5nSRURVs/sHo3R PM sn5nUR R5nURU5n[[SU55nSSS5 SRWVs/sHnSRU5PM sn5n URSUR SUSU 3$s snf!,(df  Nd=fs snf![a Sn NJf=f)N cHUVs/sHn[U5SSPM sn$s snf)Ndr!)lsis r$.SQLDatabase._get_sample_rows..s "#="QCF4CL"#=#=srz rows from z table: ) rlimitrXrrrrJconnectexecuterNrr) rCrAcommandcol columns_str connectionsample_rows_result sample_rowsrowsample_rows_strs r$rSQLDatabase._get_sample_rowss-%%d&E&EFiiU]] C]c] CD  !%%':%/%7%7%@""=?QR (#ii;(O;C3;(OPO../{5::,im2 ! '!D(')P  ! O !s;C5D4(C:D3D  D: D D DD parametersexecution_optionscU=(d 0nU=(d 0nURR5nURGbjURS:XaUR SUR4US9 GO39??3DE3Daiik3DFEFe^#)??#4L#/#7Rl>R>R>T=UFh&!O" !R%RY" !@#A"Z FE" !Z s$HJ12J, 7J1 J1,J11 KclURXXES9nUS:XaU$UVVV s/sH9nUR5VV s0sHupU[XRS9_M sn nPM; n nnn U(d(U V s/sHn [ U R 55PM n n U (dg[ U 5$s sn nfs sn nnfs sn f)zExecute a SQL command and return a string representing the results. If the statement returns rows, a string of the results is returned. If the statement returns no rows, an empty string is returned. rr)r/r)rrr0r[tuplevaluesr") rCrrinclude_columnsrrrrrvalueresrs r$runSQLDatabase.runs z  H M  &'WWY %.MF e4K4KLL%.   256#35&#C6s8O  7sB*B$B*-"B1$B*c`URU5$![anSU3sSnA$SnAff=f)rError: N)rr4)rCres r$get_table_info_no_throw#SQLDatabase.get_table_info_no_throw@s7 !&&{3 3 ! *QC=  !s -(--cdURUUUUUS9$![anSU3sSnA$SnAff=f)a Execute a SQL command and return a string representing the results. If the statement returns rows, a string of the results is returned. If the statement returns no rows, an empty string is returned. If the statement throws an error, the error message is returned. )rrr rN)rr)rCrrr rrrs r$ run_no_throwSQLDatabase.run_no_throwPsM !88%"3 /   ! *QC=  !s /*//c|[UR55nUR5nUSRU5S.$)z4Return db context that you may want in agent prompt.z, )rr)rNrTrr)rCrrs r$rySQLDatabase.get_contextls746689 113 (;9OPPr&) rQr?rJrSrRrYrLr[r]rXrKrUr\) NNNNFNFi,F)r_rr6 Optional[str]r`zOptional[MetaData]raOptional[List[str]]rbrrcrVrdboolreOptional[dict]rfrrgrVrhrr>)rozUnion[str, URL]rprrqrreturnr9)NNNNN)rr"r6r"rzrr|rrrrrrprrqrrr9)z127.0.0.1:8902rootrcnosdbpublic) rr"rr"rr"rr"rr"rr9)rr")rz Iterable[str])NF)rrrrrr")rArrr")r) rUnion[str, Executable]rLiteral['all', 'one', 'cursor']rOptional[Dict[str, Any]]rr%rz'Union[Sequence[Dict[str, Any]], Result])rF) rr#rr$r rrr%rr%r1Union[str, Sequence[Dict[str, Any]], Result[Any]])rrrr") rr"rzLiteral['all', 'one']r rrr%rr%rr&)rzDict[str, Any])__name__ __module__ __qualname____firstlineno____doc__rk classmethodrsr rrpropertyrrTrOrrrrrrrrry__static_attributes__rr&r$r9r9:sy/ !%'+-1.2)*&+,0"!$&+NNN% N + N , N$'N $N*NNN $N`'+J%J$J J  JJ ##'&*$(&*^Q^Q^Q ^Q ! ^Q $ ^Q"^Q$^Q^Q ^Q^Q@$ & && &  &  & &&P))> %=uM-N-%% QVY.YIMY Yv6  B27\ 046: \'\/\ - \ 4 \ 1\B27 % # 046:#'#/# # - #4# ;#J!&(- % ! 046:!!%! ! - !4! ;!8Qr&r9)r#z+sqlalchemy.engine.interfaces.ReflectedIndexrr")r.rr/rVr'r"rr")r6r"rr"))r+ __future__rr2typingrrrrr r r r sqlalchemylangchain_core._apir langchain_core.utilsrrrrrrrsqlalchemy.enginerrrsqlalchemy.excrrsqlalchemy.schemarsqlalchemy.sql.expressionrsqlalchemy.typesrr%r0r7r9rr&r$r9sb+" PPP*-21<)0%?D FvQvQr&