dhfQSSKrSSKrSSKrSSKJrJr SSKJrJrJ r J r J r J r SSK r SSKJr SSKJr SSKJr SSKJrJrJrJrJr S/r"S S \\5rS \ \\4S \4S jrS \ \\4S \4SjrS \ \\4S \4Sjr"SS\5r "SS\5r!S \4Sjr"S \4Sjr#S \4Sjr$S\S \%4Sjr&S\S\ \%S \4Sjr'S\S \4Sjr(\"SSSS 9"S!S\55r)g)"N)ABCabstractmethod)AnyCallableDictListMappingOptional) deprecated)CallbackManagerForLLMRun)LLM) BaseModel ConfigDictField PrivateAttrmodel_validator Databricksc \rSrSr%Sr\\S'\\S'S\S\S\S\4S jrS\S\4S jr S\S\S\4S jr \ SS\S \ \ S\4S\4Sjj5r\S\4Sj5rSrg )_DatabricksClientBasez0A base JSON API client that talks to Databricks.api_url api_tokenmethodurlrequestreturncSSUR30n[R"XXCS9nUR(d%[ SUR SUR 35eUR5$)N AuthorizationzBearer )rrheadersjsonzHTTP z error: )rrequestsrok ValueError status_codetextr )selfrrrrresponses [/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/llms/databricks.pyr_DatabricksClientBase.requestse"gdnn-=$>?##G {{uX%9%9$:(8==/RS S}}c(URSUS5$)NGETr)r&rs r(_get_DatabricksClientBase._get&s||E3--r*c&URSX5$)NPOSTr-)r&rrs r(_post_DatabricksClientBase._post)s||FC11r*Ntransform_output_fn.cgN)r&rr4s r(post_DatabricksClientBase.post,sr*cg)NFr7r&s r(llm_DatabricksClientBase.llm1sr*r7r6)__name__ __module__ __qualname____firstlineno____doc__str__annotations__rrr.r2rr rr8propertyboolr<__static_attributes__r7r*r(rrs: LNccc...22s2s2PT19(38:L1M Tr*rr'rcUSSS$)Nchoicesrr%r7r's r(_transform_completionsrK6s I q !& ))r*cUSSS$)N candidatesrr%r7rJs r(_transform_llama2_chatrN:s L !! $V ,,r*cUSSSS$)NrIrmessagecontentr7rJs r(_transform_chatrR>s I q !) ,Y 77r*c^\rSrSr%Sr\\S'\\S'\\S'Sr\\S'Sr \ \S 'Sr \ \\S 'S \4U4S jjr \S \ 4Sj5r\"SS9\S\\\4S \4Sj55rSS\S\ \S\4S \4SjjrSrU=r$) _DatabricksServingEndpointClientBz:An API client that talks to a Databricks serving endpoint.host endpoint_namedatabricks_uriNclientFexternal_or_foundationtaskdatac >[TU]"S0UD6 SSKJn U"UR5UlUR RUR5nURSS5R5S;Ul URcURS5Ul gg![ an[ S5UeSnAff=f) Nr)get_deploy_clientzMFailed to create the client. Please install mlflow with `pip install mlflow`. endpoint_type)external_modelfoundation_model_apir[r7) super__init__mlflow.deploymentsr^rXrY ImportError get_endpointrWgetlowerrZr[)r&r\r^eendpoint __class__s r(rd)_DatabricksServingEndpointClient.__init__Ls  4   <+D,?,?@DK;;++D,>,>?&.ll?B&G&M&M&OT ' # 99  V,DI  C  sB"" B=, B88B=rc URS;$)N) llm/v1/chatllm/v1/completions llama2/chat)r[r;s r(r<$_DatabricksServingEndpointClient.llmasyyPPPr*beforemodevaluesc@SU;aUSnUSnSUSUS3nXAS'U$)NrrVrWhttps://z/serving-endpoints/z /invocationsr7)clsrvrVrWrs r( set_api_url,_DatabricksServingEndpointClient.set_api_urlesA F "&>D"?3M &9- UG '9  r*rr4.cUR(akURRURUS9nU(aU"U5$URS:Xa [ U5$URS:Xa [ U5$U$SU/0nURRURUS9nUSn[U[5(aUSOUnURS:Xa [U5$U(aU"U5$U$)N)rkinputsrorpdataframe_records predictionsrrq) rZrYpredictrWr[rRrK isinstancelistrN)r&rr4respwrapped_requestr'predspreds r(r8%_DatabricksServingEndpointClient.postos  & &;;&&0B0B7&SD"*400yyM)&t,,22-d33K 3WI>O{{**++O+H]+E)%6658EDyyM)-d330C&t, M Mr*)rYrZr[r6)r>r?r@rArBrCrDrYrrZrFr[r rdrEr<r classmethodrrzrr8rG __classcell__rls@r(rTrTBsD IFC#(D(D(3--s-*QTQQ(#c3hC$QUNN19(38:L1MN NNr*rTc\rSrSr%Sr\\S'\\S'\\S'\"SS9\S\ \\ 4S \ 4S j55r SS \ S \ \ S\4S \ 4SjjrSrg )#_DatabricksClusterDriverProxyClientzBAn API client that talks to a Databricks cluster driver proxy app.rV cluster_idcluster_driver_portrsrtrvrcNSU;aUSnUSnUSnSUSUSU3nXQS'U$)NrrVrrrxz/driver-proxy-api/o/0//r7)ryrvrVrportrs r(rz/_DatabricksClusterDriverProxyClient.set_api_urlsP F "&>D -J/0D &r?r@rArBrCrDrrrrrzr rr8rGr7r*r(rrsL IO(#c3hC$QUJJ19(38:L1MJ JJr*rcPSSKJn U"5$![a [S5ef=f)z_Get the notebook REPL context if running inside a Databricks notebook. Returns None otherwise. r get_contextzBCannot access dbruntime, not running inside a Databricks notebook.)!dbruntime.databricks_repl_contextrrfrs r(get_repl_contextrs1 A}   P   s %c.[R"S5nU(d([5RnU(d [ S5eUR S5R S5RS5nU$![ an[ SU35eSnAff=f)zsGet the default Databricks workspace hostname. Raises an error if the hostname cannot be automatically determined. DATABRICKS_HOSTz(context doesn't contain browserHostName.zshost was not set and cannot be automatically inferred. Set environment variable 'DATABRICKS_HOST'. Received error: Nrxzhttp://r)osgetenvrbrowserHostNamer# Exceptionlstriprstrip)rVrjs r(get_default_hostrs 99& 'D  #%55D !KLL ;;z " ) )) 4 ; ;C @D K KKL#O  s&A77 BBBc[R"S5=n(aU$[5RnU(d [ S5eU$![ an[ SU35eSnAff=f)zsGet the default Databricks personal access token. Raises an error if the token cannot be automatically determined. DATABRICKS_TOKENz!context doesn't contain apiToken.zyapi_token was not set and cannot be automatically inferred. Set environment variable 'DATABRICKS_TOKEN'. Received error: N)rrrapiTokenr#r)rrjs r(get_default_api_tokenrs~II011y1 $&// @A A    HHIs L   s&A A'A""A'r\cp[U[5(dgSn[[R"X55$)zJChecks if a data is a valid hexadecimal string using a regular expression.Fz^[0-9a-fA-F]+$)rrCrFrematch)r\patterns r(_is_hex_stringrs+ dC G ' ((r*allow_dangerous_deserializationcU(d [S5eSSKnUR[R U55$![an[SU35eSnAff=f![an[SU35eSnAff=f)z3Loads a pickled function from a hexadecimal string.aWThis code relies on the pickle module. You will need to set allow_dangerous_deserialization=True if you want to opt-in to allow deserialization of data using pickle.Data can be compromised by a malicious actor if not handled properly to include a malicious payload that when deserialized with pickle can execute arbitrary code on your machine.rN*Please install cloudpickle>=2.0.0. Error: zFFailed to load the pickled function from a hexadecimal string. Error: )r# cloudpicklerloadsbytesfromhex)r\rrrjs r( _load_pickled_fn_from_hex_stringrs + A  K   t!455 KEaSIJJK  TUVTW X   s,=#A AAA A:'A55A:fncSSKnURU5R 5$![an[SU35eSnAff=f![an[SU35eSnAff=f)z6Pickles a function and returns the hexadecimal string.rNrzFailed to pickle the function: )rrr#dumpshex)rrrjs r(_pickle_fn_to_hex_stringrsxK@  $((** KEaSIJJK @:1#>??@s)&A A>A A#AA#z0.3.3z1.0z#databricks_langchain.ChatDatabricks)sinceremovalalternative_importc ^\rSrSr%Sr\"\S9r\\ S'\"\ S9r \\ S'Sr \ \\ S'Sr\ \\ S'Sr\ \\ S 'Sr\ \\\4\ S 'Sr\ \\ S 'Sr\ \S \4\ S 'Sr\\ S'Sr\\ S'Sr\\ S'Sr\ \\\ S'Sr\ \\ S'\"\S9r\\\4\ S'Sr \ \\ S'Sr!\"\ S'\#"5r$\%\ S'\&"SS9r'\(S\\\44Sj5r)\*"SS 9\+S!\\\4S\4S"j55r,S#\4U4S$jjr-\(S\\\44S%j5r.\(S\/\\44S&j5r0\(S\4S'j5r1S-S(\S\ \\S)\ \2S*\S\4 S+jjr3S,r4U=r5$).ri a}Databricks serving endpoint or a cluster driver proxy app for LLM. It supports two endpoint types: * **Serving endpoint** (recommended for both production and development). We assume that an LLM was deployed to a serving endpoint. To wrap it as an LLM you must have "Can Query" permission to the endpoint. Set ``endpoint_name`` accordingly and do not set ``cluster_id`` and ``cluster_driver_port``. If the underlying model is a model registered by MLflow, the expected model signature is: * inputs:: [{"name": "prompt", "type": "string"}, {"name": "stop", "type": "list[string]"}] * outputs: ``[{"type": "string"}]`` If the underlying model is an external or foundation model, the response from the endpoint is automatically transformed to the expected format unless ``transform_output_fn`` is provided. * **Cluster driver proxy app** (recommended for interactive development). One can load an LLM on a Databricks interactive cluster and start a local HTTP server on the driver node to serve the model at ``/`` using HTTP POST method with JSON input/output. Please use a port number between ``[3000, 8000]`` and let the server listen to the driver IP address or simply ``0.0.0.0`` instead of localhost only. To wrap it as an LLM you must have "Can Attach To" permission to the cluster. Set ``cluster_id`` and ``cluster_driver_port`` and do not set ``endpoint_name``. The expected server schema (using JSON schema) is: * inputs:: {"type": "object", "properties": { "prompt": {"type": "string"}, "stop": {"type": "array", "items": {"type": "string"}}}, "required": ["prompt"]}` * outputs: ``{"type": "string"}`` If the endpoint model signature is different or you want to set extra params, you can use `transform_input_fn` and `transform_output_fn` to apply necessary transformations before and after the query. )default_factoryrVrNrWrr model_kwargstransform_input_fn.r4 databricksrXg temperaturenstop max_tokens extra_paramsr[Fr_clientforbid)extrarcURURS.nUR(aURUS'URbURUS'U$)N)rrrr)rrrr)r&paramss r( _llm_paramsDatabricks._llm_paramssN ++"  99!YYF6N ?? &#'??F<  r*rsrtrvcfURS5nURS5nU(aU(a [S5eU(aSUS'O3U(aO+[5R=n(aXAS'[S5eURS5nU(aU(a [S5eU(aSUS'O,Uc [S5e[ U5S ::a[S U35eURS 5=n(aS U;dS 5eSU;dS5eU$![an[SU35eSnAff=f)NrrWz-Cannot set both endpoint_name and cluster_id.z"Context doesn't contain clusterId.zuNeither endpoint_name nor cluster_id was set. And the cluster_id cannot be automatically determined. Received error: rz6Cannot set both endpoint_name and cluster_driver_port.z? =UV V ,0F( ) (N $ % *<=PSU;a2[US5(a[USURS5S9US'SU;a2[US5(a[USURS5S9US'[TU]"S 0UD6 UR bUR b [S5eUR b[R"S[5 UR(aE[URURURURUR S9UlgUR$(aKUR&(a:[)URURUR$UR&S9Ulg[S 5e) Nrr)r\rr4z.Cannot set both extra_params and extra_params.z->-JMN N    * MMN"    ;YY.."00#22YY DL__!9!9>YY..??$($<$< DLV r*cURURURURURUR UR URURURURURURcSO[UR5URcSS.$[UR5S.$)zReturn default params.N)rVrWrrrXrrrrrrr[rr4)rVrWrrrXrrrrrrr[rrr4r;s r(_default_paramsDatabricks._default_paramssII!////#'#;#;"11 --++II// --II&&.#')$*A*AB''/$(#  &*$*B*BC'  r*cUR$r6)rr;s r(_identifying_paramsDatabricks._identifying_paramss###r*cg)zReturn type of llm.rr7r;s r( _llm_typeDatabricks._llm_typesr*r run_managerkwargsc SU0nURR(aURUR5 URUR=(d UR 5 URU5 U(aX%S'UR (aUR "S0UD6nURRXPRS9$)zAQueries the LLM endpoint with the given prompt and stop sequence.rr)r4r7) rr<updaterrrrr8r4)r&rrrrrs r(_callDatabricks._call$s$,V"4 <<   NN4++ ,t((=D,=,=>v "FO  " "--88G||  >V>V WWr*)r)NN)6r>r?r@rArBrrrVrCrDrrrWr rrrrrrrr4rXrfloatrrrrrdictrr[rrFrrrr model_configrErrrrrdrr rrr rrGrrs@r(rr s /b&67D#7+@AIsA$(M8C=' !%J $*.#- .2L(4S>*1 .2*1 9=(38"45<'NC&FK# AsJ7 $D(49 $ $J $3#(#>L$sCx.>7D(3--2#T1&1]G "2L T#s(^  (#*DcN*t*$*X,s,\ c3h  0$WS#X%6$$3%):> XXtCy!X67 X  X XXr*)*rrrabcrrtypingrrrrr r r!langchain_core._apir langchain_core.callbacksr langchain_core.language_modelsr pydanticrrrrr__all__rrCrKrNrRrTrrrrrFrrrrr7r*r(rsT #??*=. .Is@*T#s(^**-T#s(^--8d38n88FN'<FNRJ*?J2  #  #(s&)))  08   8 @ @c @  < kXkX  kXr*