andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 08:38:30 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: Sort imports to fix linting issues

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-12 06:25:41 +00:00
parent 420f826e56
commit abc71be5f6
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/crew_test.py
View File

@@ -6,6 +6,7 @@ from concurrent.futures import Future
from unittest import mock from unittest import mock
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
import agentops
import instructor import instructor
import pydantic_core import pydantic_core
import pytest import pytest
@@ -15,6 +16,7 @@ from crewai.agents.cache import CacheHandler
from crewai.crew import Crew from crewai.crew import Crew
from crewai.crews.crew_output import CrewOutput from crewai.crews.crew_output import CrewOutput
from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource from crewai.knowledge.source.string_knowledge_source import StringKnowledgeSource
from crewai.llm import LLM
from crewai.memory.contextual.contextual_memory import ContextualMemory from crewai.memory.contextual.contextual_memory import ContextualMemory
from crewai.process import Process from crewai.process import Process
from crewai.project import crew from crewai.project import crew
@@ -70,9 +72,6 @@ def test_agentops_initialization_without_api_key(monkeypatch):
def test_gemini_llm_with_agentops(monkeypatch): def test_gemini_llm_with_agentops(monkeypatch):
"""Test that Gemini LLM works correctly with agentops.""" """Test that Gemini LLM works correctly with agentops."""
from crewai.llm import LLM
import agentops
# Mock agentops # Mock agentops
mock_agentops = MagicMock() mock_agentops = MagicMock()
monkeypatch.setattr("crewai.crew.agentops", mock_agentops) monkeypatch.setattr("crewai.crew.agentops", mock_agentops)