pygeai 0.4.0b12__py3-none-any.whl → 0.6.0b3__py3-none-any.whl

pygeai/tests/migration/test_strategies.py

@@ -184,8 +184,8 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_agent.assert_called_with(project_id=self.from_project_id, agent_id=self.agent_id)
-        mock_destination.create_agent.assert_called_with(project_id=self.to_project_id, agent=mock_agent)
+        mock_source.get_agent.assert_called_with(agent_id=self.agent_id)
+        mock_destination.create_agent.assert_called_with(agent=mock_agent)
         mock_stdout.assert_called_once()
         self.assertIn("New agent detail:", mock_stdout.call_args[0][0])
         mock_stderr.assert_not_called()
@@ -211,7 +211,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_agent.assert_called_with(project_id=self.from_project_id, agent_id=self.agent_id)
+        mock_source.get_agent.assert_called_with(agent_id=self.agent_id)
         mock_destination.create_agent.assert_not_called()
         mock_stderr.assert_called_once()
         self.assertIn("Agent migration failed:", mock_stderr.call_args[0][0])
@@ -240,7 +240,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_agent.assert_called_with(project_id=self.from_project_id, agent_id=self.agent_id)
+        mock_source.get_agent.assert_called_with(agent_id=self.agent_id)
         mock_destination.create_agent.assert_not_called()
         mock_stderr.assert_called_once()
         mock_stdout.assert_called_once()
@@ -290,8 +290,8 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_tool.assert_called_with(project_id=self.from_project_id, tool_id=self.tool_id)
-        mock_destination.create_tool.assert_called_with(project_id=self.to_project_id, tool=mock_tool)
+        mock_source.get_tool.assert_called_with(tool_id=self.tool_id)
+        mock_destination.create_tool.assert_called_with(tool=mock_tool)
         mock_stdout.assert_called_once()
         self.assertIn("New tool detail:", mock_stdout.call_args[0][0])
         mock_stderr.assert_not_called()
@@ -317,7 +317,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_tool.assert_called_with(project_id=self.from_project_id, tool_id=self.tool_id)
+        mock_source.get_tool.assert_called_with(tool_id=self.tool_id)
         mock_destination.create_tool.assert_not_called()
         mock_stderr.assert_called_once()
         self.assertIn("Tool migration failed:", mock_stderr.call_args[0][0])
@@ -346,7 +346,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_tool.assert_called_with(project_id=self.from_project_id, tool_id=self.tool_id)
+        mock_source.get_tool.assert_called_with(tool_id=self.tool_id)
         mock_destination.create_tool.assert_not_called()
         mock_stderr.assert_called_once()
         mock_stdout.assert_called_once()
@@ -396,8 +396,8 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_process.assert_called_with(project_id=self.from_project_id, process_id=self.process_id)
-        mock_destination.create_process.assert_called_with(project_id=self.to_project_id, process=mock_process)
+        mock_source.get_process.assert_called_with(process_id=self.process_id)
+        mock_destination.create_process.assert_called_with(process=mock_process)
         mock_stdout.assert_called_once()
         self.assertIn("New process detail:", mock_stdout.call_args[0][0])
         mock_stderr.assert_not_called()
@@ -423,7 +423,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_process.assert_called_with(project_id=self.from_project_id, process_id=self.process_id)
+        mock_source.get_process.assert_called_with(process_id=self.process_id)
         mock_destination.create_process.assert_not_called()
         mock_stderr.assert_called_once()
         self.assertIn("Process migration failed:", mock_stderr.call_args[0][0])
@@ -452,7 +452,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_process.assert_called_with(project_id=self.from_project_id, process_id=self.process_id)
+        mock_source.get_process.assert_called_with(process_id=self.process_id)
         mock_destination.create_process.assert_not_called()
         mock_stdout.assert_called_once()
         mock_stderr.assert_called_once()
@@ -502,8 +502,8 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_task.assert_called_with(project_id=self.from_project_id, task_id=self.task_id)
-        mock_destination.create_task.assert_called_with(project_id=self.to_project_id, task=mock_task)
+        mock_source.get_task.assert_called_with(task_id=self.task_id)
+        mock_destination.create_task.assert_called_with(task=mock_task)
         mock_stdout.assert_called_once()
         self.assertIn("New task detail:", mock_stdout.call_args[0][0])
         mock_stderr.assert_not_called()
@@ -529,7 +529,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_task.assert_called_with(project_id=self.from_project_id, task_id=self.task_id)
+        mock_source.get_task.assert_called_with(task_id=self.task_id)
         mock_destination.create_task.assert_not_called()
         mock_stderr.assert_called_once()
         self.assertIn("Task migration failed:", mock_stderr.call_args[0][0])
@@ -558,7 +558,7 @@ class TestMigrationStrategies(unittest.TestCase):
         )
         strategy.migrate()
 
-        mock_source.get_task.assert_called_with(project_id=self.from_project_id, task_id=self.task_id)
+        mock_source.get_task.assert_called_with(task_id=self.task_id)
         mock_destination.create_task.assert_not_called()
         mock_stderr.assert_called_once()
         mock_stdout.assert_called_once()

pygeai/tests/organization/test_mappers.py

@@ -129,11 +129,18 @@ class TestOrganizationResponseMapper(unittest.TestCase):
         self.assertEqual(result.tokens[0].name, "Token1")
 
     def test_map_to_item_list_response_with_valid_data(self):
-        data = {"items": [{"assistant": "Assistant1", "timestamp": "2023-01-01T00:00:00", "status": "Completed"}]}
+        data = {"items": [{"assistant": "Assistant1", "timestamp": "2023-01-01T00:00:00", "status": "succeeded"}]}
         expected_item = RequestItem(
+            api_token="token123",
             assistant="Assistant1",
-            timestamp="2023-01-01T00:00:00",
-            status="Completed"
+            cost=0.5,
+            elapsed_time_ms=1000,
+            end_timestamp="2023-01-01T00:01:00",
+            module="test_module",
+            session_id="session123",
+            start_timestamp="2023-01-01T00:00:00",
+            status="succeeded",
+            timestamp="2023-01-01T00:00:00"
         )
 
         with patch('pygeai.core.base.mappers.ModelMapper.map_to_item_list', return_value=[expected_item]):
@@ -142,5 +149,5 @@ class TestOrganizationResponseMapper(unittest.TestCase):
         self.assertIsInstance(result, ProjectItemListResponse)
         self.assertEqual(len(result.items), 1)
         self.assertEqual(result.items[0].assistant, "Assistant1")
-        self.assertEqual(result.items[0].status, "Completed")
+        self.assertEqual(result.items[0].status, "succeeded")
 

pygeai/tests/organization/test_responses.py

@@ -0,0 +1,137 @@
+import unittest
+from unittest.mock import MagicMock
+from datetime import datetime
+
+from pygeai.organization.responses import (
+    AssistantListResponse,
+    ProjectListResponse,
+    ProjectDataResponse,
+    ProjectTokensResponse,
+    ProjectItemListResponse
+)
+from pygeai.core.models import Assistant, Project, ProjectToken, RequestItem
+
+
+class TestOrganizationResponses(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.organization.test_responses.TestOrganizationResponses
+    """
+    
+    def _create_request_item(self, index=1):
+        """Helper method to create a valid RequestItem for testing"""
+        return RequestItem(
+            apiToken=f"token-{index}",
+            assistant=f"assistant-{index}",
+            cost=0.01 * index,
+            elapsedTimeMs=100 * index,
+            endTimestamp=datetime.now(),
+            module=f"module-{index}",
+            sessionId=f"session-{index}",
+            startTimestamp=datetime.now(),
+            status="succeeded",
+            timestamp=datetime.now()
+        )
+
+    def test_assistant_list_response(self):
+        assistant1 = MagicMock(spec=Assistant)
+        assistant2 = MagicMock(spec=Assistant)
+        
+        response = AssistantListResponse(assistants=[assistant1, assistant2])
+        
+        self.assertEqual(len(response.assistants), 2)
+        self.assertEqual(response.assistants[0], assistant1)
+
+    def test_project_list_response(self):
+        project1 = MagicMock(spec=Project)
+        project2 = MagicMock(spec=Project)
+        
+        response = ProjectListResponse(projects=[project1, project2])
+        
+        self.assertEqual(len(response.projects), 2)
+        self.assertEqual(response.projects[0], project1)
+
+    def test_project_data_response(self):
+        project = MagicMock(spec=Project)
+        
+        response = ProjectDataResponse(project=project)
+        
+        self.assertEqual(response.project, project)
+
+    def test_project_tokens_response(self):
+        token1 = MagicMock(spec=ProjectToken)
+        token2 = MagicMock(spec=ProjectToken)
+        
+        response = ProjectTokensResponse(tokens=[token1, token2])
+        
+        self.assertEqual(len(response.tokens), 2)
+        self.assertEqual(response.tokens[0], token1)
+
+    def test_project_item_list_response_to_list(self):
+        item1 = self._create_request_item(1)
+        item2 = self._create_request_item(2)
+        
+        response = ProjectItemListResponse(items=[item1, item2])
+        result = response.to_list()
+        
+        self.assertEqual(len(result), 2)
+        self.assertIsInstance(result[0], dict)
+        self.assertIsInstance(result[1], dict)
+
+    def test_project_item_list_response_to_list_empty(self):
+        response = ProjectItemListResponse(items=[])
+        result = response.to_list()
+        
+        self.assertEqual(result, [])
+
+    def test_project_item_list_response_to_list_none(self):
+        response = ProjectItemListResponse(items=[])
+        result = response.to_list()
+        
+        self.assertEqual(result, [])
+
+    def test_project_item_list_response_getitem(self):
+        item1 = self._create_request_item(1)
+        item2 = self._create_request_item(2)
+        
+        response = ProjectItemListResponse(items=[item1, item2])
+        
+        self.assertEqual(response[0], item1)
+        self.assertEqual(response[1], item2)
+
+    def test_project_item_list_response_len(self):
+        item1 = self._create_request_item(1)
+        item2 = self._create_request_item(2)
+        
+        response = ProjectItemListResponse(items=[item1, item2])
+        
+        self.assertEqual(len(response), 2)
+
+    def test_project_item_list_response_len_empty(self):
+        response = ProjectItemListResponse(items=[])
+        
+        self.assertEqual(len(response), 0)
+
+    def test_project_item_list_response_iter(self):
+        item1 = self._create_request_item(1)
+        item2 = self._create_request_item(2)
+        
+        response = ProjectItemListResponse(items=[item1, item2])
+        
+        items = list(response)
+        self.assertEqual(len(items), 2)
+        self.assertEqual(items[0], item1)
+        self.assertEqual(items[1], item2)
+
+    def test_project_item_list_response_append(self):
+        item1 = self._create_request_item(1)
+        item2 = self._create_request_item(2)
+        
+        response = ProjectItemListResponse(items=[item1])
+        response.append(item2)
+        
+        self.assertEqual(len(response.items), 2)
+        self.assertEqual(response.items[1], item2)
+
+
+if __name__ == '__main__':
+    unittest.main()

pygeai/tests/snippets/lab/agentic_flow_example_4.py

@@ -22,23 +22,23 @@ def rollback():
     print("\n=== Initiating Rollback ===")
     if created_entities["instance_id"]:
         print(f"Deleting instance {created_entities['instance_id']}...")
-        result = manager.abort_instance(project_id=project_id, instance_id=created_entities["instance_id"])
+        result = manager.abort_instance(instance_id=created_entities["instance_id"])
         print(f"Rollback: {result}")
     if created_entities["process_id"]:
         print(f"Deleting process {created_entities['process_id']}...")
-        result = manager.delete_process(project_id=project_id, process_id=created_entities["process_id"])
+        result = manager.delete_process(process_id=created_entities["process_id"])
         print(f"Rollback: {result}")
     if created_entities["task_id"]:
         print(f"Deleting task {created_entities['task_id']}...")
-        result = manager.delete_task(project_id=project_id, task_id=created_entities["task_id"])
+        result = manager.delete_task(task_id=created_entities["task_id"])
         print(f"Rollback: {result}")
     if created_entities["tool_id"]:
         print(f"Deleting tool {created_entities['tool_id']}...")
-        result = manager.delete_tool(project_id=project_id, tool_id=created_entities["tool_id"])
+        result = manager.delete_tool(tool_id=created_entities["tool_id"])
         print(f"Rollback: {result}")
     if created_entities["agent_id"]:
         print(f"Deleting agent {created_entities['agent_id']}...")
-        result = manager.delete_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+        result = manager.delete_agent(agent_id=created_entities["agent_id"])
         print(f"Rollback: {result}")
     print("Rollback complete.")
 
@@ -82,7 +82,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_tool_result = manager.create_tool(project_id=project_id, tool=tool, automatic_publish=False)
+    create_tool_result = manager.create_tool(tool=tool, automatic_publish=False)
     if isinstance(create_tool_result, Tool):
         print(f"Success: Created Tool: {create_tool_result.name}, ID: {create_tool_result.id}")
         created_entities["tool_id"] = create_tool_result.id
@@ -107,7 +107,7 @@ def main():
             value="English"
         )
     )
-    update_tool_result = manager.update_tool(project_id=project_id, tool=tool, automatic_publish=False)
+    update_tool_result = manager.update_tool(tool=tool, automatic_publish=False)
     if isinstance(update_tool_result, Tool):
         print(f"Success: Updated Tool: {update_tool_result.description}")
     else:
@@ -116,7 +116,7 @@ def main():
         exit()
 
     print("Publishing tool revision '1'...")
-    publish_tool_result = manager.publish_tool_revision(project_id=project_id, tool_id=created_entities["tool_id"], revision="1")
+    publish_tool_result = manager.publish_tool_revision(tool_id=created_entities["tool_id"], revision="1")
     if isinstance(publish_tool_result, Tool):
         print(f"Success: Published Tool Revision: {publish_tool_result.name}")
     else:
@@ -125,7 +125,7 @@ def main():
         exit()
 
     print("Retrieving latest tool version...")
-    latest_tool = manager.get_tool(project_id=project_id, tool_id=created_entities["tool_id"])
+    latest_tool = manager.get_tool(tool_id=created_entities["tool_id"])
     if isinstance(latest_tool, Tool):
         print(f"Success: Latest Tool: {latest_tool.name}, Description: {latest_tool.description}")
     else:
@@ -184,7 +184,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_agent_result = manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    create_agent_result = manager.create_agent(agent=agent, automatic_publish=False)
     if isinstance(create_agent_result, Agent):
         print(f"Success: Created Agent: {create_agent_result.name}, ID: {create_agent_result.id}")
         created_entities["agent_id"] = create_agent_result.id
@@ -200,7 +200,7 @@ def main():
     agent.description = "Improved agent for text summarization tasks"
     agent.job_description = "Summarizes text and extracts key insights"
 
-    update_agent_result = manager.update_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    update_agent_result = manager.update_agent(agent=agent, automatic_publish=False)
     if isinstance(update_agent_result, Agent):
         print(f"Success: Updated Agent: {update_agent_result.description}")
     else:
@@ -209,7 +209,7 @@ def main():
         exit()
 
     print("Publishing agent revision '1'...")
-    publish_agent_result = manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"], revision="1")
+    publish_agent_result = manager.publish_agent_revision(agent_id=created_entities["agent_id"], revision="1")
     if isinstance(publish_agent_result, Agent):
         print(f"Success: Published Agent Revision: {publish_agent_result.name}")
     else:
@@ -218,7 +218,7 @@ def main():
         exit()
 
     print("Retrieving latest agent version...")
-    latest_agent = manager.get_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+    latest_agent = manager.get_agent(agent_id=created_entities["agent_id"])
     if isinstance(latest_agent, Agent):
         print(f"Success: Latest Agent: {latest_agent.name}, Description: {latest_agent.description}")
     else:
@@ -259,7 +259,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_task_result = manager.create_task(project_id=project_id, task=task, automatic_publish=False)
+    create_task_result = manager.create_task(task=task, automatic_publish=False)
     if isinstance(create_task_result, Task):
         print(f"Success: Created Task: {create_task_result.name}, ID: {create_task_result.id}")
         created_entities["task_id"] = create_task_result.id
@@ -275,7 +275,7 @@ def main():
     task.description = "Enhanced task for text summarization and analysis"
     task.prompt_data.instructions = "Summarize text and highlight key insights."
 
-    update_task_result = manager.update_task(project_id=project_id, task=task, automatic_publish=False)
+    update_task_result = manager.update_task(task=task, automatic_publish=False)
     if isinstance(update_task_result, Task):
         print(f"Success: Updated Task: {update_task_result.description}")
     else:
@@ -284,7 +284,7 @@ def main():
         exit()
 
     print("Publishing task revision '1'...")
-    publish_task_result = manager.publish_task_revision(project_id=project_id, task_id=created_entities["task_id"], revision="1")
+    publish_task_result = manager.publish_task_revision(task_id=created_entities["task_id"], revision="1")
     if isinstance(publish_task_result, Task):
         print(f"Success: Published Task Revision: {publish_task_result.name}")
     else:
@@ -293,7 +293,7 @@ def main():
         exit()
 
     print("Retrieving latest task version...")
-    latest_task = manager.get_task(project_id=project_id, task_id=created_entities["task_id"])
+    latest_task = manager.get_task(task_id=created_entities["task_id"])
     if isinstance(latest_task, Task):
         print(f"Success: Latest Task: {latest_task.name}, Description: {latest_task.description}")
     else:
@@ -358,7 +358,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_process_result = manager.create_process(project_id=project_id, process=process, automatic_publish=False)
+    create_process_result = manager.create_process(process=process, automatic_publish=False)
     if isinstance(create_process_result, AgenticProcess):
         print(f"Success: Created Process: {create_process_result.name}, ID: {create_process_result.id}")
         created_entities["process_id"] = create_process_result.id
@@ -373,7 +373,7 @@ def main():
     process.description = "Optimized process for text summarization workflows"
     process.agentic_activities[0].name = "Advanced Text Summarization"
 
-    update_process_result = manager.update_process(project_id=project_id, process=process, automatic_publish=False)
+    update_process_result = manager.update_process(process=process, automatic_publish=False)
     if isinstance(update_process_result, AgenticProcess):
         print(f"Success: Updated Process: {update_process_result.description}")
     else:
@@ -382,7 +382,7 @@ def main():
         exit()
 
     print("Publishing process revision '1'...")
-    publish_process_result = manager.publish_process_revision(project_id=project_id, process_id=created_entities["process_id"], revision="1")
+    publish_process_result = manager.publish_process_revision(process_id=created_entities["process_id"], revision="1")
     if isinstance(publish_process_result, AgenticProcess):
         print(f"Success: Published Process Revision: {publish_process_result.name}")
     else:
@@ -391,7 +391,7 @@ def main():
         exit()
 
     print("Retrieving latest process version...")
-    latest_process = manager.get_process(project_id=project_id, process_id=created_entities["process_id"])
+    latest_process = manager.get_process(process_id=created_entities["process_id"])
     if isinstance(latest_process, AgenticProcess):
         print(f"Success: Latest Process: {latest_process.name}, Description: {latest_process.description}")
     else:
@@ -417,7 +417,7 @@ def main():
         exit()
 
     print("Retrieving instance information...")
-    instance_info = manager.get_instance(project_id=project_id, instance_id=created_entities["instance_id"])
+    instance_info = manager.get_instance(instance_id=created_entities["instance_id"])
     if isinstance(instance_info, ProcessInstance):
         print(f"Success: Instance Info: Process: {instance_info.process.name}, Subject: {instance_info.subject}")
     else:
@@ -426,7 +426,7 @@ def main():
         exit()
 
     print("Retrieving instance history...")
-    history = manager.get_instance_history(project_id=project_id, instance_id=created_entities["instance_id"])
+    history = manager.get_instance_history(instance_id=created_entities["instance_id"])
     if isinstance(history, dict):
         print(f"Success: Instance History: {history}")
     else:

pygeai/tests/snippets/lab/samples/summarize_files.py

@@ -31,7 +31,7 @@ def rollback():
     print("\n=== Initiating Rollback ===")
     if created_entities["agent_id"]:
         print(f"Deleting agent {created_entities['agent_id']}...")
-        result = lab_manager.delete_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+        result = lab_manager.delete_agent(agent_id=created_entities["agent_id"])
         print(f"Rollback: {result}")
     if created_entities["file_id"]:
         print(f"Deleting file {created_entities['file_id']}...")
@@ -107,7 +107,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_agent_result = lab_manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    create_agent_result = lab_manager.create_agent(agent=agent, automatic_publish=False)
     if isinstance(create_agent_result, Agent):
         print(f"Success: Created Agent: {create_agent_result.name}, ID: {create_agent_result.id}")
         created_entities["agent_id"] = create_agent_result.id
@@ -117,7 +117,7 @@ def main():
         exit()
 
     print("Publishing agent revision '1'...")
-    publish_agent_result = lab_manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"], revision="1")
+    publish_agent_result = lab_manager.publish_agent_revision(agent_id=created_entities["agent_id"], revision="1")
     if isinstance(publish_agent_result, Agent):
         print(f"Success: Published Agent Revision: {publish_agent_result.name}")
     else:

pygeai/tests/snippets/lab/use_cases/file_summarizer_example.py

@@ -27,7 +27,7 @@ def rollback():
     print("\n=== Initiating Rollback ===")
     if created_entities["agent_id"]:
         print(f"Deleting agent {created_entities['agent_id']}...")
-        result = lab_manager.delete_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+        result = lab_manager.delete_agent(agent_id=created_entities["agent_id"])
         print(f"Rollback: {result}")
     if created_entities["file_id"]:
         print(f"Deleting file {created_entities['file_id']}...")
@@ -102,7 +102,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_agent_result = lab_manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    create_agent_result = lab_manager.create_agent(agent=agent, automatic_publish=False)
     if isinstance(create_agent_result, Agent):
         print(f"Success: Created Agent: {create_agent_result.name}, ID: {create_agent_result.id}")
         created_entities["agent_id"] = create_agent_result.id
@@ -112,7 +112,7 @@ def main():
         exit()
 
     print("Publishing agent revision '1'...")
-    publish_agent_result = lab_manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"], revision="1")
+    publish_agent_result = lab_manager.publish_agent_revision(agent_id=created_entities["agent_id"], revision="1")
     if isinstance(publish_agent_result, Agent):
         print(f"Success: Published Agent Revision: {publish_agent_result.name}")
     else:

pygeai/tests/snippets/lab/use_cases/file_summarizer_example_2.py

@@ -27,7 +27,7 @@ def rollback():
     print("\n=== Initiating Rollback ===")
     if created_entities["agent_id"]:
         print(f"Deleting agent {created_entities['agent_id']}...")
-        result = lab_manager.delete_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+        result = lab_manager.delete_agent(agent_id=created_entities["agent_id"])
         print(f"Rollback: {result}")
     if created_entities["file_id"]:
         print(f"Deleting file {created_entities['file_id']}...")
@@ -37,8 +37,8 @@ def rollback():
 
 
 def main():
-    # File Upload Flow
-    print("\n=== File Upload Flow ===")
+    # File Upload Step
+    print("\n=== File Upload Step ===")
     print("Uploading file 'test.txt'...")
     file_to_upload = UploadFile(
         name="test.txt",
@@ -65,8 +65,8 @@ def main():
         rollback()
         exit()
 
-    # Agent Flow
-    print("\n=== Agent Flow ===")
+    # Agent Step
+    print("\n=== Agent Step ===")
     print("Creating agent 'FileSummaryAgent2' as draft...")
     agent_id = str(uuid4())
     agent = Agent(
@@ -95,14 +95,14 @@ def main():
                 sampling=Sampling(temperature=0.8)
             ),
             models=ModelList(models=[
-                Model(name="vertex_ai/gemini-1.5-pro")
+                Model(name="openai/gpt-4o")
             ])
         ),
         is_draft=True,
         revision=1,
         status="pending"
     )
-    create_agent_result = lab_manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    create_agent_result = lab_manager.create_agent(agent=agent, automatic_publish=False)
     if isinstance(create_agent_result, Agent):
         print(f"Success: Created Agent: {create_agent_result.name}, ID: {create_agent_result.id}")
         created_entities["agent_id"] = create_agent_result.id
@@ -112,7 +112,7 @@ def main():
         exit()
 
     print("Publishing agent revision '1'...")
-    publish_agent_result = lab_manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"], revision="1")
+    publish_agent_result = lab_manager.publish_agent_revision(agent_id=created_entities["agent_id"], revision="1")
     if isinstance(publish_agent_result, Agent):
         print(f"Success: Published Agent Revision: {publish_agent_result.name}")
     else:
@@ -120,8 +120,8 @@ def main():
         rollback()
         exit()
 
-    # Chat Completion Flow
-    print("\n=== Chat Completion Flow ===")
+    # Chat Completion Step
+    print("\n=== Chat Completion Step ===")
     print(f"Asking agent about the uploaded file '{file_name}'...")
     messages = ChatMessageList(messages=[
         ChatMessage(
@@ -154,4 +154,4 @@ if __name__ == "__main__":
         rollback()
     except Exception as e:
         rollback()
-        print(f"\n# Critical error: {e}")
+        print(f"\n# Critical error: {e}")

pygeai/tests/snippets/lab/use_cases/update_web_reader.py

@@ -185,7 +185,6 @@ WEB_READING_GUIDE:
 # Update the agent
 manager = AILabManager()
 result = manager.update_agent(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     agent=agent,
     automatic_publish=True
 )
@@ -193,4 +192,4 @@ result = manager.update_agent(
 if isinstance(result, Agent):
     print(f"Agent updated successfully: {agent.to_dict()}")
 else:
-    print("Errors:", result.errors if hasattr(result, 'errors') else "Unknown error occurred")
+    print("Errors:", result.errors if hasattr(result, 'errors') else "Unknown error occurred")