github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-09 06:53:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

standard-tests: fix default fixtures (#20520)

Browse Source
This commit is contained in:
Erick Friis
2024-04-16 09:12:36 -07:00
committed by GitHub
parent 5acc7ba622
commit 77eba10f47
2 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
libs/standard-tests/langchain_standard_tests/integration_tests/chat_models.py
View File

@@ -33,13 +33,16 @@ class ChatModelIntegrationTests(ABC):
def chat_model_has_tool_calling(
self, chat_model_class: Type[BaseChatModel]
) -> bool:
return hasattr(chat_model_class, "bind_tools")
return chat_model_class.bind_tools is not BaseChatModel.bind_tools
@pytest.fixture
def chat_model_has_structured_output(
self, chat_model_class: Type[BaseChatModel]
) -> bool:
return hasattr(chat_model_class, "with_structured_output")
return (
chat_model_class.with_structured_output
is not BaseChatModel.with_structured_output
)
def test_invoke(
self, chat_model_class: Type[BaseChatModel], chat_model_params: dict