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

pygeai/tests/integration/lab/processes/test_publish_process_revision.py

@@ -0,0 +1,232 @@
+from unittest import TestCase
+import uuid
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow
+from pygeai.core.common.exceptions import InvalidAPIResponseException, MissingRequirementException, APIError
+
+ai_lab_manager: AILabManager
+
+class TestAILabPublishProcessRevisionIntegration(TestCase):  
+
+    def setUp(self):
+        self.ai_lab_manager = AILabManager()
+        # Using a known process ID that should have revisions for testing
+        self.process_id = "7e28e9ab-b9a2-417e-9e87-ed7f6ec9534b"
+        self.process_name = "Test Process For Sdk Project"
+        self.revision = "1"
+        
+
+    def __create_process(self):
+        """
+        Helper to create a process for testing publish revision functionality.
+        """
+        unique_key = str(uuid.uuid4())
+        process = AgenticProcess(
+            key="test_publish_key",
+            name=f"Test Publish Process {unique_key[:8]}",
+            description="Test process for publish revision testing",
+            kb=KnowledgeBase(name="basic-sample", artifact_type_name=["sample-artifact"]),
+            agentic_activities=[
+                AgenticActivity(
+                    key="activityOne", 
+                    name="First Step", 
+                    task_name="basic-task", 
+                    agent_name="GoogleSummarizer2", 
+                    agent_revision_id=0
+                )
+            ],
+            artifact_signals=[
+                ArtifactSignal(
+                    key="artifact.upload.1", 
+                    name="artifact.upload", 
+                    handling_type="C", 
+                    artifact_type_name=["sample-artifact"]
+                )
+            ],
+            user_signals=[
+                UserSignal(key="signal_done", name="process-completed")
+            ],
+            start_event=Event(key="artifact.upload.1", name="artifact.upload"),
+            end_event=Event(key="end", name="Done"),
+            sequence_flows=[
+                SequenceFlow(key="step1", source_key="artifact.upload.1", target_key="activityOne"),
+                SequenceFlow(key="step2", source_key="activityOne", target_key="signal_done"),
+                SequenceFlow(key="stepEnd", source_key="signal_done", target_key="end")
+            ]
+        )
+
+        return self.ai_lab_manager.create_process(
+            process=process,
+            automatic_publish=False
+        )
+
+
+    def __publish_process_revision(self, process_id: str = None, process_name: str = None, revision: str = None):
+        return self.ai_lab_manager.publish_process_revision(
+            process_id=process_id,
+            process_name=process_name,
+            revision=revision if revision is not None else self.revision
+        )
+    
+
+    def test_publish_process_revision_by_id(self):
+        """Test publishing a process revision using process_id."""
+        published_process = self.__publish_process_revision(process_id=self.process_id)
+
+        self.assertIsInstance(published_process, AgenticProcess, "Expected AgenticProcess instance")
+        self.assertEqual(published_process.id, self.process_id, "Expected process ID to match")
+        self.assertIsNotNone(published_process.revision, "Expected revision to be set")
+
+
+    def test_publish_process_revision_by_name(self):
+        """Test publishing a process revision using process_name."""
+        published_process = self.__publish_process_revision(process_name=self.process_name)
+
+        self.assertIsInstance(published_process, AgenticProcess, "Expected AgenticProcess instance")
+        self.assertEqual(published_process.name, self.process_name, "Expected process name to match")
+        self.assertIsNotNone(published_process.revision, "Expected revision to be set")
+
+
+    def test_publish_process_revision_specific_revision(self):
+        """Test publishing a specific revision number."""
+        specific_revision = "2"
+        published_process = self.__publish_process_revision(
+            process_id=self.process_id, 
+            revision=specific_revision
+        )
+
+        self.assertIsInstance(published_process, AgenticProcess, "Expected AgenticProcess instance")
+        self.assertEqual(published_process.id, self.process_id, "Expected process ID to match")
+
+
+    def test_publish_process_revision_no_parameters(self):
+        """Test error when neither process_id nor process_name is provided."""
+        with self.assertRaises(MissingRequirementException) as exception:
+            self.__publish_process_revision()
+
+        self.assertIn(
+            "Either process_id or process_name and revision must be provided",
+            str(exception.exception),
+            "Expected error message for missing parameters"
+        )
+
+
+    def test_publish_process_revision_empty_process_id(self):
+        """Test error when process_id is empty."""
+        with self.assertRaises(MissingRequirementException) as exception:
+            self.__publish_process_revision(process_id="")
+
+        self.assertIn(
+            "Either process_id or process_name and revision must be provided",
+            str(exception.exception),
+            "Expected error message for empty process_id"
+        )
+
+
+    def test_publish_process_revision_empty_process_name(self):
+        """Test error when process_name is empty."""
+        with self.assertRaises(MissingRequirementException) as exception:
+            self.__publish_process_revision(process_name="")
+
+        self.assertIn(
+            "Either process_id or process_name and revision must be provided",
+            str(exception.exception),
+            "Expected error message for empty process_name"
+        )
+
+
+    def test_publish_process_revision_no_revision(self):
+        """Test error when revision is not provided."""
+        with self.assertRaises(MissingRequirementException) as exception:
+            self.__publish_process_revision(process_id=self.process_id, revision="")
+
+        self.assertIn(
+            "Either process_id or process_name and revision must be provided",
+            str(exception.exception),
+            "Expected error message for missing revision"
+        )
+
+
+    def test_publish_process_revision_none_revision(self):
+        """Test error when revision is None."""
+        with self.assertRaises(MissingRequirementException) as exception:
+            self.__publish_process_revision(process_id=self.process_id, revision=None)
+
+        self.assertIn(
+            "Either process_id or process_name and revision must be provided",
+            str(exception.exception),
+            "Expected error message for None revision"
+        )
+
+
+    def test_publish_process_revision_invalid_process_id(self):
+        """Test error with invalid process_id."""
+        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
+        with self.assertRaises((APIError, InvalidAPIResponseException)) as exception:
+            self.__publish_process_revision(process_id=invalid_id)
+
+        self.assertTrue(
+            f"Unable to publish revision {self.revision} for process {invalid_id}" in str(exception.exception) or
+            "Error received while publishing process revision" in str(exception.exception),
+            "Expected error message for invalid process id"
+        )
+
+
+    def test_publish_process_revision_invalid_process_name(self):
+        """Test error with invalid process_name."""
+        invalid_name = "NonExistentProcessName123"
+        with self.assertRaises((APIError, InvalidAPIResponseException)) as exception:
+            self.__publish_process_revision(process_name=invalid_name)
+
+        self.assertTrue(
+            f"Unable to publish revision {self.revision} for process {invalid_name}" in str(exception.exception) or
+            "Error received while publishing process revision" in str(exception.exception),
+            "Expected error message for invalid process name"
+        )
+
+
+    def test_publish_process_revision_nonexistent_process_id(self):
+        """Test error with nonexistent process_id."""
+        nonexistent_id = "99999999-9999-9999-9999-999999999999"
+        with self.assertRaises((APIError, InvalidAPIResponseException)) as exception:
+            self.__publish_process_revision(process_id=nonexistent_id)
+
+        self.assertTrue(
+            f"Unable to publish revision {self.revision} for process {nonexistent_id}" in str(exception.exception) or
+            "Error received while publishing process revision" in str(exception.exception),
+            "Expected error message for nonexistent process id"
+        )
+
+
+    def test_publish_process_revision_invalid_revision(self):
+        """Test error with invalid revision number."""
+        invalid_revision = "999"
+        with self.assertRaises((APIError, InvalidAPIResponseException)) as exception:
+            self.__publish_process_revision(process_id=self.process_id, revision=invalid_revision)
+
+        self.assertTrue(
+            f"Unable to publish revision {invalid_revision} for process {self.process_id}" in str(exception.exception) or
+            "Error received while publishing process revision" in str(exception.exception),
+            "Expected error message for invalid revision"
+        )
+
+
+    def test_publish_process_revision_with_created_process(self):
+        """Test publishing revision with a newly created process."""
+        # Create a fresh process
+        created_process = self.__create_process()
+        
+        # Note: A newly created process might not have additional revisions to publish
+        # This test verifies the process creation worked and we can attempt to publish
+        self.assertIsNotNone(created_process.id, "Expected created process to have an ID")
+        
+        # Attempt to publish revision 0 (initial revision)
+        try:
+            published_process = self.__publish_process_revision(
+                process_id=created_process.id, 
+                revision="0"
+            )
+            self.assertIsInstance(published_process, AgenticProcess, "Expected AgenticProcess instance")
+        except (APIError, InvalidAPIResponseException):
+            # This is expected for a new process that may not have publishable revisions yet
+            pass

pygeai/tests/integration/lab/processes/test_update_process.py

@@ -15,7 +15,7 @@ class TestAILabUpdateProcessIntegration(TestCase):
         """
         Set up the test environment.
         """
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.process_to_update = self.__load_process()
         self.created_process: AgenticProcess = None
         self.updated_process: AgenticProcess = None

pygeai/tests/integration/lab/reasoning_strategies/test_get_reasoning_strategy.py

@@ -8,7 +8,7 @@ ai_lab_manager: AILabManager
 class TestAILabGetReasoningStrategyIntegration(TestCase):   
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.strategy_id = "0a3b039e-25bd-4cf9-ad67-7af2b654874b"
         self.strategy_name = "Chain of Thought"
 

pygeai/tests/integration/lab/reasoning_strategies/test_list_reasoning_strategies.py

@@ -8,7 +8,7 @@ ai_lab_manager: AILabManager
 class TestAILabListReasoningStrategiesIntegration(TestCase):    
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.filter_settings = FilterSettings(
             name="",
             start=0,

pygeai/tests/integration/lab/reasoning_strategies/test_update_reasoning_strategy.py

@@ -11,7 +11,7 @@ class TestAILabCreateReasoningStrategyIntegration(TestCase):
         """
         Set up the test environment.
         """
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.strategy_to_update = self.__load_strategy()
 
     

pygeai/tests/integration/lab/tools/test_create_tool.py

@@ -10,7 +10,7 @@ class TestAILabCreateToolIntegration(TestCase):
         """
         Set up the test environment.
         """
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.new_tool = self.__load_tool()
         self.created_tool: Tool = None
 

pygeai/tests/integration/lab/tools/test_delete_tool.py

@@ -9,7 +9,7 @@ ai_lab_manager: AILabManager
 class TestAILabDeleteToolIntegration(TestCase):  
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         
 
     def __create_tool(self):

pygeai/tests/integration/lab/tools/test_get_parameter.py

@@ -9,7 +9,7 @@ ai_lab_manager: AILabManager
 class TestAILabGetParameterIntegration(TestCase):   
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "42610b61-cb47-49bf-9475-36ee652f31f3"
         self.filter_settings = FilterSettings(
             revision="0",

pygeai/tests/integration/lab/tools/test_get_tool.py

@@ -10,7 +10,7 @@ ai_lab_manager: AILabManager
 class TestAILabGetToolIntegration(TestCase):   
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "e3e4d64f-ce52-467e-90a9-aa4d08425e82"
         self.filter_settings = FilterSettings(
             revision="0",

pygeai/tests/integration/lab/tools/test_list_tools.py

@@ -8,7 +8,7 @@ ai_lab_manager: AILabManager
 class TestAILabListToolsIntegration(TestCase):    
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.filter_settings = FilterSettings(
             allow_external=False,
             allow_drafts=True,

pygeai/tests/integration/lab/tools/test_publish_tool_revision.py

@@ -9,7 +9,7 @@ ai_lab_manager: AILabManager
 class TestAILabPublishToolRevisionIntegration(TestCase):   
 
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "c77e1f2e-0322-4dd0-b6ec-aff217f1cb32"
 
 

pygeai/tests/integration/lab/tools/test_set_parameter.py

@@ -8,7 +8,7 @@ ai_lab_manager: AILabManager
 
 class TestAILabSetParameterIntegration(TestCase):
     def setUp(self):
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_id = "42610b61-cb47-49bf-9475-36ee652f31f3"
         self.tool_parameters = self.__load_parameters()
 

pygeai/tests/integration/lab/tools/test_update_tool.py

@@ -10,7 +10,7 @@ class TestAILabUpdateToolIntegration(TestCase):
         """
         Set up the test environment.
         """
-        self.ai_lab_manager = AILabManager(alias="beta")
+        self.ai_lab_manager = AILabManager()
         self.tool_to_update = self.__load_tool()
 
     

pygeai/tests/lab/agents/test_clients.py

@@ -15,11 +15,12 @@ class TestAgentClient(unittest.TestCase):
     """
 
     def setUp(self):
-        self.client = AgentClient()
         self.project_id = "test-project-id"
+        self.client = AgentClient(api_key="test_key", base_url="https://test.url", project_id=self.project_id)
         self.agent_id = "test-agent-id"
         self.agent_data_prompt = {"instructions": "Do this task"}
         self.agent_data_llm_config = {"maxTokens": 100, "timeout": 30}
+        self.agent_data_strategy_name = "default"
         self.agent_data_models = [{"name": "gpt-4o"}]
         self.agent_data_resource_pools = [{"name": "pool1", "tools": [{"name": "tool1"}]}]
 
@@ -30,7 +31,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.list_agents(
-            project_id=self.project_id,
             status="active",
             start=0,
             count=10,
@@ -63,7 +63,7 @@ class TestAgentClient(unittest.TestCase):
         mock_response.text = "Invalid JSON response"
 
         with self.assertRaises(InvalidAPIResponseException) as context:
-            self.client.list_agents(project_id=self.project_id)
+            self.client.list_agents()
 
         self.assertEqual(str(context.exception), f"Unable to list agents for project {self.project_id}: Invalid JSON response")
         mock_get.assert_called_once()
@@ -75,7 +75,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.create_agent(
-            project_id=self.project_id,
             name="Test Agent",
             access_scope="private",
             public_name="test-agent",
@@ -84,6 +83,7 @@ class TestAgentClient(unittest.TestCase):
             description="Agent Description",
             agent_data_prompt=self.agent_data_prompt,
             agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
             agent_data_models=self.agent_data_models,
             agent_data_resource_pools=self.agent_data_resource_pools,
             automatic_publish=True
@@ -116,7 +116,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.create_agent(
-            project_id=self.project_id,
             name="Test Agent",
             access_scope="private",
             public_name="test-agent",
@@ -125,6 +124,7 @@ class TestAgentClient(unittest.TestCase):
             description="Agent Description",
             agent_data_prompt=self.agent_data_prompt,
             agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
             agent_data_models=self.agent_data_models,
             automatic_publish=False
         )
@@ -137,7 +137,6 @@ class TestAgentClient(unittest.TestCase):
     def test_create_agent_invalid_access_scope(self):
         with self.assertRaises(ValueError) as context:
             self.client.create_agent(
-                project_id=self.project_id,
                 name="Test Agent",
                 access_scope="invalid_scope",
                 public_name="test-agent",
@@ -146,6 +145,7 @@ class TestAgentClient(unittest.TestCase):
                 description="Agent Description",
                 agent_data_prompt=self.agent_data_prompt,
                 agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
                 agent_data_models=self.agent_data_models
             )
         self.assertEqual(str(context.exception), f"Access scope must be one of {', '.join(VALID_ACCESS_SCOPES)}.")
@@ -159,7 +159,6 @@ class TestAgentClient(unittest.TestCase):
 
         with self.assertRaises(InvalidAPIResponseException) as context:
             self.client.create_agent(
-                project_id=self.project_id,
                 name="Test Agent",
                 access_scope="private",
                 public_name="test-agent",
@@ -168,6 +167,7 @@ class TestAgentClient(unittest.TestCase):
                 description="Agent Description",
                 agent_data_prompt=self.agent_data_prompt,
                 agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
                 agent_data_models=self.agent_data_models
             )
 
@@ -181,7 +181,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.get_agent(
-            project_id=self.project_id,
             agent_id=self.agent_id,
             revision="1",
             version=2,
@@ -201,14 +200,9 @@ class TestAgentClient(unittest.TestCase):
         headers = mock_get.call_args[1]['headers']
         self.assertEqual(headers['ProjectId'], self.project_id)
 
-    def test_get_agent_missing_project_id(self):
-        with self.assertRaises(MissingRequirementException) as context:
-            self.client.get_agent(project_id="", agent_id=self.agent_id)
-        self.assertEqual(str(context.exception), "Cannot retrieve agent without specifying project_id")
-
     def test_get_agent_missing_agent_id(self):
         with self.assertRaises(MissingRequirementException) as context:
-            self.client.get_agent(project_id=self.project_id, agent_id="")
+            self.client.get_agent(agent_id="")
         self.assertEqual(str(context.exception), "agent_id must be specified in order to retrieve the agent")
 
     @patch("pygeai.core.services.rest.ApiService.get")
@@ -219,7 +213,7 @@ class TestAgentClient(unittest.TestCase):
         mock_response.text = "Invalid JSON response"
 
         with self.assertRaises(InvalidAPIResponseException) as context:
-            self.client.get_agent(project_id=self.project_id, agent_id=self.agent_id)
+            self.client.get_agent(agent_id=self.agent_id)
 
         self.assertEqual(str(context.exception), f"Unable to retrieve agent {self.agent_id} for project {self.project_id}: Invalid JSON response")
         mock_get.assert_called_once()
@@ -231,7 +225,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.create_sharing_link(
-            project_id=self.project_id,
             agent_id=self.agent_id
         )
 
@@ -244,14 +237,9 @@ class TestAgentClient(unittest.TestCase):
         headers = mock_get.call_args[1]['headers']
         self.assertEqual(headers['ProjectId'], self.project_id)
 
-    def test_create_sharing_link_missing_project_id(self):
-        with self.assertRaises(MissingRequirementException) as context:
-            self.client.create_sharing_link(project_id="", agent_id=self.agent_id)
-        self.assertEqual(str(context.exception), "Cannot create sharing link without specifying project_id")
-
     def test_create_sharing_link_missing_agent_id(self):
         with self.assertRaises(MissingRequirementException) as context:
-            self.client.create_sharing_link(project_id=self.project_id, agent_id="")
+            self.client.create_sharing_link(agent_id="")
         self.assertEqual(str(context.exception), "agent_id must be specified in order to create sharing link")
 
     @patch("pygeai.core.services.rest.ApiService.get")
@@ -263,7 +251,6 @@ class TestAgentClient(unittest.TestCase):
 
         with self.assertRaises(InvalidAPIResponseException) as context:
             self.client.create_sharing_link(
-                project_id=self.project_id,
                 agent_id=self.agent_id
             )
 
@@ -278,7 +265,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.publish_agent_revision(
-            project_id=self.project_id,
             agent_id=self.agent_id,
             revision=revision
         )
@@ -302,7 +288,6 @@ class TestAgentClient(unittest.TestCase):
 
         with self.assertRaises(InvalidAPIResponseException) as context:
             self.client.publish_agent_revision(
-                project_id=self.project_id,
                 agent_id=self.agent_id,
                 revision=revision
             )
@@ -312,12 +297,11 @@ class TestAgentClient(unittest.TestCase):
 
     @patch("pygeai.core.services.rest.ApiService.delete")
     def test_delete_agent_success(self, mock_delete):
-        expected_response = {"status": "deleted"}
+        expected_response = {}
         mock_response = mock_delete.return_value
-        mock_response.json.return_value = expected_response
+        mock_response.status_code = 204
 
         result = self.client.delete_agent(
-            project_id=self.project_id,
             agent_id=self.agent_id
         )
 
@@ -339,7 +323,6 @@ class TestAgentClient(unittest.TestCase):
 
         with self.assertRaises(InvalidAPIResponseException) as context:
             self.client.delete_agent(
-                project_id=self.project_id,
                 agent_id=self.agent_id
             )
 
@@ -353,7 +336,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.update_agent(
-            project_id=self.project_id,
             agent_id=self.agent_id,
             name="Updated Agent",
             access_scope="public",
@@ -363,6 +345,7 @@ class TestAgentClient(unittest.TestCase):
             description="Updated Description",
             agent_data_prompt=self.agent_data_prompt,
             agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
             agent_data_models=self.agent_data_models,
             agent_data_resource_pools=self.agent_data_resource_pools,
             automatic_publish=True,
@@ -396,7 +379,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.update_agent(
-            project_id=self.project_id,
             agent_id=self.agent_id,
             name="Upserted Agent",
             access_scope="private",
@@ -406,6 +388,7 @@ class TestAgentClient(unittest.TestCase):
             description="Upserted Description",
             agent_data_prompt=self.agent_data_prompt,
             agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
             agent_data_models=self.agent_data_models,
             automatic_publish=False,
             upsert=True
@@ -425,7 +408,6 @@ class TestAgentClient(unittest.TestCase):
         mock_response.json.return_value = expected_response
 
         result = self.client.update_agent(
-            project_id=self.project_id,
             agent_id=self.agent_id,
             name="Updated Agent No Pools",
             access_scope="private",
@@ -435,6 +417,7 @@ class TestAgentClient(unittest.TestCase):
             description="Updated Description",
             agent_data_prompt=self.agent_data_prompt,
             agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
             agent_data_models=self.agent_data_models,
             automatic_publish=False,
             upsert=False
@@ -448,7 +431,6 @@ class TestAgentClient(unittest.TestCase):
     def test_update_agent_invalid_access_scope(self):
         with self.assertRaises(ValueError) as context:
             self.client.update_agent(
-                project_id=self.project_id,
                 agent_id=self.agent_id,
                 name="Updated Agent",
                 access_scope="invalid_scope",
@@ -458,6 +440,7 @@ class TestAgentClient(unittest.TestCase):
                 description="Updated Description",
                 agent_data_prompt=self.agent_data_prompt,
                 agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
                 agent_data_models=self.agent_data_models
             )
         self.assertEqual(str(context.exception), f"Access scope must be one of {', '.join(VALID_ACCESS_SCOPES)}.")
@@ -471,7 +454,6 @@ class TestAgentClient(unittest.TestCase):
 
         with self.assertRaises(InvalidAPIResponseException) as context:
             self.client.update_agent(
-                project_id=self.project_id,
                 agent_id=self.agent_id,
                 name="Updated Agent",
                 access_scope="private",
@@ -481,6 +463,7 @@ class TestAgentClient(unittest.TestCase):
                 description="Updated Description",
                 agent_data_prompt=self.agent_data_prompt,
                 agent_data_llm_config=self.agent_data_llm_config,
+            agent_data_strategy_name=self.agent_data_strategy_name,
                 agent_data_models=self.agent_data_models
             )