From 0918b6fd1fd53b2542facafb8d25fb18c6b08c69 Mon Sep 17 00:00:00 2001 From: Vardaan Grover Date: Fri, 24 Jan 2025 16:13:49 +0530 Subject: [PATCH] feat: add field validator to flatten context in DelegateWorkToolSchema --- src/crewai/tools/agent_tools/delegate_work_tool.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/crewai/tools/agent_tools/delegate_work_tool.py b/src/crewai/tools/agent_tools/delegate_work_tool.py index 9dbf6c920..c48f06661 100644 --- a/src/crewai/tools/agent_tools/delegate_work_tool.py +++ b/src/crewai/tools/agent_tools/delegate_work_tool.py @@ -1,6 +1,6 @@ from typing import Optional -from pydantic import BaseModel, Field +from pydantic import BaseModel, Field, field_validator from crewai.tools.agent_tools.base_agent_tools import BaseAgentTool @@ -12,6 +12,9 @@ class DelegateWorkToolSchema(BaseModel): ..., description="The role/name of the coworker to delegate to" ) + @field_validator("context") + def flatten(cls, v): + return " ".join(v) if isinstance(v, list) else v class DelegateWorkTool(BaseAgentTool): """Tool for delegating work to coworkers"""