enable test on ollama

langchain-ai · Dec 17, 2024 · a61cca2 · a61cca2
1 parent e002031
commit a61cca2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/libs/partners/ollama/tests/integration_tests/chat_models/test_chat_models_standard.py b/libs/partners/ollama/tests/integration_tests/chat_models/test_chat_models_standard.py
@@ -22,6 +22,10 @@ def chat_model_params(self) -> dict:
     def supports_image_inputs(self) -> bool:
         return True
 
+    @property
+    def supports_json_mode(self) -> bool:
+        return True
+
     @pytest.mark.xfail(
         reason=(
             "Fails with 'AssertionError'. Ollama does not support 'tool_choice' yet."