Refactor imports in test files to use new agent structure and add utility function for JSON serialization

.gitignore

@@ -4,4 +4,6 @@ agency-swarm-main/
 .cursor
 temp/
 __pycache__/
-settings.json
+settings.json
+.DS_Store
+.idea/

agency_ai_demo/{tools/clickup_tools.py → agents/ClickUpAgent/tools/ClickUpTools.py}

@@ -1,4 +1,5 @@
 import os
+import sys
 import requests
 from dotenv import load_dotenv
 from langchain.tools import StructuredTool, BaseTool, tool
@@ -9,37 +10,6 @@ from typing import Any, Type, List, Dict
 import datetime
 import json
 import re
-
-
-# Add a utility function to ensure all values are JSON serializable
-def _ensure_serializable(obj):
-    """
-    Recursively ensure that an object and all its contents are JSON serializable.
-    Handles Pydantic Field objects by extracting their values.
-    """
-    # Handle None
-    if obj is None:
-        return None
-
-    # Check if it's a Field object (has certain common Field attributes)
-    if hasattr(obj, "default") and hasattr(obj, "description"):
-        # Return the default value or None
-        if obj.default is not None:
-            return obj.default
-        return None
-
-    # Handle dictionaries
-    if isinstance(obj, dict):
-        return {k: _ensure_serializable(v) for k, v in obj.items()}
-
-    # Handle lists
-    if isinstance(obj, list):
-        return [_ensure_serializable(item) for item in obj]
-
-    # Return other objects as is
-    return obj
-
-
 from composio.tools.local.clickup.actions.base import OpenAPIAction
 from composio.tools.local.clickup.actions.create_task import (
     CreateTask,
@@ -76,7 +46,11 @@ from composio.tools.local.clickup.actions.get_task import (
     GetTaskResponse,
 )
 
+# Add the parent directory to sys.path to enable imports
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
 from utils.tool_wrapper import convert_langchain_tools
+from utils.ensure_serializable import ensure_serializable
 
 load_dotenv()
 CLICKUP_TOKEN = os.getenv("CLICKUP_TOKEN")
@@ -306,7 +280,7 @@ class CreateTaskTool(BaseTool):
         # Add any other parameters from kwargs
         for key, value in kwargs.items():
             if value is not None and key != "kwargs" and key != "list_id":
-                params[key] = _ensure_serializable(value)
+                params[key] = ensure_serializable(value)
 
         # For testing, ensure we have the minimum required parameters
         if "name" not in params:
@@ -461,7 +435,7 @@ class DeleteTaskTool(BaseTool):
 
         # Make sure all parameters are JSON serializable
         params = {
-            key: _ensure_serializable(value)
+            key: ensure_serializable(value)
             for key, value in kwargs.items()
             if value is not None and key != "kwargs" and key != "task_id"
         }
@@ -772,13 +746,11 @@ class UpdateTaskTool(BaseTool):
         # Add update parameters from kwargs
         for key, value in kwargs.items():
             if value is not None and key != "kwargs" and key != "task_id":
-                update_params[key] = _ensure_serializable(value)
+                update_params[key] = ensure_serializable(value)
 
         # Make sure all parameters are JSON serializable
         params = {
-            k: _ensure_serializable(v)
-            for k, v in update_params.items()
-            if v is not None
+            k: ensure_serializable(v) for k, v in update_params.items() if v is not None
         }
 
         print(f"Update parameters: {params}")
@@ -967,7 +939,7 @@ class AddDependencyTool(BaseTool):
 
         # Make sure all parameters are JSON serializable
         params = {
-            key: _ensure_serializable(value)
+            key: ensure_serializable(value)
             for key, value in kwargs.items()
             if value is not None and key != "kwargs" and key != "task_id"
         }
@@ -1139,7 +1111,7 @@ class GetTasksTool(BaseTool):
 
         # Make sure all parameters are JSON serializable
         query_params = {
-            k: _ensure_serializable(v)
+            k: ensure_serializable(v)
             for k, v in kwargs.items()
             if v is not None and k != "kwargs"
         }
@@ -1186,170 +1158,6 @@ class GetTasksTool(BaseTool):
         return filtered_response
 
 
-class GetTaskTool(BaseTool):
-    name: str = "get_task_tool"
-    description: str = """
-    Tool to retrieve details of a specific task from ClickUp based on its ID.
-    - Get Task:
-        Invoke: "GetTaskTool" with the appropriate parameters.
-        
-    Parameters:
-    - task_id (required): The ID of the task to retrieve
-    - custom_task_ids (optional): Whether to use custom task IDs
-    - team_id (optional): Team ID for the task
-    - include_subtasks (optional): Whether to include subtasks
-    """
-    args_schema: Type[BaseModel] = GetTaskRequest
-    headers: dict = {"Authorization": f"{CLICKUP_TOKEN}"}
-
-    def __init__(self, **data):
-        super().__init__(**data)
-
-    def _run(self, **kwargs) -> Any:
-        """Executes task retrieval from ClickUp"""
-
-        # Log the received parameters to help debug
-        print("\n==== GetTaskTool._run received parameters: ====")
-        print(f"kwargs: {kwargs}")
-
-        # Extract task_id from different possible locations
-        task_id = None
-        task_name = None
-
-        # 1. Direct task_id parameter
-        if "task_id" in kwargs:
-            task_id = kwargs.get("task_id")
-            print(f"Found task_id in direct parameter: {task_id}")
-
-        # 2. Check if task_id is inside nested kwargs
-        if "kwargs" in kwargs and isinstance(kwargs["kwargs"], dict):
-            task_id = task_id or kwargs["kwargs"].get("task_id")
-            print(f"Found task_id in nested kwargs: {task_id}")
-
-        # 3. Check if task_id is in FieldInfo format
-        if "kwargs" in kwargs and hasattr(kwargs["kwargs"], "task_id"):
-            if hasattr(kwargs["kwargs"].task_id, "default"):
-                task_id = kwargs["kwargs"].task_id.default
-                print(f"Found task_id in FieldInfo default: {task_id}")
-
-        # 4. Check for task_id in description or raw query
-        if "kwargs" in kwargs and hasattr(kwargs["kwargs"], "description"):
-            desc = kwargs["kwargs"].description
-            # Look for task ID pattern in the description
-            task_id_match = re.search(r'task_id[=:]\s*["\']?([0-9a-z]{8,})["\']?', desc)
-            if task_id_match:
-                task_id = task_id_match.group(1)
-                print(f"Found task_id in description: {task_id}")
-
-            # Look for task name in the description
-            task_name_match = re.search(r'task\s+["\']([^"\']+)["\']', desc)
-            if task_name_match:
-                task_name = task_name_match.group(1).strip()
-                print(f"Found task_name in description: {task_name}")
-
-        # 5. Check any string parameters for task_id
-        for k, v in kwargs.items():
-            if isinstance(v, str):
-                # Check if the parameter contains a task ID pattern
-                task_id_match = re.search(
-                    r'task_id[=:]\s*["\']?([0-9a-z]{8,})["\']?', v
-                )
-                if task_id_match:
-                    task_id = task_id_match.group(1)
-                    print(f"Found task_id in string parameter: {task_id}")
-                    break
-
-                # Check for task name pattern in the string
-                task_name_match = re.search(r'task\s+["\']([^"\']+)["\']', v)
-                if task_name_match:
-                    task_name = task_name_match.group(1).strip()
-                    print(f"Found task_name in string parameter: {task_name}")
-                    break
-
-        # 6. If task name found but no ID, try to lookup ID by name
-        if not task_id and task_name:
-            try:
-                # Get all tasks in the list to find the task ID by name
-                get_tasks_tool = GetTasksTool()
-                tasks = get_tasks_tool._run(list_id=901307715461)
-
-                # Find the task by name
-                for task in tasks:
-                    if task.get("name") == task_name:
-                        task_id = task.get("id")
-                        print(f"Found task_id {task_id} for task name '{task_name}'")
-                        break
-            except Exception as e:
-                print(f"Error getting task ID from name: {e}")
-
-        # 7. Hardcoded fallback for testing
-        if not task_id and task_name:
-            if task_name == "TEST TASK 2":
-                task_id = "86a702gha"  # Known ID of TEST TASK 2
-                print(f"Using hardcoded task_id for 'TEST TASK 2': {task_id}")
-            elif task_name == "TEST TASK":
-                task_id = "86a700c6e"  # Known ID of TEST TASK
-                print(f"Using hardcoded task_id for 'TEST TASK': {task_id}")
-
-        if not task_id:
-            raise ToolException("task_id is required for getting a task")
-
-        print(f"task_id being used: {task_id}")
-        print("==== End parameters ====\n")
-
-        action = GetTask()
-
-        url = f"{action.url}{action.path}".format(task_id=task_id)
-        print(f"URL being used: {url}")
-
-        # Make sure all parameters are JSON serializable
-        params = {
-            key: _ensure_serializable(value)
-            for key, value in kwargs.items()
-            if value is not None and key != "kwargs" and key != "task_id"
-        }
-
-        response = requests.get(url, headers=self.headers, params=params)
-        print(f"Response status code: {response.status_code}")
-
-        if response.status_code == 200:
-            response_json = response.json()
-        else:
-            try:
-                response_json = response.json()
-                print(f"Error response: {response_json}")
-            except requests.JSONDecodeError:
-                response_json = {"error": "Invalid JSON response"}
-                print("Could not decode JSON response")
-
-        task_details = GetTaskResponse(data=response_json)
-
-        # Format the response for better readability
-        if task_details.data and isinstance(task_details.data, dict):
-            task = task_details.data
-            formatted_response = {
-                "name": task.get("name", "N/A"),
-                "id": task.get("id", "N/A"),
-                "status": task.get("status", {}).get("status", "N/A"),
-                "assignees": [
-                    a.get("username", "N/A") for a in task.get("assignees", [])
-                ],
-                "description": task.get("description", "N/A"),
-                "due_date": task.get("due_date", "N/A"),
-                "time_estimate": task.get("time_estimate", "N/A"),
-            }
-
-            print(f"Found task: {formatted_response}")
-            return formatted_response
-        else:
-            error_message = "Task not found or API error occurred"
-            if "err" in task_details.data:
-                error_message = f"Error: {task_details.data['err']}"
-
-            print(f"Result: {error_message}")
-            return error_message
-
-
 # Util for converting dates
 @tool
 def date_to_timestamp(date_str: str) -> int:

agency_ai_demo/agents/ClickUpAgent/tools/ExampleTool.py

@@ -0,0 +1,30 @@
+from agency_swarm.tools import BaseTool
+from pydantic import Field
+import os
+
+account_id = "MY_ACCOUNT_ID"
+api_key = os.getenv("MY_API_KEY") # or access_token = os.getenv("MY_ACCESS_TOKEN")
+
+class ExampleTool(BaseTool):
+    """
+    A brief description of what the custom tool does.
+    The docstring should clearly explain the tool's purpose and functionality.
+    It will be used by the agent to determine when to use this tool.
+    """
+
+    # Define the fields with descriptions using Pydantic Field
+    example_field: str = Field(
+        ..., description="Description of the example field, explaining its purpose and usage for the Agent."
+    )
+
+    def run(self):
+        """
+        The implementation of the run method, where the tool's main functionality is executed.
+        This method should utilize the fields defined above to perform the task.
+        Docstring is not required for this method and will not be used by the agent.
+        """
+        # Your custom tool logic goes here
+        # do_something(self.example_field, api_key, account_id)
+
+        # Return the result of the tool's operation as a string
+        return "Result of ExampleTool operation"

agency_ai_demo/agents/ClickUpAgent/tools/GetTaskTool.py

@@ -0,0 +1,78 @@
+import os
+import sys
+import requests
+from dotenv import load_dotenv
+from langchain.tools import BaseTool
+from langchain_core.pydantic_v1 import BaseModel
+from langchain_core.tools import ToolException
+from typing import Any, Type
+import re
+from composio.tools.local.clickup.actions.get_task import (
+    GetTask,
+    GetTaskRequest,
+    GetTaskResponse,
+)
+
+# Add the parent directory to sys.path to enable imports
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+# Add project utils
+from utils.tool_wrapper import convert_langchain_tools
+from utils.ensure_serializable import ensure_serializable
+
+load_dotenv()
+
+
+CLICKUP_TOKEN = os.getenv("CLICKUP_TOKEN")
+
+
+# TODO: FIX THIS TOOL
+class GetTaskTool(BaseTool):
+    name: str = "get_task_tool"
+    description: str = """
+    Ferramenta para visualizar detalhes de uma tarefa no ClickUp.
+    - Obter detalhes da tarefa:
+        Invocar: "GetTaskTool" com o ID da tarefa e parâmetros opcionais.
+    """
+    args_schema: Type[BaseModel] = GetTaskRequest
+    headers: dict = {"Authorization": f"{CLICKUP_TOKEN}"}
+
+    def _init_(self, **data):
+        super()._init_(**data)
+
+    def _run(self, **kwargs) -> Any:
+        """Executa a requisição para obter detalhes de uma tarefa no ClickUp"""
+
+        action = GetTask()
+
+        url = f"{action.url}{action.path}".format(task_id=kwargs.get("task_id"))
+
+        query_params = {k: v for k, v in kwargs.items() if v is not None}
+
+        response = requests.get(url, headers=self.headers, params=query_params)
+
+        if response.status_code == 200:
+            response_json = response.json()
+        else:
+            try:
+                response_json = response.json()
+            except requests.JSONDecodeError:
+                response_json = {"error": "Invalid JSON response"}
+
+        response = GetTaskResponse(data=response_json)
+
+        filtered_response = {
+            "id": response.data.get("id"),
+            "name": response.data.get("name"),
+            "status": response.data.get("status", {}).get("status"),
+            "due_date": response.data.get("due_date"),
+            "date_created": response.data.get("date_created"),
+            "url": response.data.get("url"),
+            "assignees": [
+                {"id": assignee.get("id"), "username": assignee.get("username")}
+                for assignee in response.data.get("assignees", [])
+            ],
+            "error": response.data.get("error"),
+        }
+
+        return filtered_response

agency_ai_demo/agents/ResearchAndReportAgent/tools/ExampleTool.py

@@ -0,0 +1,30 @@
+from agency_swarm.tools import BaseTool
+from pydantic import Field
+import os
+
+account_id = "MY_ACCOUNT_ID"
+api_key = os.getenv("MY_API_KEY") # or access_token = os.getenv("MY_ACCESS_TOKEN")
+
+class ExampleTool(BaseTool):
+    """
+    A brief description of what the custom tool does.
+    The docstring should clearly explain the tool's purpose and functionality.
+    It will be used by the agent to determine when to use this tool.
+    """
+
+    # Define the fields with descriptions using Pydantic Field
+    example_field: str = Field(
+        ..., description="Description of the example field, explaining its purpose and usage for the Agent."
+    )
+
+    def run(self):
+        """
+        The implementation of the run method, where the tool's main functionality is executed.
+        This method should utilize the fields defined above to perform the task.
+        Docstring is not required for this method and will not be used by the agent.
+        """
+        # Your custom tool logic goes here
+        # do_something(self.example_field, api_key, account_id)
+
+        # Return the result of the tool's operation as a string
+        return "Result of ExampleTool operation"

agency_ai_demo/agents/TechnicalProjectManager/tools/ExampleTool.py

@@ -0,0 +1,30 @@
+from agency_swarm.tools import BaseTool
+from pydantic import Field
+import os
+
+account_id = "MY_ACCOUNT_ID"
+api_key = os.getenv("MY_API_KEY") # or access_token = os.getenv("MY_ACCESS_TOKEN")
+
+class ExampleTool(BaseTool):
+    """
+    A brief description of what the custom tool does.
+    The docstring should clearly explain the tool's purpose and functionality.
+    It will be used by the agent to determine when to use this tool.
+    """
+
+    # Define the fields with descriptions using Pydantic Field
+    example_field: str = Field(
+        ..., description="Description of the example field, explaining its purpose and usage for the Agent."
+    )
+
+    def run(self):
+        """
+        The implementation of the run method, where the tool's main functionality is executed.
+        This method should utilize the fields defined above to perform the task.
+        Docstring is not required for this method and will not be used by the agent.
+        """
+        # Your custom tool logic goes here
+        # do_something(self.example_field, api_key, account_id)
+
+        # Return the result of the tool's operation as a string
+        return "Result of ExampleTool operation"

agency_ai_demo/demo.ipynb

@@ -0,0 +1,533 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "True"
+      ]
+     },
+     "execution_count": 1,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "import os\n",
+    "import requests\n",
+    "from dotenv import load_dotenv\n",
+    "# from langchain.tools import StructuredTool, BaseTool, tool\n",
+    "from langchain_core.pydantic_v1 import BaseModel, Field\n",
+    "from langchain_core.tools import ToolException\n",
+    "from pydantic import ValidationError\n",
+    "from typing import Any, Type, List, Dict\n",
+    "import datetime\n",
+    "\n",
+    "from openai import AzureOpenAI\n",
+    "from agency_swarm import set_openai_client\n",
+    "from agency_swarm import Agent, Agency\n",
+    "\n",
+    "\n",
+    "from composio.tools.local.clickup.actions.base import OpenAPIAction\n",
+    "from composio.tools.local.clickup.actions.create_task import CreateTask, CreateTaskResponse\n",
+    "\n",
+    "from agents.ClickUpAgent.tools.ClickUpTools import initialize_clickup_tools\n",
+    "\n",
+    "\n",
+    "load_dotenv()"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "client = AzureOpenAI(\n",
+    "    api_key=os.getenv(\"AZURE_OPENAI_API_KEY\"),\n",
+    "    api_version=os.getenv(\"OPENAI_API_VERSION\"),\n",
+    "    azure_endpoint=os.getenv(\"AZURE_OPENAI_ENDPOINT\"),\n",
+    "    timeout=5,\n",
+    "    max_retries=5,\n",
+    ")\n",
+    "\n",
+    "set_openai_client(client)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 3,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "project_manager = Agent(\n",
+    "    name=\"project_manager\",\n",
+    "    description=\"I am the Project Manager\",\n",
+    "    model=\"gpt-4o\",\n",
+    "    )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "clickup_agent_instructions = f\"\"\"# PROJECT_MANAGEMENT\n",
+    "Your team uses ClickUp to manage projects and tasks.\n",
+    "Users often refer to lists and tasks by name, and sometimes there may not be an exact match, so you should look for the closest ones.\n",
+    "If in doubt, provide the user with valid options to choose from or ask for more information if necessary.\n",
+    "\n",
+    "## CLICKUP_STRUCTURE\n",
+    "-> Workspace: The highest level of organization in ClickUp. Contains all your Spaces.\n",
+    "--> Space: A collection of Folders and Lists. It is a way to group related work.\n",
+    "---> Folder: Used to group Lists.\n",
+    "----> List: Used to group tasks. Lists can be in a Folder or directly in a Space.\n",
+    "-----> Task: The basic unit of work in ClickUp, assignable with due dates.\n",
+    "------> Subtask: A child task of a parent Task, assignable to different people.\n",
+    "\n",
+    "## DEFAULT_CLICKUP_IDS\n",
+    "Use these IDs unless specified otherwise\n",
+    "- Team ID: {os.getenv(\"CLICKUP_TEAM_ID\")}\n",
+    "- List ID 'test_clickup_tool': {os.getenv(\"CLICKUP_LIST_ID\")}\n",
+    "\n",
+    "## CLICKUP_USERS\n",
+    "- Your id: {os.getenv(\"CLICKUP_AGENT_USER_ID\")}\n",
+    "- Main user id: {os.getenv(\"CLICKUP_USER_ID\")}\n",
+    "\"\"\""
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "[<class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>, <class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>, <class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>, <class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>, <class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>, <class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>, <class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>, <class 'utils.tool_wrapper._convert_function_tool.<locals>.ConvertedFunctionTool'>]\n"
+     ]
+    }
+   ],
+   "source": [
+    "# clickup_tools = initialize_clickup_tools()\n",
+    "# print(clickup_tools)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from agents.ClickUpAgent.tools.GetTaskTool import GetTaskTool\n",
+    "from utils.tool_wrapper import convert_langchain_tools\n",
+    "get_task_tool = GetTaskTool()\n",
+    "clickup_tools = convert_langchain_tools([get_task_tool])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 6,
+   "metadata": {},
+   "outputs": [
+    {
+     "data": {
+      "text/plain": [
+       "'Create_task_toolTool'"
+      ]
+     },
+     "execution_count": 6,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "clickup_tools[0].__name__"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 7,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "\n",
+    "clickup_agent = Agent(\n",
+    "    name=\"clickup_agent\",\n",
+    "    description=\"I am a simple agent\",\n",
+    "    instructions=clickup_agent_instructions,\n",
+    "    model=\"gpt-4o\",\n",
+    "    tools=clickup_tools,\n",
+    "    )"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 8,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Updating agent... clickup_agent\n"
+     ]
+    }
+   ],
+   "source": [
+    "agency = Agency(\n",
+    "    agency_chart=[\n",
+    "    project_manager,\n",
+    "    [project_manager, clickup_agent],\n",
+    "],\n",
+    "    # shared_instructions=\"agency_manifesto.md\"\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 9,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/Users/morossini/Projects/professional_demos/agency_swarm_demo/.venv/lib/python3.11/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+      "  from .autonotebook import tqdm as notebook_tqdm\n",
+      "/Users/morossini/Projects/professional_demos/agency_swarm_demo/.venv/lib/python3.11/site-packages/gradio/components/chatbot.py:285: UserWarning: You have not specified a value for the `type` parameter. Defaulting to the 'tuples' format for chatbot messages, but this is deprecated and will be removed in a future version of Gradio. Please set type='messages' instead, which uses openai-style dictionaries with 'role' and 'content' keys.\n",
+      "  warnings.warn(\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "* Running on local URL:  http://127.0.0.1:7860\n",
+      "\n",
+      "To create a public link, set `share=True` in `launch()`.\n"
+     ]
+    },
+    {
+     "data": {
+      "text/html": [
+       "<div><iframe src=\"http://127.0.0.1:7860/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
+      ],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/plain": [
+       "Gradio Blocks instance: 6 backend functions\n",
+       "-------------------------------------------\n",
+       "fn_index=0\n",
+       " inputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       " outputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       "fn_index=1\n",
+       " inputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       " |-<gradio.components.dropdown.Dropdown object at 0x123608550>\n",
+       " outputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       " |-<gradio.components.dropdown.Dropdown object at 0x123608550>\n",
+       "fn_index=2\n",
+       " inputs:\n",
+       " |-<gradio.components.dropdown.Dropdown object at 0x123608550>\n",
+       " outputs:\n",
+       "fn_index=3\n",
+       " inputs:\n",
+       " |-<gradio.templates.Files object at 0x122a95b50>\n",
+       " outputs:\n",
+       "fn_index=4\n",
+       " inputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       " outputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       "fn_index=5\n",
+       " inputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       " |-<gradio.components.dropdown.Dropdown object at 0x123608550>\n",
+       " outputs:\n",
+       " |-<gradio.components.textbox.Textbox object at 0x11f7e8c50>\n",
+       " |-<gradio.components.chatbot.Chatbot object at 0x11f2cd950>\n",
+       " |-<gradio.components.dropdown.Dropdown object at 0x123608550>"
+      ]
+     },
+     "execution_count": 9,
+     "metadata": {},
+     "output_type": "execute_result"
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Message files:  []\n",
+      "Images:  []\n",
+      "THREAD:[ user -> project_manager ]: URL https://platform.openai.com/playground/assistants?assistant=asst_q3bAPsDDQmwfijHqK7v8Z655&mode=assistant&thread=thread_awYwQAYJsii7m4eCHOOtpQtX\n",
+      "THREAD:[ project_manager -> clickup_agent ]: URL https://platform.openai.com/playground/assistants?assistant=asst_NQWCwFwP37B8T5rATBm6Fi6E&mode=assistant&thread=thread_tT4NsmS9eZuM0FmYLzwfwHMl\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "[2025-02-27 18:56:41,742][ERROR] Tool execution failed: list_id is required for getting list information\n",
+      "Traceback (most recent call last):\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/utils/tool_wrapper.py\", line 409, in run\n",
+      "    result = tool_func._run(**kwargs)\n",
+      "             ^^^^^^^^^^^^^^^^^^^^^^^^\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/agents/ClickUpAgent/tools/ClickUpTools.py\", line 1055, in _run\n",
+      "    raise ToolException(\"list_id is required for getting list information\")\n",
+      "langchain_core.tools.base.ToolException: list_id is required for getting list information\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "==== GetListTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n",
+      "\n",
+      "==== GetTasksTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n",
+      "No list_id found in parameters, using hardcoded value 901307715461\n",
+      "list_id being used: 901307715461\n",
+      "==== End parameters ====\n",
+      "\n",
+      "URL being used: https://api.clickup.com/api/v2/list/901307715461/task\n",
+      "Response status code: 200\n",
+      "Returning filtered response with 1 items\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "[2025-02-27 18:56:46,276][ERROR] Tool execution failed: task_id is required for getting a task\n",
+      "Traceback (most recent call last):\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/utils/tool_wrapper.py\", line 409, in run\n",
+      "    result = tool_func._run(**kwargs)\n",
+      "             ^^^^^^^^^^^^^^^^^^^^^^^^\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/agents/ClickUpAgent/tools/ClickUpTools.py\", line 1299, in _run\n",
+      "    raise ToolException(\"task_id is required for getting a task\")\n",
+      "langchain_core.tools.base.ToolException: task_id is required for getting a task\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "==== GetTaskTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "[2025-02-27 18:56:47,808][ERROR] Tool execution failed: task_id is required for getting a task\n",
+      "Traceback (most recent call last):\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/utils/tool_wrapper.py\", line 409, in run\n",
+      "    result = tool_func._run(**kwargs)\n",
+      "             ^^^^^^^^^^^^^^^^^^^^^^^^\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/agents/ClickUpAgent/tools/ClickUpTools.py\", line 1299, in _run\n",
+      "    raise ToolException(\"task_id is required for getting a task\")\n",
+      "langchain_core.tools.base.ToolException: task_id is required for getting a task\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "==== GetTaskTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n",
+      "Message files:  []\n",
+      "Images:  []\n",
+      "THREAD:[ user -> project_manager ]: URL https://platform.openai.com/playground/assistants?assistant=asst_q3bAPsDDQmwfijHqK7v8Z655&mode=assistant&thread=thread_awYwQAYJsii7m4eCHOOtpQtX\n",
+      "THREAD:[ project_manager -> clickup_agent ]: URL https://platform.openai.com/playground/assistants?assistant=asst_NQWCwFwP37B8T5rATBm6Fi6E&mode=assistant&thread=thread_tT4NsmS9eZuM0FmYLzwfwHMl\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "[2025-02-27 18:58:01,081][ERROR] Tool execution failed: task_id is required for getting a task\n",
+      "Traceback (most recent call last):\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/utils/tool_wrapper.py\", line 409, in run\n",
+      "    result = tool_func._run(**kwargs)\n",
+      "             ^^^^^^^^^^^^^^^^^^^^^^^^\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/agents/ClickUpAgent/tools/ClickUpTools.py\", line 1299, in _run\n",
+      "    raise ToolException(\"task_id is required for getting a task\")\n",
+      "langchain_core.tools.base.ToolException: task_id is required for getting a task\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "==== GetTaskTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "[2025-02-27 18:58:04,314][ERROR] Tool execution failed: task_id is required for getting a task\n",
+      "Traceback (most recent call last):\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/utils/tool_wrapper.py\", line 409, in run\n",
+      "    result = tool_func._run(**kwargs)\n",
+      "             ^^^^^^^^^^^^^^^^^^^^^^^^\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/agents/ClickUpAgent/tools/ClickUpTools.py\", line 1299, in _run\n",
+      "    raise ToolException(\"task_id is required for getting a task\")\n",
+      "langchain_core.tools.base.ToolException: task_id is required for getting a task\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "==== GetTaskTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "[2025-02-27 18:58:08,100][ERROR] Tool execution failed: task_id is required for getting a task\n",
+      "Traceback (most recent call last):\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/utils/tool_wrapper.py\", line 409, in run\n",
+      "    result = tool_func._run(**kwargs)\n",
+      "             ^^^^^^^^^^^^^^^^^^^^^^^^\n",
+      "  File \"/Users/morossini/Projects/professional_demos/agency_swarm_demo/agency_ai_demo/agents/ClickUpAgent/tools/ClickUpTools.py\", line 1299, in _run\n",
+      "    raise ToolException(\"task_id is required for getting a task\")\n",
+      "langchain_core.tools.base.ToolException: task_id is required for getting a task\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "==== GetTaskTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n",
+      "Message files:  []\n",
+      "Images:  []\n",
+      "THREAD:[ user -> project_manager ]: URL https://platform.openai.com/playground/assistants?assistant=asst_q3bAPsDDQmwfijHqK7v8Z655&mode=assistant&thread=thread_awYwQAYJsii7m4eCHOOtpQtX\n",
+      "THREAD:[ project_manager -> clickup_agent ]: URL https://platform.openai.com/playground/assistants?assistant=asst_NQWCwFwP37B8T5rATBm6Fi6E&mode=assistant&thread=thread_tT4NsmS9eZuM0FmYLzwfwHMl\n",
+      "\n",
+      "==== CreateTaskTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n",
+      "No list_id found in parameters, using hardcoded value 901307715461\n",
+      "list_id being used: 901307715461\n",
+      "==== End parameters ====\n",
+      "\n",
+      "URL being used: https://api.clickup.com/api/v2/list/901307715461/task\n",
+      "Request parameters: {'name': 'API TEST TASK', 'assignees': [81918955]}\n",
+      "Response status code: 200\n",
+      "Error response: {'id': '86a706rnx', 'custom_id': None, 'custom_item_id': 0, 'name': 'API TEST TASK', 'text_content': '', 'description': '', 'status': {'id': 'c90131663060_Uk5SstGY', 'status': 'backlog', 'color': '#9c09bc', 'orderindex': 0, 'type': 'open'}, 'orderindex': '120305513.00000000000000000000000000000000', 'date_created': '1740693545341', 'date_updated': '1740693545341', 'date_closed': None, 'date_done': None, 'archived': False, 'creator': {'id': 84141406, 'username': 'Ana Beatriz', 'color': '#455963', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/84141406_ZkZ.jpg'}, 'assignees': [{'id': 81918955, 'username': 'João Guilherme Silva Morossini', 'color': '#81b1ff', 'initials': 'JM', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/81918955_kcK.jpg'}], 'group_assignees': [], 'watchers': [{'id': 81918955, 'username': 'João Guilherme Silva Morossini', 'color': '#81b1ff', 'initials': 'JM', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/81918955_kcK.jpg'}, {'id': 84141406, 'username': 'Ana Beatriz', 'color': '#455963', 'initials': 'AB', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/84141406_ZkZ.jpg'}], 'checklists': [], 'tags': [], 'parent': None, 'top_level_parent': None, 'priority': None, 'due_date': None, 'start_date': None, 'points': None, 'time_estimate': None, 'time_spent': 0, 'custom_fields': [], 'dependencies': [], 'linked_tasks': [], 'locations': [], 'team_id': '12927880', 'url': 'https://app.clickup.com/t/86a706rnx', 'sharing': {'public': False, 'public_share_expires_on': None, 'public_fields': ['assignees', 'priority', 'due_date', 'content', 'comments', 'attachments', 'customFields', 'subtasks', 'tags', 'checklists', 'coverimage'], 'token': None, 'seo_optimized': False}, 'permission_level': 'create', 'list': {'id': '901307715461', 'name': 'test_clickup_tool', 'access': True}, 'project': {'id': '90131663060', 'name': 'Agentes', 'hidden': False, 'access': True}, 'folder': {'id': '90131663060', 'name': 'Agentes', 'hidden': False, 'access': True}, 'space': {'id': '54804921'}}\n",
+      "Returning filtered response: {\n",
+      "  \"id\": \"86a706rnx\",\n",
+      "  \"name\": \"API TEST TASK\",\n",
+      "  \"status\": \"backlog\",\n",
+      "  \"assignees\": [\n",
+      "    {\n",
+      "      \"id\": 81918955,\n",
+      "      \"username\": \"Jo\\u00e3o Guilherme Silva Morossini\",\n",
+      "      \"color\": \"#81b1ff\",\n",
+      "      \"initials\": \"JM\",\n",
+      "      \"email\": \"[email protected]\",\n",
+      "      \"profilePicture\": \"https://attachments.clickup.com/profilePictures/81918955_kcK.jpg\"\n",
+      "    }\n",
+      "  ],\n",
+      "  \"due_date\": null,\n",
+      "  \"error\": null\n",
+      "}\n",
+      "Message files:  []\n",
+      "Images:  []\n",
+      "THREAD:[ user -> project_manager ]: URL https://platform.openai.com/playground/assistants?assistant=asst_q3bAPsDDQmwfijHqK7v8Z655&mode=assistant&thread=thread_awYwQAYJsii7m4eCHOOtpQtX\n",
+      "THREAD:[ project_manager -> clickup_agent ]: URL https://platform.openai.com/playground/assistants?assistant=asst_NQWCwFwP37B8T5rATBm6Fi6E&mode=assistant&thread=thread_tT4NsmS9eZuM0FmYLzwfwHMl\n",
+      "\n",
+      "==== CreateTaskTool._run received parameters: ====\n",
+      "kwargs: {'kwargs': FieldInfo(annotation=NoneType, required=False, default=None, description='Parameter: kwargs')}\n",
+      "No list_id found in parameters, using hardcoded value 901307715461\n",
+      "list_id being used: 901307715461\n",
+      "==== End parameters ====\n",
+      "\n",
+      "URL being used: https://api.clickup.com/api/v2/list/901307715461/task\n",
+      "Request parameters: {'name': 'API TEST TASK', 'assignees': [81918955]}\n",
+      "Response status code: 200\n",
+      "Error response: {'id': '86a706rxv', 'custom_id': None, 'custom_item_id': 0, 'name': 'API TEST TASK', 'text_content': '', 'description': '', 'status': {'id': 'c90131663060_Uk5SstGY', 'status': 'backlog', 'color': '#9c09bc', 'orderindex': 0, 'type': 'open'}, 'orderindex': '120305514.00000000000000000000000000000000', 'date_created': '1740693581177', 'date_updated': '1740693581177', 'date_closed': None, 'date_done': None, 'archived': False, 'creator': {'id': 84141406, 'username': 'Ana Beatriz', 'color': '#455963', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/84141406_ZkZ.jpg'}, 'assignees': [{'id': 81918955, 'username': 'João Guilherme Silva Morossini', 'color': '#81b1ff', 'initials': 'JM', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/81918955_kcK.jpg'}], 'group_assignees': [], 'watchers': [{'id': 81918955, 'username': 'João Guilherme Silva Morossini', 'color': '#81b1ff', 'initials': 'JM', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/81918955_kcK.jpg'}, {'id': 84141406, 'username': 'Ana Beatriz', 'color': '#455963', 'initials': 'AB', 'email': '[email protected]', 'profilePicture': 'https://attachments.clickup.com/profilePictures/84141406_ZkZ.jpg'}], 'checklists': [], 'tags': [], 'parent': None, 'top_level_parent': None, 'priority': None, 'due_date': None, 'start_date': None, 'points': None, 'time_estimate': None, 'time_spent': 0, 'custom_fields': [], 'dependencies': [], 'linked_tasks': [], 'locations': [], 'team_id': '12927880', 'url': 'https://app.clickup.com/t/86a706rxv', 'sharing': {'public': False, 'public_share_expires_on': None, 'public_fields': ['assignees', 'priority', 'due_date', 'content', 'comments', 'attachments', 'customFields', 'subtasks', 'tags', 'checklists', 'coverimage'], 'token': None, 'seo_optimized': False}, 'permission_level': 'create', 'list': {'id': '901307715461', 'name': 'test_clickup_tool', 'access': True}, 'project': {'id': '90131663060', 'name': 'Agentes', 'hidden': False, 'access': True}, 'folder': {'id': '90131663060', 'name': 'Agentes', 'hidden': False, 'access': True}, 'space': {'id': '54804921'}}\n",
+      "Returning filtered response: {\n",
+      "  \"id\": \"86a706rxv\",\n",
+      "  \"name\": \"API TEST TASK\",\n",
+      "  \"status\": \"backlog\",\n",
+      "  \"assignees\": [\n",
+      "    {\n",
+      "      \"id\": 81918955,\n",
+      "      \"username\": \"Jo\\u00e3o Guilherme Silva Morossini\",\n",
+      "      \"color\": \"#81b1ff\",\n",
+      "      \"initials\": \"JM\",\n",
+      "      \"email\": \"[email protected]\",\n",
+      "      \"profilePicture\": \"https://attachments.clickup.com/profilePictures/81918955_kcK.jpg\"\n",
+      "    }\n",
+      "  ],\n",
+      "  \"due_date\": null,\n",
+      "  \"error\": null\n",
+      "}\n"
+     ]
+    }
+   ],
+   "source": [
+    "agency.demo_gradio()"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.11.3"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

agency_ai_demo/tests/test_clickup_agent.py

@@ -5,7 +5,7 @@ from dotenv import load_dotenv
 from openai import AzureOpenAI
 from agency_swarm import set_openai_client, Agent, Agency
 
-from tools.clickup_tools import initialize_clickup_tools
+from agents.ClickUpAgent.tools.ClickUpTools import initialize_clickup_tools
 
 load_dotenv()
 

agency_ai_demo/tests/test_clickup_tool_conversion.py

@@ -15,7 +15,7 @@ sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 from utils.tool_wrapper import convert_langchain_tool, LangChainBaseTool
 
 # Import all tools from the clickup_tools module
-from tools.clickup_tools import (
+from agents.ClickUpAgent.tools.ClickUpTools import (
     CreateTaskTool,
     DeleteTaskTool,
     UpdateTaskTool,

agency_ai_demo/tests/test_clickup_tools.py

@@ -6,7 +6,7 @@ import json
 # Add the parent directory to sys.path to import modules
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
-from agency_ai_demo.tools.clickup_tools import (
+from agents.ClickUpAgent.tools.ClickUpTools import (
     CreateTaskTool,
     DeleteTaskTool,
     UpdateTaskTool,

agency_ai_demo/tests/test_list_tasks.py

@@ -4,7 +4,7 @@ from openai import AzureOpenAI
 from agency_swarm import set_openai_client
 from agency_swarm import Agent
 
-from tools.clickup_tools import initialize_clickup_tools
+from agents.ClickUpAgent.tools.ClickUpTools import initialize_clickup_tools
 
 load_dotenv()
 
@@ -34,7 +34,7 @@ def main():
 
     print("\nTesting get_tasks_tool...")
     # Test GetTasksTool directly without using the agent
-    from tools.clickup_tools import GetTasksTool
+    from agents.ClickUpAgent.tools.ClickUpTools import GetTasksTool
 
     tool = GetTasksTool()
     result = tool._run(list_id=901307715461)

agency_ai_demo/tests/test_tool_directly.py

@@ -6,7 +6,11 @@ from dotenv import load_dotenv
 from openai import AzureOpenAI
 from agency_swarm import set_openai_client, Agent, Agency
 
-from tools.clickup_tools import initialize_clickup_tools, GetTasksTool, GetTaskTool
+from agents.ClickUpAgent.tools.ClickUpTools import (
+    initialize_clickup_tools,
+    GetTasksTool,
+    GetTaskTool,
+)
 
 load_dotenv()
 

agency_ai_demo/utils/ensure_serializable.py

@@ -0,0 +1,30 @@
+import json
+
+
+# Add a utility function to ensure all values are JSON serializable
+def ensure_serializable(obj):
+    """
+    Recursively ensure that an object and all its contents are JSON serializable.
+    Handles Pydantic Field objects by extracting their values.
+    """
+    # Handle None
+    if obj is None:
+        return None
+
+    # Check if it's a Field object (has certain common Field attributes)
+    if hasattr(obj, "default") and hasattr(obj, "description"):
+        # Return the default value or None
+        if obj.default is not None:
+            return obj.default
+        return None
+
+    # Handle dictionaries
+    if isinstance(obj, dict):
+        return {k: ensure_serializable(v) for k, v in obj.items()}
+
+    # Handle lists
+    if isinstance(obj, list):
+        return [ensure_serializable(item) for item in obj]
+
+    # Return other objects as is
+    return obj