pygeai 0.5.0__py3-none-any.whl → 0.6.0b3__py3-none-any.whl

pygeai/tests/cli/commands/lab/test_spec.py

@@ -25,21 +25,14 @@ class TestSpec(unittest.TestCase):
             show_help()
             mock_stdout.assert_called_once()
 
-    def test_load_agent_missing_project_id(self):
-        option_list = [(Option("file", ["--file"], "", True), "file.json")]
-        with self.assertRaises(MissingRequirementException) as cm:
-            load_agent(option_list)
-        self.assertEqual(str(cm.exception), "Project ID must be defined.")
-
     def test_load_agent_missing_file(self):
-        option_list = [(Option("project_id", ["--project-id"], "", True), "proj123")]
+        option_list = []
         with self.assertRaises(MissingRequirementException) as cm:
             load_agent(option_list)
         self.assertEqual(str(cm.exception), "Cannot load agent definition without specifying path to JSON file.")
 
     def test_load_agent_single_dict(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -50,11 +43,10 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.AgentParser.get_agent', return_value=agent_mock), \
              patch('pygeai.cli.commands.lab.spec.create_agent') as mock_create:
             load_agent(option_list)
-            mock_create.assert_called_once_with("proj123", agent_mock, False)
+            mock_create.assert_called_once_with(agent_mock, False)
 
     def test_load_agent_list(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -66,8 +58,8 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.create_agent') as mock_create:
             load_agent(option_list)
             self.assertEqual(mock_create.call_count, 2)
-            mock_create.assert_any_call("proj123", agent_mock1, False)
-            mock_create.assert_any_call("proj123", agent_mock2, False)
+            mock_create.assert_any_call(agent_mock1, False)
+            mock_create.assert_any_call(agent_mock2, False)
 
     def test_create_agent_success(self):
         agent_mock = MagicMock()
@@ -76,7 +68,7 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.Console.write_stdout') as mock_stdout:
             mock_manager.return_value = MagicMock()
             mock_manager.return_value.create_agent.return_value = created_agent_mock
-            create_agent("proj123", agent_mock, False)
+            create_agent(agent_mock, False)
             mock_stdout.assert_called_once_with(f"Created agent detail: \n{created_agent_mock}")
 
     def test_create_agent_failure(self):
@@ -84,25 +76,18 @@ class TestSpec(unittest.TestCase):
         with patch('pygeai.cli.commands.lab.spec.AILabManager', side_effect=Exception("Error")), \
              patch('pygeai.cli.commands.lab.spec.Console.write_stderr') as mock_stderr, \
              patch('pygeai.cli.commands.lab.spec.logger.error') as mock_logger:
-            create_agent("proj123", agent_mock, False)
+            create_agent(agent_mock, False)
             mock_stderr.assert_called_once_with(f"Error creating agent: \n{agent_mock}")
             mock_logger.assert_called_once()
 
-    def test_load_tool_missing_project_id(self):
-        option_list = [(Option("file", ["--file"], "", True), "file.json")]
-        with self.assertRaises(MissingRequirementException) as cm:
-            load_tool(option_list)
-        self.assertEqual(str(cm.exception), "Project ID must be defined.")
-
     def test_load_tool_missing_file(self):
-        option_list = [(Option("project_id", ["--project-id"], "", True), "proj123")]
+        option_list = []
         with self.assertRaises(MissingRequirementException) as cm:
             load_tool(option_list)
         self.assertEqual(str(cm.exception), "Cannot load tool definition without specifying path to JSON file.")
 
     def test_load_tool_single_dict(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -113,11 +98,10 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.ToolParser.get_tool', return_value=tool_mock), \
              patch('pygeai.cli.commands.lab.spec.create_tool') as mock_create:
             load_tool(option_list)
-            mock_create.assert_called_once_with("proj123", tool_mock, False)
+            mock_create.assert_called_once_with(tool_mock, False)
 
     def test_load_tool_list(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -129,8 +113,8 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.create_tool') as mock_create:
             load_tool(option_list)
             self.assertEqual(mock_create.call_count, 2)
-            mock_create.assert_any_call("proj123", tool_mock1, False)
-            mock_create.assert_any_call("proj123", tool_mock2, False)
+            mock_create.assert_any_call(tool_mock1, False)
+            mock_create.assert_any_call(tool_mock2, False)
 
     def test_create_tool_success(self):
         tool_mock = MagicMock()
@@ -139,7 +123,7 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.Console.write_stdout') as mock_stdout:
             mock_manager.return_value = MagicMock()
             mock_manager.return_value.create_tool.return_value = created_tool_mock
-            create_tool("proj123", tool_mock, False)
+            create_tool(tool_mock, False)
             mock_stdout.assert_called_once_with(f"Created tool detail: \n{created_tool_mock}")
 
     def test_create_tool_failure(self):
@@ -147,25 +131,18 @@ class TestSpec(unittest.TestCase):
         with patch('pygeai.cli.commands.lab.spec.AILabManager', side_effect=Exception("Error")), \
              patch('pygeai.cli.commands.lab.spec.Console.write_stderr') as mock_stderr, \
              patch('pygeai.cli.commands.lab.spec.logger.error') as mock_logger:
-            create_tool("proj123", tool_mock, False)
+            create_tool(tool_mock, False)
             mock_stderr.assert_called_once_with(f"Error creating tool: \n{tool_mock}")
             mock_logger.assert_called_once()
 
-    def test_load_task_missing_project_id(self):
-        option_list = [(Option("file", ["--file"], "", True), "file.json")]
-        with self.assertRaises(MissingRequirementException) as cm:
-            load_task(option_list)
-        self.assertEqual(str(cm.exception), "Project ID must be defined.")
-
     def test_load_task_missing_file(self):
-        option_list = [(Option("project_id", ["--project-id"], "", True), "proj123")]
+        option_list = []
         with self.assertRaises(MissingRequirementException) as cm:
             load_task(option_list)
         self.assertEqual(str(cm.exception), "Cannot load task definition without specifying path to JSON file.")
 
     def test_load_task_single_dict(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -176,11 +153,10 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.TaskParser.get_task', return_value=task_mock), \
              patch('pygeai.cli.commands.lab.spec.create_task') as mock_create:
             load_task(option_list)
-            mock_create.assert_called_once_with("proj123", task_mock, False)
+            mock_create.assert_called_once_with(task_mock, False)
 
     def test_load_task_list(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -192,8 +168,8 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.create_task') as mock_create:
             load_task(option_list)
             self.assertEqual(mock_create.call_count, 2)
-            mock_create.assert_any_call("proj123", task_mock1, False)
-            mock_create.assert_any_call("proj123", task_mock2, False)
+            mock_create.assert_any_call(task_mock1, False)
+            mock_create.assert_any_call(task_mock2, False)
 
     def test_create_task_success(self):
         task_mock = MagicMock()
@@ -202,7 +178,7 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.Console.write_stdout') as mock_stdout:
             mock_manager.return_value = MagicMock()
             mock_manager.return_value.create_task.return_value = created_task_mock
-            create_task("proj123", task_mock, False)
+            create_task(task_mock, False)
             mock_stdout.assert_called_once_with(f"Created task detail: \n{created_task_mock}")
 
     def test_create_task_failure(self):
@@ -210,25 +186,18 @@ class TestSpec(unittest.TestCase):
         with patch('pygeai.cli.commands.lab.spec.AILabManager', side_effect=Exception("Error")), \
              patch('pygeai.cli.commands.lab.spec.Console.write_stderr') as mock_stderr, \
              patch('pygeai.cli.commands.lab.spec.logger.error') as mock_logger:
-            create_task("proj123", task_mock, False)
+            create_task(task_mock, False)
             mock_stderr.assert_called_once_with(f"Error creating task: \n{task_mock}")
             mock_logger.assert_called_once()
 
-    def test_load_agentic_process_missing_project_id(self):
-        option_list = [(Option("file", ["--file"], "", True), "file.json")]
-        with self.assertRaises(MissingRequirementException) as cm:
-            load_agentic_process(option_list)
-        self.assertEqual(str(cm.exception), "Project ID must be defined.")
-
     def test_load_agentic_process_missing_file(self):
-        option_list = [(Option("project_id", ["--project-id"], "", True), "proj123")]
+        option_list = []
         with self.assertRaises(MissingRequirementException) as cm:
             load_agentic_process(option_list)
         self.assertEqual(str(cm.exception), "Cannot load agentic process definition without specifying path to JSON file.")
 
     def test_load_agentic_process_single_dict(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -239,11 +208,10 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.AgenticProcessParser.get_agentic_process', return_value=process_mock), \
              patch('pygeai.cli.commands.lab.spec.create_agentic_process') as mock_create:
             load_agentic_process(option_list)
-            mock_create.assert_called_once_with("proj123", process_mock, False)
+            mock_create.assert_called_once_with(process_mock, False)
 
     def test_load_agentic_process_list(self):
         option_list = [
-            (Option("project_id", ["--project-id"], "", True), "proj123"),
             (Option("file", ["--file"], "", True), "file.json"),
             (Option("automatic_publish", ["--automatic-publish"], "", True), False)
         ]
@@ -255,8 +223,8 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.create_agentic_process') as mock_create:
             load_agentic_process(option_list)
             self.assertEqual(mock_create.call_count, 2)
-            mock_create.assert_any_call("proj123", process_mock1, False)
-            mock_create.assert_any_call("proj123", process_mock2, False)
+            mock_create.assert_any_call(process_mock1, False)
+            mock_create.assert_any_call(process_mock2, False)
 
     def test_create_agentic_process_success(self):
         process_mock = MagicMock()
@@ -265,7 +233,7 @@ class TestSpec(unittest.TestCase):
              patch('pygeai.cli.commands.lab.spec.Console.write_stdout') as mock_stdout:
             mock_manager.return_value = MagicMock()
             mock_manager.return_value.create_process.return_value = created_process_mock
-            create_agentic_process("proj123", process_mock, False)
+            create_agentic_process(process_mock, False)
             mock_stdout.assert_called_once_with(f"Created agentic process detail: \n{created_process_mock}")
 
     def test_create_agentic_process_failure(self):
@@ -273,6 +241,6 @@ class TestSpec(unittest.TestCase):
         with patch('pygeai.cli.commands.lab.spec.AILabManager', side_effect=Exception("Error")), \
              patch('pygeai.cli.commands.lab.spec.Console.write_stderr') as mock_stderr, \
              patch('pygeai.cli.commands.lab.spec.logger.error') as mock_logger:
-            create_agentic_process("proj123", process_mock, False)
+            create_agentic_process(process_mock, False)
             mock_stderr.assert_called_once_with(f"Error creating agentic process: \n{process_mock}")
             mock_logger.assert_called_once()

pygeai/tests/cli/commands/test_chat.py

@@ -52,13 +52,19 @@ class TestChatCommands(unittest.TestCase):
         mock_iris.assert_called_once()
         mock_iris_instance.stream_answer.assert_called()
 
+    @patch('pygeai.cli.commands.chat.get_project_id')
+    @patch('pygeai.cli.commands.chat.AgentClient')
     @patch('pygeai.cli.commands.chat.AgentChatSession')
     @patch('prompt_toolkit.PromptSession.prompt')
-    def test_chat_with_agent_success(self, mock_prompt, mock_agent_session):
+    def test_chat_with_agent_success(self, mock_prompt, mock_agent_session, mock_agent_client, mock_get_project_id):
         mock_prompt.side_effect = ["Hello", "exit"]
         mock_session_instance = mock_agent_session.return_value
         mock_session_instance.get_answer.return_value = "Hello, I'm your agent!"
         mock_session_instance.stream_answer.return_value = iter(["Hi there!"])
+        
+        mock_get_project_id.return_value = "test-project-id"
+        mock_client_instance = mock_agent_client.return_value
+        mock_client_instance.get_agent.return_value = {"agentId": "123", "agentName": "test-agent"}
 
         option_list = [
             (Option("agent_name", ["--agent-name"], "", True), "test-agent")
@@ -73,16 +79,22 @@ class TestChatCommands(unittest.TestCase):
         with self.assertRaises(MissingRequirementException):
             chat_with_agent(option_list)
 
+    @patch('pygeai.cli.commands.chat.get_project_id')
+    @patch('pygeai.cli.commands.chat.AgentClient')
     @patch('pygeai.cli.commands.chat.AgentChatSession')
     @patch('prompt_toolkit.PromptSession.prompt')
     @patch('builtins.open', new_callable=MagicMock)
     @patch('json.load')
-    def test_chat_with_agent_restore_session_success(self, mock_json_load, mock_open, mock_prompt, mock_agent_session):
+    def test_chat_with_agent_restore_session_success(self, mock_json_load, mock_open, mock_prompt, mock_agent_session, mock_agent_client, mock_get_project_id):
         mock_prompt.side_effect = ["Hello", "exit"]
         mock_session_instance = mock_agent_session.return_value
         mock_session_instance.get_answer.return_value = "Hello, I'm your agent!"
         mock_session_instance.stream_answer.return_value = iter(["Hi there!"])
         mock_json_load.return_value = [{"role": "user", "content": "Hi"}, {"role": "assistant", "content": "Hello!"}]
+        
+        mock_get_project_id.return_value = "test-project-id"
+        mock_client_instance = mock_agent_client.return_value
+        mock_client_instance.get_agent.return_value = {"agentId": "123", "agentName": "test-agent"}
 
         option_list = [
             (Option("agent_name", ["--agent-name"], "", True), "test-agent"),
@@ -92,15 +104,21 @@ class TestChatCommands(unittest.TestCase):
         chat_with_agent(option_list)
         mock_json_load.assert_called_once()
 
+    @patch('pygeai.cli.commands.chat.get_project_id')
+    @patch('pygeai.cli.commands.chat.AgentClient')
     @patch('pygeai.cli.commands.chat.AgentChatSession')
     @patch('prompt_toolkit.PromptSession.prompt')
     @patch('builtins.open', new_callable=MagicMock)
     @patch('json.dump')
-    def test_chat_with_agent_save_session_success(self, mock_json_dump, mock_open, mock_prompt, mock_agent_session):
+    def test_chat_with_agent_save_session_success(self, mock_json_dump, mock_open, mock_prompt, mock_agent_session, mock_agent_client, mock_get_project_id):
         mock_prompt.side_effect = ["Hello", "exit"]
         mock_session_instance = mock_agent_session.return_value
         mock_session_instance.get_answer.return_value = "Hello, I'm your agent!"
         mock_session_instance.stream_answer.return_value = iter(["Hi there!"])
+        
+        mock_get_project_id.return_value = "test-project-id"
+        mock_client_instance = mock_agent_client.return_value
+        mock_client_instance.get_agent.return_value = {"agentId": "123", "agentName": "test-agent"}
 
         option_list = [
             (Option("agent_name", ["--agent-name"], "", True), "test-agent"),