Add tests for ClickUp tools and agent functionality

agency_ai_demo/tests/test_clickup_agent.py

@@ -0,0 +1,94 @@
+import os
+import sys
+import traceback
+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
+
+load_dotenv()
+
+
+def main():
+    try:
+        # Accept query from command line arguments if provided
+        if len(sys.argv) > 1:
+            query = sys.argv[1]
+        else:
+            # Default query if no argument provided
+            query = 'Use the create_task_tool to create a new task in list_id 901307715461 with name "API TEST TASK" and assignees [81918955]. Make sure to pass the list_id parameter correctly.'
+
+        print("Initializing OpenAI client...")
+        client = AzureOpenAI(
+            api_key=os.getenv("AZURE_OPENAI_API_KEY"),
+            api_version=os.getenv("OPENAI_API_VERSION"),
+            azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
+            timeout=10,
+            max_retries=5,
+        )
+
+        set_openai_client(client)
+
+        print("Converting ClickUp tools...")
+        clickup_tools = initialize_clickup_tools()
+
+        # Instructions similar to what's in the notebook
+        clickup_instructions = """
+        1. GERENCIAMENTO_DE_PROJETOS:
+            A equipe usa o ClickUp para gerenciar projetos e tarefas.
+            
+        1.1 ESTRUTURA_DO_CLICKUP
+            -> Workspace: O nível mais alto de organização no ClickUp. Contém todos os seus Espaços.
+            --> Space: Uma coleção de Pastas e Listas. É uma maneira de agrupar trabalhos relacionados.
+            ---> Folder: Usado para agrupar Listas.
+            ----> List: Usado para agrupar tarefas. As listas podem estar em uma Pasta ou diretamente em um Espaço.
+            -----> Task: A unidade básica de trabalho no ClickUp, atribuível com datas de vencimento.
+            ------> Subtask: Uma tarefa filha de uma Tarefa pai, atribuível a diferentes pessoas.
+
+        1.1.1 IDS_PADRÃO_DO_CLICKUP
+            Use esses IDs, a menos que especificado de outra forma
+            - 'Workspace' (também conhecido como "team_id"): 12927880
+            - ID do espaço 'Projetos': 54804921
+            - ID da pasta 'Agentes': 90131663060
+            - ID da lista 'test_clickup_tool': 901307715461
+            
+        1.1.2 USUÁRIOS_ATIVOS_DO_CLICKUP (id, nome de usuário, email)
+            - 81918955, João Guilherme Silva Morossini, [email protected]
+        """
+
+        print("Creating ClickUp agent...")
+        clickup_agent = Agent(
+            name="clickup_agent",
+            description="I am a ClickUp agent that helps manage tasks and projects",
+            instructions=clickup_instructions,
+            model="gpt-4o",
+            tools=clickup_tools,
+        )
+
+        # Create the agency with a single agent
+        print("Creating agency...")
+        agency = Agency(
+            agency_chart=[
+                clickup_agent,  # Just the clickup agent
+            ]
+        )
+
+        # Send the query to the agent
+        print("Sending query to agent...")
+        print(f"Query: '{query}'")
+        print("Using agency.get_completion() method...")
+        response = agency.get_completion(query)
+
+        print(f"\nAgent response: {response}")
+
+        print("\nDone!")
+
+    except Exception as e:
+        print(f"\nAn error occurred: {e}")
+        print("\nTraceback:")
+        traceback.print_exception(*sys.exc_info())
+
+
+if __name__ == "__main__":
+    main()

agency_ai_demo/tests/test_clickup_tool_conversion.py

@@ -0,0 +1,145 @@
+import os
+import sys
+import inspect
+import logging
+from typing import List, Type, Any, Callable, Dict, Tuple
+
+# Setup logging
+logging.basicConfig(level=logging.INFO, format="%(levelname)s: %(message)s")
+logger = logging.getLogger(__name__)
+
+# Add the parent directory to sys.path to enable imports
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+# Import the tool wrapper
+from utils.tool_wrapper import convert_langchain_tool, LangChainBaseTool
+
+# Import all tools from the clickup_tools module
+from tools.clickup_tools import (
+    CreateTaskTool,
+    DeleteTaskTool,
+    UpdateTaskTool,
+    CreateTaskCommentTool,
+    GetTaskCommentsTool,
+    AddTaskToListTool,
+    RemoveTaskFromListTool,
+    CreateFolderlessListTool,
+    GetListsTool,
+    AddDependencyTool,
+    GetAuthorizedTeamsWorkspacesTool,
+    GetAuthorizedUserTool,
+    GetFolderTool,
+    GetFoldersTool,
+    GetListTool,
+    GetListMembersTool,
+    GetTaskMembersTool,
+    GetSpaceTool,
+    GetSpacesTool,
+    GetFilteredTeamTasksTool,
+    GetTasksTool,
+    GetTaskTool,
+    GetWorkspaceSeatsTool,
+    date_to_timestamp,
+)
+
+
+def test_tool_conversion():
+    """
+    Test the conversion of all ClickUp tools to agency-swarm compatible tools.
+    """
+    # Define all tools to test explicitly
+    class_tools = [
+        CreateTaskTool,
+        DeleteTaskTool,
+        UpdateTaskTool,
+        CreateTaskCommentTool,
+        GetTaskCommentsTool,
+        AddTaskToListTool,
+        RemoveTaskFromListTool,
+        CreateFolderlessListTool,
+        GetListsTool,
+        AddDependencyTool,
+        GetAuthorizedTeamsWorkspacesTool,
+        GetAuthorizedUserTool,
+        GetFolderTool,
+        GetFoldersTool,
+        GetListTool,
+        GetListMembersTool,
+        GetTaskMembersTool,
+        GetSpaceTool,
+        GetSpacesTool,
+        GetFilteredTeamTasksTool,
+        GetTasksTool,
+        GetTaskTool,
+        GetWorkspaceSeatsTool,
+    ]
+
+    function_tools = [date_to_timestamp]
+
+    logger.info(
+        f"Testing {len(class_tools)} class-based tools and {len(function_tools)} function-based tools"
+    )
+
+    # Test conversion of class-based tools
+    successful_conversions = []
+    failed_conversions = []
+
+    # Convert and test each class-based tool
+    for tool_cls in class_tools:
+        tool_name = tool_cls.__name__
+        try:
+            logger.info(f"Converting {tool_name}...")
+            converted_tool = convert_langchain_tool(tool_cls)
+            successful_conversions.append((tool_name, converted_tool))
+            logger.info(f"✅ Successfully converted {tool_name}")
+        except Exception as e:
+            logger.error(f"❌ Failed to convert {tool_name}: {e}")
+            failed_conversions.append((tool_name, str(e)))
+
+    # Convert and test each function-based tool
+    for tool_func in function_tools:
+        # For function tools, use the function's name directly
+        if hasattr(tool_func, "__name__"):
+            tool_name = tool_func.__name__
+        else:
+            tool_name = "function_tool"
+
+        try:
+            logger.info(f"Converting {tool_name}...")
+            converted_tool = convert_langchain_tool(tool_func)
+            successful_conversions.append((tool_name, converted_tool))
+            logger.info(f"✅ Successfully converted {tool_name}")
+        except Exception as e:
+            logger.error(f"❌ Failed to convert {tool_name}: {e}")
+            failed_conversions.append((tool_name, str(e)))
+
+    # Print summary report
+    print("\n" + "=" * 50)
+    print("CONVERSION SUMMARY REPORT")
+    print("=" * 50)
+    print(f"Total tools tested: {len(class_tools) + len(function_tools)}")
+    print(f"Successfully converted: {len(successful_conversions)}")
+    print(f"Failed conversions: {len(failed_conversions)}")
+
+    if successful_conversions:
+        print("\n" + "-" * 25)
+        print("SUCCESSFULLY CONVERTED TOOLS:")
+        print("-" * 25)
+        for name, _ in successful_conversions:
+            print(f"✅ {name}")
+
+    if failed_conversions:
+        print("\n" + "-" * 25)
+        print("FAILED CONVERSIONS:")
+        print("-" * 25)
+        for name, error in failed_conversions:
+            print(f"❌ {name}: {error}")
+
+    return successful_conversions, failed_conversions
+
+
+if __name__ == "__main__":
+    successful_tools, failed_tools = test_tool_conversion()
+
+    # Exit with non-zero code if any conversions failed
+    sys.exit(1 if failed_tools else 0)

agency_ai_demo/tests/test_clickup_tools.py

@@ -0,0 +1,170 @@
+import os
+import sys
+from dotenv import load_dotenv
+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 (
+    CreateTaskTool,
+    DeleteTaskTool,
+    UpdateTaskTool,
+    AddDependencyTool,
+    GetListTool,
+    GetTasksTool,
+    GetTaskTool,
+)
+
+# Load environment variables
+load_dotenv()
+
+
+def test_create_task():
+    print("\n===== Testing CreateTaskTool =====")
+    tool = CreateTaskTool()
+    result = tool._run(
+        list_id="901307715461",
+        name="TEST TASK FROM DIRECT TOOL",
+        description="This is a test task created using the CreateTaskTool directly",
+        assignees=[81918955],
+    )
+    print("Result:", result)
+    return result
+
+
+def test_get_tasks():
+    print("\n===== Testing GetTasksTool =====")
+    tool = GetTasksTool()
+    result = tool._run(list_id="901307715461")
+    print("Tasks found:", len(result))
+
+    # Print the first task details as an example
+    if result and len(result) > 0:
+        print("Example task:")
+        task = result[0]
+        print(f"ID: {task.get('id')}")
+        print(f"Name: {task.get('name')}")
+        print(f"Status: {task.get('status')}")
+        print(f"Assignees: {task.get('assignees')}")
+
+    return result
+
+
+def test_get_task(task_id=None):
+    print("\n===== Testing GetTaskTool =====")
+    # If no task_id is provided, get the first task from the list
+    if not task_id:
+        tasks = test_get_tasks()
+        if tasks and len(tasks) > 0:
+            task_id = tasks[0].get("id")
+            print(f"Using first task ID: {task_id}")
+
+    tool = GetTaskTool()
+    result = tool._run(task_id=task_id)
+    print("Result:", json.dumps(result, indent=2))
+    return result
+
+
+def test_update_task(task_id=None):
+    print("\n===== Testing UpdateTaskTool =====")
+    # If no task_id is provided, get the first task from the list
+    if not task_id:
+        tasks = test_get_tasks()
+        if tasks and len(tasks) > 0:
+            task_id = tasks[0].get("id")
+            print(f"Using first task ID: {task_id}")
+
+    tool = UpdateTaskTool()
+    result = tool._run(
+        task_id=task_id,
+        name="UPDATED TASK NAME",
+        description="This task was updated using the UpdateTaskTool directly",
+    )
+    print("Result:", result)
+    return result
+
+
+def test_add_dependency(task_id=None, depends_on=None):
+    print("\n===== Testing AddDependencyTool =====")
+    # If no task_ids are provided, get the first two tasks from the list
+    if not task_id or not depends_on:
+        tasks = test_get_tasks()
+        if tasks and len(tasks) >= 2:
+            task_id = tasks[0].get("id")
+            depends_on = tasks[1].get("id")
+            print(f"Using task ID: {task_id}")
+            print(f"Using depends_on ID: {depends_on}")
+
+    tool = AddDependencyTool()
+    result = tool._run(task_id=task_id, depends_on=depends_on)
+    print("Result:", result)
+    return result
+
+
+def test_delete_task(task_id=None):
+    print("\n===== Testing DeleteTaskTool =====")
+    # If no task_id is provided, create a new task to delete
+    if not task_id:
+        created_task = test_create_task()
+        if isinstance(created_task, dict) and "id" in created_task:
+            task_id = created_task["id"]
+            print(f"Using newly created task ID: {task_id}")
+
+    tool = DeleteTaskTool()
+    result = tool._run(task_id=task_id)
+    print("Result:", result)
+    return result
+
+
+def test_get_list():
+    print("\n===== Testing GetListTool =====")
+    tool = GetListTool()
+    result = tool._run(list_id="901307715461")
+    print("Result:", json.dumps(result, indent=2))
+    return result
+
+
+def test_all_tools():
+    print("Starting ClickUp tools tests...")
+
+    # Step 1: Get all tasks to see what's already there
+    tasks = test_get_tasks()
+
+    # Step 2: Create a new task
+    created_task = test_create_task()
+    if isinstance(created_task, dict) and "id" in created_task:
+        new_task_id = created_task.get("id")
+        print(f"Created new task with ID: {new_task_id}")
+
+        # Step 3: Get the task details
+        test_get_task(new_task_id)
+
+        # Step 4: Update the task
+        test_update_task(new_task_id)
+
+        # Step 5: Create another task for dependency testing
+        second_task = test_create_task()
+        if isinstance(second_task, dict) and "id" in second_task:
+            second_task_id = second_task.get("id")
+            print(f"Created second task with ID: {second_task_id}")
+
+            # Step 6: Add dependency between tasks
+            test_add_dependency(new_task_id, second_task_id)
+
+            # Step 7: Get list details
+            test_get_list()
+
+            # Step 8: Delete the tasks we created (cleanup)
+            test_delete_task(new_task_id)
+            test_delete_task(second_task_id)
+        else:
+            print("Failed to create second task, skipping dependency test")
+    else:
+        print("Failed to create task, skipping remaining tests")
+
+    print("\nAll tests completed!")
+
+
+if __name__ == "__main__":
+    test_all_tools()

agency_ai_demo/tests/test_list_tasks.py

@@ -0,0 +1,47 @@
+import os
+from dotenv import load_dotenv
+from openai import AzureOpenAI
+from agency_swarm import set_openai_client
+from agency_swarm import Agent
+
+from tools.clickup_tools import initialize_clickup_tools
+
+load_dotenv()
+
+
+def main():
+    print("Initializing OpenAI client...")
+    client = AzureOpenAI(
+        api_key=os.getenv("AZURE_OPENAI_API_KEY"),
+        api_version=os.getenv("OPENAI_API_VERSION"),
+        azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
+        timeout=5,
+        max_retries=5,
+    )
+
+    set_openai_client(client)
+
+    print("Converting ClickUp tools...")
+    clickup_tools = initialize_clickup_tools()
+
+    print("Creating test agent...")
+    clickup_agent = Agent(
+        name="clickup_agent",
+        description="I am a simple agent for testing tools",
+        model="gpt-4o",
+        tools=clickup_tools,
+    )
+
+    print("\nTesting get_tasks_tool...")
+    # Test GetTasksTool directly without using the agent
+    from tools.clickup_tools import GetTasksTool
+
+    tool = GetTasksTool()
+    result = tool._run(list_id=901307715461)
+    print(f"Result from GetTasksTool: {result}")
+
+    print("\nDone!")
+
+
+if __name__ == "__main__":
+    main()

agency_ai_demo/tests/test_tool_directly.py

@@ -0,0 +1,86 @@
+import os
+import sys
+import json
+import traceback
+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
+
+load_dotenv()
+
+
+def main():
+    try:
+        print("Initializing OpenAI client...")
+        client = AzureOpenAI(
+            api_key=os.getenv("AZURE_OPENAI_API_KEY"),
+            api_version=os.getenv("OPENAI_API_VERSION"),
+            azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
+            timeout=10,
+            max_retries=5,
+        )
+
+        set_openai_client(client)
+
+        # First test the tool directly
+        print("\n==== Testing GetTasksTool directly ====")
+        tool = GetTasksTool()
+        result = tool._run(list_id=901307715461)
+        print(f"Direct result: {json.dumps(result, indent=2)}")
+
+        print("\n==== Testing GetTaskTool directly ====")
+        get_task_tool = GetTaskTool()
+        task_id = "86a700c6e"
+        direct_result = get_task_tool._run(task_id=task_id)
+        print(f"Direct result: {json.dumps(direct_result, indent=2)}")
+
+        # Now test through Agency Swarm
+        print("\n==== Testing through Agency Swarm ====")
+
+        # Convert tools
+        print("Converting ClickUp tools...")
+        clickup_tools = initialize_clickup_tools()
+
+        # Instructions
+        clickup_instructions = """
+        You need to use the get_tasks_tool to retrieve tasks from list_id 901307715461.
+        Always show the full output to the user.
+        """
+
+        print("Creating ClickUp agent...")
+        clickup_agent = Agent(
+            name="clickup_agent",
+            description="I help with ClickUp tasks",
+            instructions=clickup_instructions,
+            model="gpt-4o",
+            tools=clickup_tools,
+        )
+
+        # Create the agency with a single agent
+        print("Creating agency...")
+        agency = Agency(
+            agency_chart=[
+                clickup_agent,
+            ]
+        )
+
+        # Run the query through the agency
+        print("\nSending query to agent...")
+        user_query = "Use the get_tasks_tool to get all tasks in list_id 901307715461 and show me the complete results without summarizing."
+        print(f"Query: '{user_query}'")
+
+        response = agency.get_completion(user_query)
+        print(f"\nAgent response: {response}")
+
+        print("\nDone!")
+
+    except Exception as e:
+        print(f"\nAn error occurred: {e}")
+        print("\nTraceback:")
+        traceback.print_exception(*sys.exc_info())
+
+
+if __name__ == "__main__":
+    main()