mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
06e524416c
# Powerbi API wrapper bug fix + integration tests - Bug fix by removing `TYPE_CHECKING` in in utilities/powerbi.py - Added integration test for power bi api in utilities/test_powerbi_api.py - Added integration test for power bi agent in agent/test_powerbi_agent.py - Edited .env.examples to help set up power bi related environment variables - Updated demo notebook with working code in docs../examples/powerbi.ipynb - AzureOpenAI -> ChatOpenAI Notes: Chat models (gpt3.5, gpt4) are much more capable than davinci at writing DAX queries, so that is important to getting the agent to work properly. Interestingly, gpt3.5-turbo needed the examples=DEFAULT_FEWSHOT_EXAMPLES to write consistent DAX queries, so gpt4 seems necessary as the smart llm. Fixes #4325 ## Before submitting Azure-core and Azure-identity are necessary dependencies check integration tests with the following: `pytest tests/integration_tests/utilities/test_powerbi_api.py` `pytest tests/integration_tests/agent/test_powerbi_agent.py` You will need a power bi account with a dataset id + table name in order to test. See .env.examples for details. ## Who can review? @hwchase17 @vowelparrot --------- Co-authored-by: aditya-pethe <adityapethe1@gmail.com>
11 lines
509 B
Python
11 lines
509 B
Python
def test_power_bi_can_be_imported() -> None:
|
|
"""Test that powerbi tools can be imported.
|
|
|
|
The goal of this test is to verify that langchain users will not get import errors
|
|
when loading powerbi related code if they don't have optional dependencies
|
|
installed.
|
|
"""
|
|
from langchain.tools.powerbi.tool import QueryPowerBITool # noqa
|
|
from langchain.agents.agent_toolkits import PowerBIToolkit, create_pbi_agent # noqa
|
|
from langchain.utilities.powerbi import PowerBIDataset # noqa
|