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

@@ -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):