updating tests

This commit is contained in:
João Moura
2024-09-23 03:58:41 -03:00
parent 7d981ba8ce
commit 59e51f18fd
98 changed files with 36353 additions and 16646 deletions

View File

@@ -44,7 +44,7 @@ class TestCrewEvaluator:
== "Evaluator agent for crew evaluation with precise capabilities to evaluate the performance of the agents in the crew based on the tasks they have performed"
)
assert agent.verbose is False
assert agent.llm == "gpt-4o-mini"
assert agent.llm.model == "gpt-4o-mini"
def test_evaluation_task(self, crew_planner):
evaluator_agent = Agent(

View File

@@ -2,6 +2,7 @@ import json
from unittest.mock import MagicMock, Mock, patch
import pytest
from crewai.llm import LLM
from crewai.utilities.converter import (
Converter,
ConverterError,
@@ -213,11 +214,13 @@ def test_get_conversion_instructions_non_gpt():
# Tests for is_gpt
def test_is_gpt_true():
assert is_gpt("gpt-4") is True
llm = LLM(model="gpt-4")
assert is_gpt(llm) is True
def test_is_gpt_false():
assert is_gpt("lol-4") is False
llm = LLM(model="lol-4")
assert is_gpt(llm) is False
class CustomConverter(Converter):