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

pygeai/core/files/responses.py

@@ -1,10 +1,11 @@
+from typing import Optional
 from pydantic.main import BaseModel
 
 
 class UploadFileResponse(BaseModel):
-    id: str
-    url: str
-    success: bool
+    id: Optional[str] = None
+    url: Optional[str] = None
+    success: Optional[bool] = None
 
     def to_dict(self):
         return {

pygeai/lab/clients.py

@@ -8,12 +8,12 @@ class AILabClient(BaseClient):
 
     def __init__(self, api_key: str = None, base_url: str = None, alias: str = None, project_id: str = None):
         super().__init__(api_key, base_url, alias)
-        self.project_id = project_id if project_id else self.__get_project_id()
+        self.project_id = project_id if project_id else self.__get_project_id(api_key, base_url, alias)
 
-    def __get_project_id(self):
+    def __get_project_id(self, api_key: str = None, base_url: str = None, alias: str = None):
         response = None
         try:
-            response = AdminClient().validate_api_token()
+            response = AdminClient(api_key=api_key, base_url=base_url, alias=alias).validate_api_token()
             return response.get("projectId")
         except Exception as e:
             logger.error(f"Error retrieving project_id from GEAI. Response: {response}: {e}")

pygeai/lab/tools/clients.py

@@ -261,8 +261,8 @@ class ToolClient(AILabClient):
             raise APIResponseError(f"API returned an error: {response.text}")
 
         if response.status_code != 204:
-            logger.error(f"Unable to delete tool {tool_id or tool_name} from project {self.project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to delete tool {tool_id or tool_name} from project {self.project_id}: {response.text}")
+            logger.error(f"Unable to delete tool {tool_id or tool_name} in project {self.project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
+            raise InvalidAPIResponseException(f"Unable to delete tool {tool_id or tool_name} in project {self.project_id}: {response.text}")
         else:
             return {}
 
@@ -522,8 +522,8 @@ class ToolClient(AILabClient):
             data=data
         )
         if response.status_code != 204:
-            logger.error(f"Unable to set parameter for tool {tool_id or tool_public_name} from project {self.project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to set parameter for tool {tool_id or tool_public_name} from project {self.project_id}: {response.text}")
+            logger.error(f"Unable to set parameters for tool {tool_id or tool_public_name} in project {self.project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
+            raise InvalidAPIResponseException(f"Unable to set parameters for tool {tool_id or tool_public_name} in project {self.project_id}: {response.text}")
         else:
             return {}
 

pygeai/tests/admin/test_clients.py

@@ -0,0 +1,143 @@
+import unittest
+from unittest.mock import patch, MagicMock
+from json import JSONDecodeError
+
+from pygeai.admin.clients import AdminClient
+from pygeai.core.common.exceptions import InvalidAPIResponseException
+
+
+class TestAdminClient(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.admin.test_clients.TestAdminClient
+    """
+
+    def setUp(self):
+        self.client = AdminClient()
+        self.mock_response = MagicMock()
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_validate_api_token_success(self, mock_get):
+        self.mock_response.json.return_value = {"organizationId": "org-123", "projectId": "proj-123"}
+        mock_get.return_value = self.mock_response
+        
+        result = self.client.validate_api_token()
+        
+        mock_get.assert_called_once()
+        self.assertEqual(result, {"organizationId": "org-123", "projectId": "proj-123"})
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_validate_api_token_json_decode_error(self, mock_get):
+        self.mock_response.json.side_effect = JSONDecodeError("error", "doc", 0)
+        self.mock_response.status_code = 500
+        self.mock_response.text = "Invalid response"
+        mock_get.return_value = self.mock_response
+        
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            self.client.validate_api_token()
+        self.assertIn("Unable to validate API token", str(context.exception))
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_authorized_organizations_success(self, mock_get):
+        self.mock_response.json.return_value = {"organizations": ["org1", "org2"]}
+        mock_get.return_value = self.mock_response
+        
+        result = self.client.get_authorized_organizations()
+        
+        mock_get.assert_called_once()
+        self.assertEqual(result, {"organizations": ["org1", "org2"]})
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_authorized_organizations_json_decode_error(self, mock_get):
+        self.mock_response.json.side_effect = JSONDecodeError("error", "doc", 0)
+        self.mock_response.status_code = 500
+        self.mock_response.text = "Invalid response"
+        mock_get.return_value = self.mock_response
+        
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            self.client.get_authorized_organizations()
+        self.assertIn("Unable to retrieve authorized organizations", str(context.exception))
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_authorized_projects_by_organization_success(self, mock_get):
+        self.mock_response.json.return_value = {"projects": ["proj1", "proj2"]}
+        mock_get.return_value = self.mock_response
+        
+        result = self.client.get_authorized_projects_by_organization("org-123")
+        
+        mock_get.assert_called_once()
+        call_args = mock_get.call_args
+        self.assertEqual(call_args[1]['params']['organization'], "org-123")
+        self.assertEqual(result, {"projects": ["proj1", "proj2"]})
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_authorized_projects_by_organization_json_decode_error(self, mock_get):
+        self.mock_response.json.side_effect = JSONDecodeError("error", "doc", 0)
+        self.mock_response.status_code = 500
+        self.mock_response.text = "Invalid response"
+        mock_get.return_value = self.mock_response
+        
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            self.client.get_authorized_projects_by_organization("org-123")
+        self.assertIn("Unable to retrieve authorized projects for organization", str(context.exception))
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_project_visibility_success(self, mock_get):
+        self.mock_response.json.return_value = {}
+        mock_get.return_value = self.mock_response
+        
+        result = self.client.get_project_visibility(
+            organization="org-123",
+            project="proj-456",
+            access_token="token-789"
+        )
+        
+        mock_get.assert_called_once()
+        call_args = mock_get.call_args
+        self.assertEqual(call_args[1]['params']['organization'], "org-123")
+        self.assertEqual(call_args[1]['params']['project'], "proj-456")
+        self.assertEqual(call_args[1]['params']['accessToken'], "token-789")
+        self.assertEqual(result, {})
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_project_visibility_json_decode_error(self, mock_get):
+        self.mock_response.json.side_effect = JSONDecodeError("error", "doc", 0)
+        self.mock_response.status_code = 403
+        self.mock_response.text = "Forbidden"
+        mock_get.return_value = self.mock_response
+        
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            self.client.get_project_visibility("org-123", "proj-456", "token-789")
+        self.assertIn("Unable to retrieve project visibility", str(context.exception))
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_project_api_token_success(self, mock_get):
+        self.mock_response.json.return_value = {"apiToken": "api-token-123"}
+        mock_get.return_value = self.mock_response
+        
+        result = self.client.get_project_api_token(
+            organization="org-123",
+            project="proj-456",
+            access_token="token-789"
+        )
+        
+        mock_get.assert_called_once()
+        call_args = mock_get.call_args
+        self.assertEqual(call_args[1]['params']['organization'], "org-123")
+        self.assertEqual(call_args[1]['params']['project'], "proj-456")
+        self.assertEqual(call_args[1]['params']['accessToken'], "token-789")
+        self.assertEqual(result, {"apiToken": "api-token-123"})
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_project_api_token_json_decode_error(self, mock_get):
+        self.mock_response.json.side_effect = JSONDecodeError("error", "doc", 0)
+        self.mock_response.status_code = 401
+        self.mock_response.text = "Unauthorized"
+        mock_get.return_value = self.mock_response
+        
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            self.client.get_project_api_token("org-123", "proj-456", "token-789")
+        self.assertIn("Unable to retrieve project API token", str(context.exception))
+
+
+if __name__ == '__main__':
+    unittest.main()

pygeai/tests/auth/test_clients.py

@@ -0,0 +1,105 @@
+import unittest
+from unittest.mock import patch, MagicMock
+from json import JSONDecodeError
+
+from pygeai.auth.clients import AuthClient
+from pygeai.core.common.exceptions import InvalidAPIResponseException
+
+
+class TestAuthClient(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.auth.test_clients.TestAuthClient
+    """
+
+    def setUp(self):
+        self.client = AuthClient()
+        self.mock_response = MagicMock()
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_oauth2_access_token_success(self, mock_get):
+        self.mock_response.json.return_value = {"access_token": "token-123", "token_type": "Bearer"}
+        mock_get.return_value = self.mock_response
+        
+        result = self.client.get_oauth2_access_token(
+            client_id="client-123",
+            username="user@example.com",
+            password="password123"
+        )
+        
+        mock_get.assert_called_once()
+        call_args = mock_get.call_args
+        self.assertEqual(call_args[1]['params']['client_id'], "client-123")
+        self.assertEqual(call_args[1]['params']['username'], "user@example.com")
+        self.assertEqual(call_args[1]['params']['password'], "password123")
+        self.assertEqual(call_args[1]['params']['scope'], "gam_user_data gam_user_roles")
+        self.assertEqual(result, {"access_token": "token-123", "token_type": "Bearer"})
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_oauth2_access_token_custom_scope(self, mock_get):
+        self.mock_response.json.return_value = {"access_token": "token-123"}
+        mock_get.return_value = self.mock_response
+        
+        result = self.client.get_oauth2_access_token(
+            client_id="client-123",
+            username="user@example.com",
+            password="password123",
+            scope="custom_scope"
+        )
+        
+        call_args = mock_get.call_args
+        self.assertEqual(call_args[1]['params']['scope'], "custom_scope")
+
+    @patch('pygeai.core.services.rest.ApiService.get')
+    def test_get_oauth2_access_token_json_decode_error(self, mock_get):
+        self.mock_response.json.side_effect = JSONDecodeError("error", "doc", 0)
+        self.mock_response.status_code = 401
+        self.mock_response.text = "Invalid credentials"
+        mock_get.return_value = self.mock_response
+        
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            self.client.get_oauth2_access_token(
+                client_id="client-123",
+                username="user@example.com",
+                password="wrong"
+            )
+        self.assertIn("Unable to  obtain Oauth2 access token", str(context.exception))
+
+    @patch.object(AuthClient, 'api_service', create=True)
+    def test_get_user_profile_information_success(self, mock_api_service):
+        self.mock_response.json.return_value = {
+            "user_id": "user-123",
+            "email": "user@example.com",
+            "name": "Test User"
+        }
+        mock_api_service.get.return_value = self.mock_response
+        mock_api_service.token = None
+        
+        client = AuthClient()
+        client.api_service = mock_api_service
+        result = client.get_user_profile_information("access-token-123")
+        
+        mock_api_service.get.assert_called_once()
+        self.assertEqual(result, {
+            "user_id": "user-123",
+            "email": "user@example.com",
+            "name": "Test User"
+        })
+
+    @patch.object(AuthClient, 'api_service', create=True)
+    def test_get_user_profile_information_json_decode_error(self, mock_api_service):
+        self.mock_response.json.side_effect = JSONDecodeError("error", "doc", 0)
+        self.mock_response.status_code = 401
+        self.mock_response.text = "Invalid token"
+        mock_api_service.get.return_value = self.mock_response
+        mock_api_service.token = None
+        
+        client = AuthClient()
+        client.api_service = mock_api_service
+        
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            client.get_user_profile_information("invalid-token")
+        self.assertIn("Unable to retrieve user profile information", str(context.exception))
+
+
+if __name__ == '__main__':
+    unittest.main()

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

@@ -61,11 +61,17 @@ class TestAILab(unittest.TestCase):
             show_help()
             mock_stdout.assert_called_once()
 
-    def test_list_agents_missing_project_id(self):
+    def test_list_agents_auto_fetch_project_id(self):
         option_list = []
-        with self.assertRaises(MissingRequirementException) as cm:
+        with patch('pygeai.cli.commands.lab.ai_lab.AgentClient') as mock_client, \
+             patch('pygeai.cli.commands.lab.ai_lab.Console.write_stdout') as mock_stdout:
+            mock_client_instance = MagicMock()
+            mock_client_instance.project_id = "auto_fetched_proj"
+            mock_client.return_value = mock_client_instance
+            mock_client_instance.list_agents.return_value = []
             list_agents(option_list)
-        self.assertEqual(str(cm.exception), "Project ID must be specified.")
+            mock_client.assert_called_once_with(project_id=None)
+            mock_stdout.assert_called_once_with("Agent list: \n[]")
 
     def test_list_agents_success(self):
         option_list = [(Option("project_id", ["--project-id"], "", True), "proj123")]
@@ -76,7 +82,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.list_agents.return_value = [{"id": "agent1"}]
             list_agents(option_list)
             mock_client_instance.list_agents.assert_called_once_with(
-                project_id="proj123", status="", start="", count="", access_scope="public",
+                status="", start="", count="", access_scope="public",
                 allow_drafts=True, allow_external=False
             )
             mock_stdout.assert_called_once_with("Agent list: \n[{'id': 'agent1'}]")
@@ -126,7 +132,7 @@ class TestAILab(unittest.TestCase):
         option_list = []
         with self.assertRaises(MissingRequirementException) as cm:
             get_agent(option_list)
-        self.assertEqual(str(cm.exception), "Project ID and Agent id must be specified.")
+        self.assertEqual(str(cm.exception), "Agent ID must be specified.")
 
     def test_get_agent_success(self):
         option_list = [
@@ -140,7 +146,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_agent.return_value = {"id": "agent1"}
             get_agent(option_list)
             mock_client_instance.get_agent.assert_called_once_with(
-                project_id="proj123", agent_id="agent1", revision=0, version=0, allow_drafts=True
+                agent_id="agent1", revision=0, version=0, allow_drafts=True
             )
             mock_stdout.assert_called_once_with("Agent detail: \n{'id': 'agent1'}")
 
@@ -156,7 +162,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.create_sharing_link.return_value = {"token": "share123"}
             create_sharing_link(option_list)
             mock_client_instance.create_sharing_link.assert_called_once_with(
-                project_id="proj123", agent_id="agent1"
+                agent_id="agent1"
             )
             mock_stdout.assert_called_once_with("Sharing token: \n{'token': 'share123'}")
 
@@ -173,7 +179,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.publish_agent_revision.return_value = {"status": "published"}
             publish_agent_revision(option_list)
             mock_client_instance.publish_agent_revision.assert_called_once_with(
-                project_id="proj123", agent_id="agent1", revision="1"
+                agent_id="agent1", revision="1"
             )
             mock_stdout.assert_called_once_with("Published revision detail: \n{'status': 'published'}")
 
@@ -189,7 +195,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.delete_agent.return_value = {"status": "deleted"}
             delete_agent(option_list)
             mock_client_instance.delete_agent.assert_called_once_with(
-                project_id="proj123", agent_id="agent1"
+                agent_id="agent1"
             )
             mock_stdout.assert_called_once_with("Deleted agent detail: \n{'status': 'deleted'}")
 
@@ -254,7 +260,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.list_tools.return_value = [{"id": "tool1"}]
             list_tools(option_list)
             mock_client_instance.list_tools.assert_called_once_with(
-                project_id="proj123", id="", count="100", access_scope="public",
+                id="", count="100", access_scope="public",
                 allow_drafts=True, scope="api", allow_external=True
             )
             mock_stdout.assert_called_once_with("Tool list: \n[{'id': 'tool1'}]")
@@ -271,7 +277,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_tool.return_value = {"id": "tool1"}
             get_tool(option_list)
             mock_client_instance.get_tool.assert_called_once_with(
-                project_id="proj123", tool_id="tool1", revision=0, version=0, allow_drafts=True
+                tool_id="tool1", revision=0, version=0, allow_drafts=True
             )
             mock_stdout.assert_called_once_with("Tool detail: \n{'id': 'tool1'}")
 
@@ -287,7 +293,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.delete_tool.return_value = {"status": "deleted"}
             delete_tool(option_list)
             mock_client_instance.delete_tool.assert_called_once_with(
-                project_id="proj123", tool_id="tool1", tool_name=None
+                tool_id="tool1", tool_name=None
             )
             mock_stdout.assert_called_once_with("Deleted tool detail: \n{'status': 'deleted'}")
 
@@ -320,7 +326,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.publish_tool_revision.return_value = {"status": "published"}
             publish_tool_revision(option_list)
             mock_client_instance.publish_tool_revision.assert_called_once_with(
-                project_id="proj123", tool_id="tool1", revision="1"
+                tool_id="tool1", revision="1"
             )
             mock_stdout.assert_called_once_with("Published revision detail: \n{'status': 'published'}")
 
@@ -336,7 +342,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_parameter.return_value = {"key": "param1"}
             get_parameter(option_list)
             mock_client_instance.get_parameter.assert_called_once_with(
-                project_id="proj123", tool_id="tool1", tool_public_name=None, revision=0, version=0, allow_drafts=True
+                tool_id="tool1", tool_public_name=None, revision=0, version=0, allow_drafts=True
             )
             mock_stdout.assert_called_once_with("Parameter detail: \n{'key': 'param1'}")
 
@@ -354,7 +360,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.set_parameter.return_value = {"status": "set"}
             set_parameter(option_list)
             mock_client_instance.set_parameter.assert_called_once_with(
-                project_id="proj123", tool_id="tool1", tool_public_name=None, parameters=[{"key": "param1", "dataType": "String"}]
+                tool_id="tool1", tool_public_name=None, parameters=[{"key": "param1", "dataType": "String"}]
             )
             mock_stdout.assert_called_once_with("Set parameter detail: \n{'status': 'set'}")
 
@@ -419,7 +425,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_reasoning_strategy.return_value = {"id": "rs1"}
             get_reasoning_strategy(option_list)
             mock_client_instance.get_reasoning_strategy.assert_called_once_with(
-                project_id="proj123", reasoning_strategy_id="rs1", reasoning_strategy_name=None
+                reasoning_strategy_id="rs1", reasoning_strategy_name=None
             )
             mock_stdout.assert_called_once_with("Reasoning strategy detail: \n{'id': 'rs1'}")
 
@@ -466,7 +472,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_process.return_value = {"id": "proc1"}
             get_process(option_list)
             mock_client_instance.get_process.assert_called_once_with(
-                project_id="proj123", process_id="proc1", process_name=None, revision="0", version=0, allow_drafts=True
+                process_id="proc1", process_name=None, revision="0", version=0, allow_drafts=True
             )
             mock_stdout.assert_called_once_with("Process detail: \n{'id': 'proc1'}")
 
@@ -479,7 +485,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.list_processes.return_value = [{"id": "proc1"}]
             list_processes(option_list)
             mock_client_instance.list_processes.assert_called_once_with(
-                project_id="proj123", id=None, name=None, status=None, start="0", count="100", allow_draft=True
+                id=None, name=None, status=None, start="0", count="100", allow_draft=True
             )
             mock_stdout.assert_called_once_with("Process list: \n[{'id': 'proc1'}]")
 
@@ -495,7 +501,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.list_process_instances.return_value = [{"id": "inst1"}]
             list_processes_instances(option_list)
             mock_client_instance.list_process_instances.assert_called_once_with(
-                project_id="proj123", process_id="proc1", is_active=True, start="0", count="10"
+                process_id="proc1", is_active=True, start="0", count="10"
             )
             mock_stdout.assert_called_once_with("Process instances list: \n[{'id': 'inst1'}]")
 
@@ -511,7 +517,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.delete_process.return_value = {"status": "deleted"}
             delete_process(option_list)
             mock_client_instance.delete_process.assert_called_once_with(
-                project_id="proj123", process_id="proc1", process_name=None
+                process_id="proc1", process_name=None
             )
             mock_stdout.assert_called_once_with("Delete process result: \n{'status': 'deleted'}")
 
@@ -528,7 +534,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.publish_process_revision.return_value = {"status": "published"}
             publish_process_revision(option_list)
             mock_client_instance.publish_process_revision.assert_called_once_with(
-                project_id="proj123", process_id="proc1", process_name=None, revision="1"
+                process_id="proc1", process_name=None, revision="1"
             )
             mock_stdout.assert_called_once_with("Published process revision detail: \n{'status': 'published'}")
 
@@ -559,7 +565,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_task.return_value = {"id": "task1"}
             get_task(option_list)
             mock_client_instance.get_task.assert_called_once_with(
-                project_id="proj123", task_id="task1", task_name=None
+                task_id="task1", task_name=None
             )
             mock_stdout.assert_called_once_with("Task detail: \n{'id': 'task1'}")
 
@@ -572,7 +578,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.list_tasks.return_value = [{"id": "task1"}]
             list_tasks(option_list)
             mock_client_instance.list_tasks.assert_called_once_with(
-                project_id="proj123", id=None, start="0", count="100", allow_drafts=True
+                id=None, start="0", count="100", allow_drafts=True
             )
             mock_stdout.assert_called_once_with("Task list: \n[{'id': 'task1'}]")
 
@@ -603,7 +609,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.delete_task.return_value = {"status": "deleted"}
             delete_task(option_list)
             mock_client_instance.delete_task.assert_called_once_with(
-                project_id="proj123", task_id="task1", task_name=None
+                task_id="task1", task_name=None
             )
             mock_stdout.assert_called_once_with("Delete task result: \n{'status': 'deleted'}")
 
@@ -620,7 +626,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.publish_task_revision.return_value = {"status": "published"}
             publish_task_revision(option_list)
             mock_client_instance.publish_task_revision.assert_called_once_with(
-                project_id="proj123", task_id="task1", task_name=None, revision="1"
+                task_id="task1", task_name=None, revision="1"
             )
             mock_stdout.assert_called_once_with("Published task revision detail: \n{'status': 'published'}")
 
@@ -636,7 +642,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.start_instance.return_value = {"id": "inst1"}
             start_instance(option_list)
             mock_client_instance.start_instance.assert_called_once_with(
-                project_id="proj123", process_name="proc1", subject=None, variables=None
+                process_name="proc1", subject=None, variables=None
             )
             mock_stdout.assert_called_once_with("Started instance detail: \n{'id': 'inst1'}")
 
@@ -652,7 +658,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.abort_instance.return_value = {"status": "aborted"}
             abort_instance(option_list)
             mock_client_instance.abort_instance.assert_called_once_with(
-                project_id="proj123", instance_id="inst1"
+                instance_id="inst1"
             )
             mock_stdout.assert_called_once_with("Abort instance result: \n{'status': 'aborted'}")
 
@@ -668,7 +674,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_instance.return_value = {"id": "inst1"}
             get_instance(option_list)
             mock_client_instance.get_instance.assert_called_once_with(
-                project_id="proj123", instance_id="inst1"
+                instance_id="inst1"
             )
             mock_stdout.assert_called_once_with("Instance detail: \n{'id': 'inst1'}")
 
@@ -684,7 +690,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_thread_information.return_value = {"id": "thread1"}
             get_thread_information(option_list)
             mock_client_instance.get_thread_information.assert_called_once_with(
-                project_id="proj123", thread_id="thread1"
+                thread_id="thread1"
             )
             mock_stdout.assert_called_once_with("Thread information: \n{'id': 'thread1'}")
 
@@ -701,7 +707,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.send_user_signal.return_value = {"status": "sent"}
             send_user_signal(option_list)
             mock_client_instance.send_user_signal.assert_called_once_with(
-                project_id="proj123", instance_id="inst1", signal_name="signal1"
+                instance_id="inst1", signal_name="signal1"
             )
             mock_stdout.assert_called_once_with("Send user signal result: \n{'status': 'sent'}")
 
@@ -717,7 +723,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.create_kb.return_value = {"id": "kb1"}
             create_kb(option_list)
             mock_client_instance.create_kb.assert_called_once_with(
-                project_id="proj123", name="kb1", artifacts=None, metadata=None
+                name="kb1", artifacts=None, metadata=None
             )
             mock_stdout.assert_called_once_with("Created knowledge base detail: \n{'id': 'kb1'}")
 
@@ -733,7 +739,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.get_kb.return_value = {"id": "kb1"}
             get_kb(option_list)
             mock_client_instance.get_kb.assert_called_once_with(
-                project_id="proj123", kb_id="kb1", kb_name=None
+                kb_id="kb1", kb_name=None
             )
             mock_stdout.assert_called_once_with("Knowledge base detail: \n{'id': 'kb1'}")
 
@@ -746,7 +752,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.list_kbs.return_value = [{"id": "kb1"}]
             list_kbs(option_list)
             mock_client_instance.list_kbs.assert_called_once_with(
-                project_id="proj123", name=None, start="0", count="100"
+                name=None, start="0", count="100"
             )
             mock_stdout.assert_called_once_with("Knowledge base list: \n[{'id': 'kb1'}]")
 
@@ -762,7 +768,7 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.delete_kb.return_value = {"status": "deleted"}
             delete_kb(option_list)
             mock_client_instance.delete_kb.assert_called_once_with(
-                project_id="proj123", kb_id="kb1", kb_name=None
+                kb_id="kb1", kb_name=None
             )
             mock_stdout.assert_called_once_with("Delete knowledge base result: \n{'status': 'deleted'}")
 
@@ -775,6 +781,6 @@ class TestAILab(unittest.TestCase):
             mock_client_instance.list_jobs.return_value = [{"id": "job1"}]
             list_jobs(option_list)
             mock_client_instance.list_jobs.assert_called_once_with(
-                project_id="proj123", start="0", count="100", topic=None, token=None, name=None
+                start="0", count="100", topic=None, token=None, name=None
             )
             mock_stdout.assert_called_once_with("Job list: \n[{'id': 'job1'}]")