7|he+dZddlZddlmZddlmZmZmZmZddl m Z m Z ddl m Z ddlmZmZmZddlmZdd lmZmZmZdd lmZmZej6eZGd d e ZGd de ZGdde Z y)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 eZdZUdZdZeed<dZeed<dZe ed<e d Z e ed <e Zeeed <e ed Zeee fed <dZeed<dZeed<dZeeed<edZedededefdZdedeefdZ d#dedeede defdZ d#dedeede defdZ deee fde!eeee ffd Z"d!ede!e#effd"Z$y)$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)modevaluesreturnc|d}|jjD]&}|dvstd|jj|S)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 e/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/tools/powerbi/tool.py"validate_llm_chain_input_variablesz3QueryPowerBITool.validate_llm_chain_input_variables.s[ ;' ##33 CII I$$44   r+c>||jvry|j|S)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 r3 _check_cachezQueryPowerBITool._check_cache=s& T// /!!*--r5 run_managerkwargsc |j|x}rtjd|||S tjd||jj ||j j|j j|j|r|jnd}|dk(r||j|<|j|Stjd|t}|j j|}t} tjd |tjd | |z d |j|\} } | "d | vrd |j|<|j|S|j!dd} | r<| |j"kr-|j%t'j(||| || dzS| r| nt+j(| |j|<|j|S#t$r+}d||j|<|j|cYd}~Sd}~wwxYw):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+r9rIrL)r8loggerdebuginforpredictrget_table_names get_schemasr get_child Exceptionrrrun _parse_outputgetr!_runrformatr r7r+r9r:cacherKexc start_time pbi_resultend_timeresultrLrIs r3r[zQueryPowerBITool._runFs3%%j1 15 1 LL9:u ML 2 KK? LNN**%||335 0025@+//1d +E * *-2D  z *%%j1 1 $e,!^ \\%%e%4 > %z2 +Hz,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 H= H82H=8H=cK|j|x}rtjd|||dS tjd||jj ||j j|j j|j|r|jndd{}|dk(r||j|<|j|Stjd|t}|j j| d{}t} tjd |tjd | |z d |j|\} } | &d | vsd| vrd|j|<|j|S|j!dd} | rD| |j"kr5|j%t'j(||| || dzd{S| r| nt+j(| |j|<|j|S7r#t$r+}d||j|<|j|cYd}~Sd}~wwxYw7E7ww)r<r=z3, from cache, you have already asked this question.r>Nr?rArBz PBI Query: %srCrErFrGrH TokenErrorzkAuthentication token expired or invalid, please try to reauthenticate or check the scope of the credential.rIrrJrMrNrO)r8rPrQrRrapredictrrTrUrrVrWrrarunrYrZr!_arunrr\rr]s r3rhzQueryPowerBITool._arun|sc%%j1 15 1 LL9:u MWOP P 2 KK? L..11%||335 0025@+//1d 2E * *-2D  z *%%j1 1 OU+!^ <<,,U,;; > %z2 +Hz,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<sg0I)BH-8H*9H-=A"I)I$ CI)/I'0:I)*H-- I!6 II!I)I!!I)'I)racNd|vro|ddddd}t|dk(rtjdyt|}|j |\}}|rd|d|j d d fS|d fSd |vr+d |d vrd |d d vrd |d d d ddfSd |d fSd |fS)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.NrLz pbi.errordetailsdetail)lenrPrRr_result_too_larger")r7rarkrctoo_longlengths r3rYzQueryPowerBITool._parse_outputs#  "i(+H5a8@D4yA~ CD %F#55f= HfqrxqyzUVZVmVmUnnvw4<  j z'22G!4[!AAZ0=iHKHUUUG,, ,Zr5rcc|jrft}|j|j}t|j |}t j d|||jkD|fSy)z!Tokenize the output of the query.zResult length: %s)Fr)r#rencoding_for_modelrnencoderPrRr")r7rc tiktoken_encodingrqs r3roz"QueryPowerBITool._result_too_largesd  # #(*I 33D4L4LMH01F KK+V 4D333V; ;r5N)%__name__ __module__ __qualname____doc__rstr__annotations__rrrr rrrrrdictrrr!intr"r#r model_configr classmethodr4r8r r[rrhrrYboolror5r3rrs/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@r5rceZdZUdZdZeed<dZeed<edZ e ed<e d Z dd ed e ed efdZ dd ed e ed efdZy )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+r9r)cV|jj|jdS)z4Get the schema for tables in a comma-separated list., )rget_table_infosplitr7r+r9s r3r[zInfoPowerBITool._runs$ ||**:+;+;D+ABBr5crK|jj|jdd{S7w)Nr)raget_table_inforrs r3rhzInfoPowerBITool._aruns. \\11*2B2B42HIIIIs .757rwrxryrzr{rr|r}rr rrr rrr r[rrhrr5r3rrs< D# K $D1G^1 $L<@CC78C CAEJJ<=J Jr5rceZdZUdZdZeed<dZeed<edZ e ed<e d Z dd e ed e ed efdZ dd e ed e ed efdZy )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+r9r)cTdj|jjSzGet the names of the tables.rjoinrrTrs r3r[zListPowerBITool._runs yy55788r5c\Kdj|jjSwrrrs r3rhzListPowerBITool._aruns$ yy55788s*,)NNrrr5r3rrs(%D#%nKn#D1G^1 $L %);?9SM9789 9%)@D9SM9<=9 9r5r)!r{loggingtimertypingrrrrlangchain_core.callbacksrr langchain_core.toolsr pydanticr r r &langchain_community.chat_models.openair(langchain_community.tools.powerbi.promptrrr%langchain_community.utilities.powerbirr getLoggerrxrPrrrrr5r3rss4--*77C M   8 $xDJhJ@9h9r5