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

fix: Fix type checking and import issues

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-02-14 21:10:53 +00:00
parent 32c1c81978
commit 2fbff69c81
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/crewai/crew.py
View File

@@ -36,7 +36,11 @@ from crewai.task import Task
from crewai.tasks.conditional_task import ConditionalTask
from crewai.tasks.task_output import TaskOutput
from crewai.telemetry import Telemetry
from crewai.tools.agent_tools.agent_tools import AgentTools
from crewai.tools.agent_tools.agent_tools import (
AgentTools,
ASK_QUESTION_TOOL,
DELEGATE_WORK_TOOL,
)
from crewai.tools.base_tool import Tool
from crewai.types.usage_metrics import UsageMetrics
from crewai.utilities import I18N, FileHandler, Logger, RPMController
@@ -965,7 +969,7 @@ class Crew(BaseModel):
# Keep track of delegation tools
delegation_tools = [
tool for tool in existing_tools
if tool.name in {"Delegate Work", "Ask Question"}
if tool.name in {DELEGATE_WORK_TOOL, ASK_QUESTION_TOOL}
]
# Create mapping of tool names to new tools
@@ -1001,7 +1005,7 @@ class Crew(BaseModel):
code_tools = agent.get_code_execution_tools()
return self._merge_tools(tools, code_tools)
def _add_delegation_tools(self, task: Task, tools: List[Tool]):
def _add_delegation_tools(self, task: Task, tools: List[Tool]) -> List[Tool]:
"""Add delegation tools for the task's agent.
Args:

1
src/crewai/tools/agent_tools/agent_tools.py
View File

@@ -1,7 +1,6 @@
from crewai.agents.agent_builder.base_agent import BaseAgent
from crewai.tools.base_tool import BaseTool
from crewai.utilities import I18N, Logger
from .ask_question_tool import AskQuestionTool
from .delegate_work_tool import DelegateWorkTool