github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-24 03:52:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

experimental[patch]: fix zero-shot pandas agent (#17442)

Browse Source
This commit is contained in:
Bagatur
2024-02-12 21:58:35 -08:00
committed by GitHub
parent 37e1275f9e
commit c0ce93236a
7 changed files with 135 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
libs/experimental/tests/unit_tests/agents/__init__.py Normal file
View File

0
libs/experimental/tests/unit_tests/agents/agent_toolkits/__init__.py Normal file
View File

0
libs/experimental/tests/unit_tests/agents/agent_toolkits/pandas/__init__.py Normal file
View File

15
libs/experimental/tests/unit_tests/agents/agent_toolkits/pandas/test_base.py Normal file
View File

@@ -0,0 +1,15 @@
import sys
import pytest
from langchain_experimental.agents import create_pandas_dataframe_agent
from tests.unit_tests.fake_llm import FakeLLM
@pytest.mark.requires("pandas", "tabulate")
@pytest.mark.skipif(sys.version_info < (3, 9), reason="requires python3.9 or higher")
def test_create_pandas_dataframe_agent() -> None:
import pandas as pd
create_pandas_dataframe_agent(FakeLLM(), pd.DataFrame())
create_pandas_dataframe_agent(FakeLLM(), [pd.DataFrame(), pd.DataFrame()])