.ipynb
.pdf
PowerBI Dataset Agent
 Contents 
Some notes
Initialization
Example: describing a table
Example: simple query on a table
Example: running queries
Example: add your own few-shot prompts
PowerBI Dataset Agent#
This notebook showcases an agent designed to interact with a Power BI Dataset. The agent is designed to answer more general questions about a dataset, as well as recover from errors.
Note that, as this agent is in active development, all answers might not be correct. It runs against the executequery endpoint, which does not allow deletes.
Some notes#
It relies on authentication with the azure.identity package, which can be installed with pip install azure-identity. Alternatively you can create the powerbi dataset with a token as a string without supplying the credentials.
You can also supply a username to impersonate for use with datasets that have RLS enabled.
The toolkit uses a LLM to create the query from the question, the agent uses the LLM for the overall execution.
Testing was done mostly with a text-davinci-003 model, codex models did not seem to perform ver well.
Initialization#
from langchain.agents.agent_toolkits import create_pbi_agent
from langchain.agents.agent_toolkits import PowerBIToolkit
from langchain.utilities.powerbi import PowerBIDataset
from langchain.chat_models import ChatOpenAI
from langchain.agents import AgentExecutor
from azure.identity import DefaultAzureCredential
fast_llm = ChatOpenAI(temperature=0.5, max_tokens=1000, model_name="gpt-3.5-turbo", verbose=True)
smart_llm = ChatOpenAI(temperature=0, max_tokens=100, model_name="gpt-4", verbose=True)
toolkit = PowerBIToolkit(
rtdocs_stable/api.python.langchain.com/en/stable/modules/agents/toolkits/examples/powerbi.html