Checking supports_function_calling isntead of gpt models

2026-01-24 23:58:15 +00:00 · 2024-09-23 16:23:38 -03:00
parent 3b6d1838b4
commit 493f046c03
8 changed files with 34 additions and 61 deletions
--- a/tests/utilities/evaluators/test_task_evaluator.py
+++ b/tests/utilities/evaluators/test_task_evaluator.py
@@ -25,6 +25,7 @@ def test_evaluate_training_data(converter_mock):
    }
    agent_id = "agent_id"
    original_agent = MagicMock()
+    original_agent.llm.supports_function_calling.return_value = False
    function_return_value = TrainingTaskEvaluation(
        suggestions=[
            "The initial output was already good, having a detailed explanation. However, the improved output "