From a32875647b76499cfc27a3ac858bdee731dedfb6 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Wed, 16 Jul 2025 23:01:02 +0100
Subject: [PATCH] Fixed integration tests

---
 .../test_agent_manager_integration.py          | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/tests/integration/test_agent_manager_integration.py b/tests/integration/test_agent_manager_integration.py
index f3450df2..1f3966d1 100644
--- a/tests/integration/test_agent_manager_integration.py
+++ b/tests/integration/test_agent_manager_integration.py
@@ -69,39 +69,39 @@ class TestAgentManagerIntegration:
             "knowledge_query": Tool(
                 name="knowledge_query",
                 description="Query the knowledge graph for information",
-                arguments={
-                    "question": Argument(
+                arguments=[
+                    Argument(
                         name="question",
                         type="string", 
                         description="The question to ask the knowledge graph"
                     )
-                },
+                ],
                 implementation=KnowledgeQueryImpl,
                 config={}
             ),
             "text_completion": Tool(
                 name="text_completion",
                 description="Generate text completion using LLM",
-                arguments={
-                    "question": Argument(
+                arguments=[
+                    Argument(
                         name="question",
                         type="string",
                         description="The question to ask the LLM"
                     )
-                },
+                ],
                 implementation=TextCompletionImpl,
                 config={}
             ),
             "web_search": Tool(
                 name="web_search",
                 description="Search the web for information",
-                arguments={
-                    "query": Argument(
+                arguments=[
+                    Argument(
                         name="query",
                         type="string",
                         description="The search query"
                     )
-                },
+                ],
                 implementation=lambda context: AsyncMock(invoke=AsyncMock(return_value="Web search results")),
                 config={}
             )