Fix tests

crewAIInc · Jan 29, 2024 · 2845ecf · 2845ecf
1 parent 43a700a
commit 2845ecf
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,7 +18,7 @@ Repository = "https://github.com/joaomdmoura/crewai"
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
 pydantic = "^2.4.2"
-langchain = "^0.1.0"
+langchain = "0.1.0"
 openai = "^1.7.1"
 langchain-openai = "^0.0.2"
 pyright = "1.1.333"
@@ -35,6 +35,7 @@ mkdocs-material = "^9.5.3"
 profile = "black"
 known_first_party = ["crewai"]
 
+
 [tool.poetry.group.test.dependencies]
 pytest = "^7.4"
 pytest-vcr = "^1.0.2"

diff --git a/src/crewai/agents/cache/cache_handler.py b/src/crewai/agents/cache/cache_handler.py
@@ -1,7 +1,5 @@
 from typing import Optional
 
-from pydantic import PrivateAttr
-
 
 class CacheHandler:
     """Callback handler for tool usage."""

diff --git a/tests/task_test.py b/tests/task_test.py
@@ -67,7 +67,6 @@ def test_task_prompt_includes_expected_output():
         description="Give me a list of 5 interesting ideas to explore for na article, what makes them unique and interesting.",
         expected_output="Bullet point list of 5 interesting ideas.",
         agent=researcher,
-        allow_delegation=False,
     )
 
     from unittest.mock import patch