dh+SrSSKrSSKJr SSKJrJrJrJr SSK J r J r SSK J r SSKJrJrJr SSKJr SS KJrJrJr SS KJrJr \R6"\5r"S S \ 5r"S S\ 5r"SS\ 5r g)z.Tools for interacting with a Power BI dataset.N) perf_counter)AnyDictOptionalTuple)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseTool) ConfigDictFieldmodel_validator)_import_tiktoken)BAD_REQUEST_RESPONSEDEFAULT_FEWSHOT_EXAMPLESRETRY_RESPONSE)PowerBIDataset json_to_mdc \rSrSr%SrSr\\S'Sr\\S'Sr \ \S'\ "S S 9r \ \S '\r\\\S '\ "\S S 9r\\\ 4\S'Sr\\S'Sr\\S'Sr\\\S'\"S S9r\"SS9\S\S\4Sj55rS\S\\4SjrS%S\S\\S\ S\4SjjrS%S\S\\ S\ S\4Sjjr!S \\\ 4S\"\\\\ 44S!jr#S"\S\"\$\44S#jr%S$r&g)&QueryPowerBIToolz%Tool for querying a Power BI Dataset. query_powerbinamea Input to this tool is a detailed question about the dataset, output is a result from the dataset. It will try to answer the question using the dataset, and if it cannot, it will ask for clarification. Example Input: "How many rows are in table1?" descriptionN llm_chainTexcludepowerbiexamples)default_factoryr session_cachemax_iterationsioutput_token_limittiktoken_model_namearbitrary_types_allowedbefore)modevaluesreturncUSnURRH)nUS;dM [SURR5e U$)z8Make sure the LLM chain has the correct input variables.r) tool_inputtablesschemasrzrLLM chain for QueryPowerBITool must have input variables ['tool_input', 'tables', 'schemas', 'examples'], found %s)promptinput_variables ValueError)clsr)rvars ^/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/tools/powerbi/tool.py"validate_llm_chain_input_variables3QueryPowerBITool.validate_llm_chain_input_variables.sV ;' ##33CII I$$444  r,c@XR;agURU$)zCheck if the input is present in the cache. If the value is a bad request, overwrite with the escalated version, if not present return None.N)r )selfr,s r4 _check_cacheQueryPowerBITool._check_cache=s$ // /!!*--r7 run_managerkwargsc URU5=n(a[RSX5 U$[RSU5 URR UUR R5UR R5URU(aUR5OSS9nUS:XaXPRU'URU$[RSU5 [5nUR RUS9n[5n [RS U5 [RS X- S 35 URU5upU b$S U ;aS URU'URU$UR!SS5n U (a6XR":a'UR%[&R("XU S9UU S-S9$U (aU O[*R("U S9URU'URU$![a+nSU3URU'URUsSnA$SnAff=f):Execute the query, return the results or an error message.Found cached result for %s: %s%Running PBI Query Tool with input: %sNr,r-r.r callbacksError on call to LLM: I cannot answer thisz PBI Query: %scommandPBI Result: %sPBI Query duration: 0.6f TokenExpiredzCAuthentication token expired or invalid, please try reauthenticate. iterationsrr,queryerrorr,r<rLrO)r:loggerdebuginforpredictrget_table_names get_schemasr get_child Exceptionr rrun _parse_outputgetr"_runrformatr r9r,r<r=cacherNexc start_time pbi_resultend_timeresultrOrLs r4r^QueryPowerBITool._runFs'%%j1 15 1 LL9: ML 2 KK? LNN**%||335 0025@+//1d +E * *-2  z *%%j1 1 $e,!^ \\%%e%4 > %z2 +H,A$+GHI**:6   5!8U   z *%%j1 1ZZ a0 Z"5"5599)00)e(%>  F"6"="=E"J :&!!*--A 2/EcU-KD  z *%%j1 1 2sBH I H?9I?Ic# URU5=n(a[RSX5 US3$[RSU5 URR UUR R5UR R5URU(aUR5OSS9IShvN nUS:XaXPRU'URU$[RSU5 [5nUR RUS 9IShvN n[5n [RS U5 [RS X- S 35 URU5upU b*S U ;dSU ;aSURU'URU$UR!SS5n U (a>XR":a/UR%[&R("XU S9UU S-S9IShvN $U (aU O[*R("U S9URU'URU$GNt![a+nSU3URU'URUsSnA$SnAff=fGNKN{7f)r?r@z3, from cache, you have already asked this question.rANrBrDrEz PBI Query: %srFrHrIrJrK TokenErrorzkAuthentication token expired or invalid, please try to reauthenticate or check the scope of the credential.rLrrMrPrQrR)r:rSrTrUrapredictrrWrXrrYrZr rarunr\r]r"_arunrr_rr`s r4rlQueryPowerBITool._arun|sU%%j1 15 1 LL9: MWOP P 2 KK? L..11%||335 0025@+//1d 2E * *-2  z *%%j1 1 OU+!^ <<,,U,;; > %z2 +H,A$+GHI**:6   .E"9\U=R}   z *%%j1 1ZZ a0 Z"5"55)00)e(%> $ F"6"="=E"J :&!!*--Q 2/EcU-KD  z *%%j1 1 2<sg4I2BH5>H2?H5AI2"I-#CI24I05=I22H55 I*? I%I* I2%I**I20I2rdchSU;atUSSSSSn[U5S:Xa[RS5 g[U5nUR U5upEU(aSUSUR S 3S 4$US 4$S U;a/S US ;aS US S ;aS US S S SS4$S US 4$S U4$)z2Parse the output of the query to a markdown table.resultsrr-rowsz%0 records in result, query was valid.)Nz^0 rows returned, this might be correct, but please validate if all filter values were correct?z[Result too large, please try to be more specific or use the `TOPN` function. The result is z tokens long, the limit is z tokens.NrOz pbi.errordetailsdetail)lenrSrUr_result_too_larger#)r9rdrprftoo_longlengths r4r\QueryPowerBITool._parse_outputs!  "i(+H5a8@D4yA~ CD %F#55f= HqrxqyzUVZVmVmUnnvw4<  j z'22G!4[!AAZ0=iHKHUUUG,, ,Zr7rfcUR(ae[5nURUR5n[UR U55n[ R SU5 X@R:U4$g)z!Tokenize the output of the query.zResult length: %s)Fr)r$rencoding_for_modelrsencoderSrUr#)r9rf tiktoken_encodingrvs r4rt"QueryPowerBITool._result_too_largesb  # #(*I 33D4L4LMH01F KK+V 4333V; ;r7N)'__name__ __module__ __qualname____firstlineno____doc__rstr__annotations__rrrr rrrrrdictr rr"intr#r$r model_configr classmethodr5r:r r^rrlrr\boolrt__static_attributes__r~r7r4rrs/D#K Is#D1G^16Hhsm6$)$$MM4S>MNC"")-#- $L(#    $ .s.x}.<@4.4.784. 4. 4.rAE4.4.<=4. 4. 4.l sCx. x}hsm+ , <dCi0@r7rc\rSrSr%SrSr\\S'Sr\\S'\ "SS9r \ \S '\ "SS 9r SS \S \\S\4SjjrSS \S \\S\4SjjrSrg )InfoPowerBIToolz2Tool for getting metadata about a PowerBI Dataset.schema_powerbirz Input to this tool is a comma-separated list of tables, output is the schema and sample rows for those tables. Be sure that the tables actually exist by calling list_tables_powerbi first! Example Input: "table1, table2, table3" rTrrr%Nr,r<r*cVURRURS55$)z4Get the schema for tables in a comma-separated list., )rget_table_infosplitr9r,r<s r4r^InfoPowerBITool._runs$ ||**:+;+;D+ABBr7cr# URRURS55IShvN $N7f)Nr)raget_table_inforrs r4rlInfoPowerBITool._aruns- \\11*2B2B42HIIIIs .757r~rrrrrrrrrrr rrr rrr r^rrlrr~r7r4rrs< D# K $D1G^1 $L<@CC78C CAEJJ<=J JJr7rc\rSrSr%SrSr\\S'Sr\\S'\ "SS9r \ \S '\ "SS 9r SS \\S \\S\4SjjrSS \\S \\S\4SjjrSrg )ListPowerBIToolzTool for getting tables names.list_tables_powerbirzUInput is an empty string, output is a comma separated list of tables in the database.rTrrr%Nr,r<r*cTSRURR55$zGet the names of the tables.rjoinrrWrs r4r^ListPowerBITool._runs yy55788r7c\# SRURR55$7frrrs r4rlListPowerBITool._aruns# yy55788s*,r~)NNrr~r7r4rrs(%D#%nKn#D1G^1 $L %);?9SM9789 9%)@D9SM9<=9 99r7r)!rloggingtimertypingrrrrlangchain_core.callbacksrr langchain_core.toolsr pydanticr r r &langchain_community.chat_models.openair(langchain_community.tools.powerbi.promptrrr%langchain_community.utilities.powerbirr getLoggerrrSrrrr~r7r4rsq4--*77C M   8 $xDJhJ@9h9r7