github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-28 01:19:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: fix tool calling integration tests for AI21 Jamba models (#25771)

Browse Source 
Ignore specific integration tests that handles specific tool calling
cases that will soon be fixed.
This commit is contained in:
pazshalev 2024-08-27 15:54:51 +03:00 committed by GitHub
parent 99f9a664a5
commit 995305fdd5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
libs/partners/ai21/tests/integration_tests/test_standard.py
View File

@ -71,3 +71,19 @@ class TestAI21Jamba1_5(BaseTestAI21):
return { return {
"model": "jamba-1.5-mini", "model": "jamba-1.5-mini",
} }
@pytest.mark.xfail(reason="Prompt doesn't generate tool calls for Jamba 1.5.")
def test_tool_calling(self, model: BaseChatModel) -> None:
super().test_tool_calling(model)
@pytest.mark.xfail(reason="Prompt doesn't generate tool calls for Jamba 1.5.")
def test_tool_calling_with_no_arguments(self, model: BaseChatModel) -> None:
super().test_tool_calling_with_no_arguments(model)
@pytest.mark.xfail(reason="Requires tool calling & stream - still WIP")
def test_structured_output(self, model: BaseChatModel) -> None:
super().test_structured_output(model)
@pytest.mark.xfail(reason="Requires tool calling & stream - still WIP")
def test_structured_output_pydantic_2_v1(self, model: BaseChatModel) -> None:
super().test_structured_output_pydantic_2_v1(model)