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

Fix type-checking issues in to_structured_dict method

Browse Source 
Co-Authored-By: Joe Moura <joao@crewai.com>
This commit is contained in:
Devin AI
2025-03-11 07:50:34 +00:00
parent 0b3a5a6c81
commit 39834680e8
1 changed files with 14 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/crewai/crew.py
View File

@@ -1263,19 +1263,22 @@ class Crew(BaseModel):
Returns: Returns:
Dict[str, Any]: Dictionary representation of the task Dict[str, Any]: Dictionary representation of the task
""" """
task_info = { task_info: Dict[str, Any] = {
"id": str(task.id), "id": str(task.id),
"description": task.description, "description": task.description,
"expected_output": task.expected_output "expected_output": task.expected_output
} }
# Safely add agent information if available # Safely add agent information if available
if getattr(task, 'agent', None): agent = getattr(task, 'agent', None)
task_info["agent"] = task.agent.role if agent is not None:
task_info["agent"] = agent.role
# Safely add async_execution if available # Safely add async_execution if available
if hasattr(task, 'async_execution'): if hasattr(task, 'async_execution'):
task_info["async_execution"] = task.async_execution async_execution = getattr(task, 'async_execution', None)
if async_execution is not None:
task_info["async_execution"] = async_execution
return task_info return task_info
@@ -1291,7 +1294,7 @@ class Crew(BaseModel):
agents, tasks, and their relationships. agents, tasks, and their relationships.
""" """
# Basic crew information # Basic crew information
result = { result: Dict[str, Any] = {
"id": str(self.id), "id": str(self.id),
"name": self.name, "name": self.name,
"process": str(self.process), "process": str(self.process),
@@ -1313,18 +1316,20 @@ class Crew(BaseModel):
result["tasks"].append(task_info) result["tasks"].append(task_info)
# Add task relationships if context exists # Add task relationships if context exists
if getattr(task, 'context', None): context = getattr(task, 'context', None)
if context is not None:
relationship = { relationship = {
"task_id": str(task.id), "task_id": str(task.id),
"depends_on": [str(context_task.id) for context_task in task.context] "depends_on": [str(context_task.id) for context_task in context]
} }
result["task_relationships"].append(relationship) result["task_relationships"].append(relationship)
except Exception as e: except Exception as e:
self._logger.log("warning", f"Error processing task {getattr(task, 'id', 'unknown')}: {str(e)}") self._logger.log("warning", f"Error processing task {getattr(task, 'id', 'unknown')}: {str(e)}")
# Add manager agent if exists # Add manager agent if exists
if getattr(self, 'manager_agent', None): manager_agent = getattr(self, 'manager_agent', None)
result["manager_agent"] = self._agent_to_dict(self.manager_agent) if manager_agent is not None:
result["manager_agent"] = self._agent_to_dict(manager_agent)
return result return result