Ng dZddlmZddlmZmZmZmZmZddl m Z ddl m Z ddl mZmZddlmZddlmZerdd lmZd d d eed d d d d d f d(d'Zd S))zPower BI agent.) annotations) TYPE_CHECKINGAnyDictListOptional)BaseCallbackManager)BaseLanguageModel)POWERBI_PREFIXPOWERBI_SUFFIX)PowerBIToolkit)PowerBIDataset AgentExecutorN Fllmr toolkitOptional[PowerBIToolkit]powerbiOptional[PowerBIDataset]callback_managerOptional[BaseCallbackManager]prefixstrsuffixformat_instructions Optional[str]examplesinput_variablesOptional[List[str]]top_kintverboseboolagent_executor_kwargsOptional[Dict[str, Any]]kwargsrreturnrc ddlm} ddlm}ddlm}|#|t dt|||}|}|r|j n |j j }|d|ini}|d|||j |f| |  | ||d ||| d |Dd| }| j d|||| d| piS)a(Construct a Power BI agent from an LLM and tools. Args: llm: The language model to use. toolkit: Optional. The Power BI toolkit. Default is None. powerbi: Optional. The Power BI dataset. Default is None. callback_manager: Optional. The callback manager. Default is None. prefix: Optional. The prefix for the prompt. Default is POWERBI_PREFIX. suffix: Optional. The suffix for the prompt. Default is POWERBI_SUFFIX. format_instructions: Optional. The format instructions for the prompt. Default is None. examples: Optional. The examples for the prompt. Default is None. input_variables: Optional. The input variables for the prompt. Default is None. top_k: Optional. The top k for the prompt. Default is 10. verbose: Optional. Whether to print verbose output. Default is False. agent_executor_kwargs: Optional. The agent executor kwargs. Default is None. kwargs: Any. Additional keyword arguments. Returns: The agent executor. rr) ZeroShotAgent)LLMChainNz0Must provide either a toolkit or powerbi dataset)rrrr)r!)tables)rrr)rpromptrr#cg|] }|j S)name).0tools k/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/agent_toolkits/powerbi/base.py z$create_pbi_agent..Us333Tty333) llm_chain allowed_tools)agenttoolsrr#r/)langchain.agentsrlangchain.agents.mrkl.baser*langchain.chains.llmr+ ValueErrorr get_tools table_namesr create_promptformatfrom_agent_and_tools)rrrrrrrrrr!r#r%r'rr*r+r9r, prompt_paramsr8s r3create_pbi_agentrDsH/.....888888------ ?OPP P cHMMM     E$+ LW 1LF  *  344  M (.=.}}5}11888GG /   .    43U333     E .= - )    ! &B   r5)rr rrrrrrrrrrrrrrrr r!r"r#r$r%r&r'rr(r)__doc__ __future__rtypingrrrrrlangchain_core.callbacksr langchain_core.language_modelsr 1langchain_community.agent_toolkits.powerbi.promptr r 2langchain_community.agent_toolkits.powerbi.toolkitr %langchain_community.utilities.powerbirr:rrDr/r5r3rMs!"""""";;;;;;;;;;;;;;888888<<<<<<NMMMMM@@@@@@/...... )-(,6:  )-"+/6:IIIIIIIr5