pygeai 0.4.0b8__py3-none-any.whl → 0.4.0b9__py3-none-any.whl

pygeai/cli/commands/lab/ai_lab.py

@@ -9,11 +9,13 @@ from pygeai.cli.commands.lab.common import get_agent_data_prompt_inputs, get_age
 from pygeai.cli.texts.help import AI_LAB_HELP_TEXT
 from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError
 from pygeai.core.utils.console import Console
+from pygeai.cli.commands.lab.options import PROJECT_ID_OPTION
 from pygeai.lab.agents.clients import AgentClient
 from pygeai.lab.processes.clients import AgenticProcessClient
 from pygeai.lab.strategies.clients import ReasoningStrategyClient
 from pygeai.lab.tools.clients import ToolClient
 from pygeai.lab.constants import VALID_SCOPES
+from pygeai.tests.snippets.lab.crud_ui import project_id
 
 
 def show_help():
@@ -25,6 +27,7 @@ def show_help():
 
 
 def list_agents(option_list: list):
+    project_id = None
     status = ""
     start = ""
     count = ""
@@ -33,6 +36,8 @@ def list_agents(option_list: list):
     allow_external = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "status":
             status = option_arg
         if option_flag.name == "start":
@@ -46,7 +51,7 @@ def list_agents(option_list: list):
         if option_flag.name == "allow_external":
             allow_external = get_boolean_value(option_arg)
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.list_agents(
         status=status,
         start=start,
@@ -59,6 +64,7 @@ def list_agents(option_list: list):
 
 
 list_agents_options = [
+    PROJECT_ID_OPTION,
     Option(
         "status",
         ["--status"],
@@ -99,6 +105,7 @@ list_agents_options = [
 
 
 def create_agent(option_list: list):
+    project_id = None
     name = None
     access_scope = None
     public_name = None
@@ -120,6 +127,8 @@ def create_agent(option_list: list):
     automatic_publish = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "name":
             name = option_arg
         if option_flag.name == "access_scope":
@@ -231,7 +240,7 @@ def create_agent(option_list: list):
         {"name": agent_data_model_name}
     ]
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.create_agent(
         name=name,
         access_scope=access_scope,
@@ -250,6 +259,7 @@ def create_agent(option_list: list):
 
 
 create_agent_options = [
+    PROJECT_ID_OPTION,
     Option(
         "name",
         ["--name", "-n"],
@@ -370,12 +380,15 @@ create_agent_options = [
 
 
 def get_agent(option_list: list):
+    project_id = None
     agent_id = None
     revision = 0
     version = 0
     allow_drafts = True
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "agent_id":
             agent_id = option_arg
         if option_flag.name == "revision":
@@ -388,7 +401,7 @@ def get_agent(option_list: list):
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.get_agent(
         agent_id=agent_id,
         revision=revision,
@@ -399,6 +412,7 @@ def get_agent(option_list: list):
 
 
 get_agent_options = [
+    PROJECT_ID_OPTION,
     Option(
         "agent_id",
         ["--agent-id", "--aid"],
@@ -427,10 +441,13 @@ get_agent_options = [
 
 
 def export_agent(option_list: list):
+    project_id = None
     agent_id = None
     file = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "agent_id":
             agent_id = option_arg
         if option_flag.name == "file":
@@ -439,7 +456,7 @@ def export_agent(option_list: list):
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.export_agent(
         agent_id=agent_id,
     )
@@ -458,6 +475,7 @@ def export_agent(option_list: list):
 
 
 export_agent_options = [
+    PROJECT_ID_OPTION,
     Option(
         "agent_id",
         ["--agent-id", "--aid"],
@@ -474,9 +492,12 @@ export_agent_options = [
 
 
 def import_agent(option_list: list):
+    project_id = None
     file = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "file":
             file = option_arg
 
@@ -491,7 +512,7 @@ def import_agent(option_list: list):
         logger.error(f"File is not in JSON format: {e}")
         Console.write_stderr(f"File is not in JSON format.")
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.import_agent(
         data=agent_data
     )
@@ -501,6 +522,7 @@ def import_agent(option_list: list):
 
 
 import_agent_options = [
+    PROJECT_ID_OPTION,
     Option(
         "file",
         ["--file", "-f"],
@@ -511,16 +533,19 @@ import_agent_options = [
 
 
 def create_sharing_link(option_list: list):
+    project_id = None
     agent_id = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "agent_id":
             agent_id = option_arg
 
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.create_sharing_link(
         agent_id=agent_id,
     )
@@ -528,6 +553,7 @@ def create_sharing_link(option_list: list):
 
 
 create_sharing_link_options = [
+    PROJECT_ID_OPTION,
     Option(
         "agent_id",
         ["--agent-id", "--aid"],
@@ -538,10 +564,13 @@ create_sharing_link_options = [
 
 
 def publish_agent_revision(option_list: list):
+    project_id = None
     agent_id = None
     revision = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "agent_id":
             agent_id = option_arg
         if option_flag.name == "revision":
@@ -550,7 +579,7 @@ def publish_agent_revision(option_list: list):
     if not (agent_id and revision):
         raise MissingRequirementException("Agent ID and revision must be specified.")
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.publish_agent_revision(
         agent_id=agent_id,
         revision=revision
@@ -559,6 +588,7 @@ def publish_agent_revision(option_list: list):
 
 
 publish_agent_revision_options = [
+    PROJECT_ID_OPTION,
     Option(
         "agent_id",
         ["--agent-id", "--aid"],
@@ -575,16 +605,19 @@ publish_agent_revision_options = [
 
 
 def delete_agent(option_list: list):
+    project_id = None
     agent_id = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "agent_id":
             agent_id = option_arg
 
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.delete_agent(
         agent_id=agent_id,
     )
@@ -592,6 +625,7 @@ def delete_agent(option_list: list):
 
 
 delete_agent_options = [
+    PROJECT_ID_OPTION,
     Option(
         "agent_id",
         ["--agent-id", "--aid"],
@@ -602,6 +636,7 @@ delete_agent_options = [
 
 
 def update_agent(option_list: list):
+    project_id = None
     agent_id = None
     name = None
     access_scope = None
@@ -625,6 +660,8 @@ def update_agent(option_list: list):
     upsert = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "agent_id":
             agent_id = option_arg
         if option_flag.name == "name":
@@ -740,7 +777,7 @@ def update_agent(option_list: list):
         {"name": agent_data_model_name}
     ]
 
-    client = AgentClient()
+    client = AgentClient(project_id=project_id)
     result = client.update_agent(
         agent_id=agent_id,
         name=name,
@@ -761,6 +798,7 @@ def update_agent(option_list: list):
 
 
 update_agent_options = [
+    PROJECT_ID_OPTION,
     Option(
         "agent_id",
         ["--agent-id", "--aid"],
@@ -894,6 +932,7 @@ update_agent_options = [
 
 
 def create_tool(option_list: list):
+    project_id = None
     name = None
     description = None
     scope = None
@@ -907,6 +946,8 @@ def create_tool(option_list: list):
     automatic_publish = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "name":
             name = option_arg
         if option_flag.name == "description":
@@ -959,7 +1000,7 @@ def create_tool(option_list: list):
 
     tool_parameters = get_tool_parameters(parameters)
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.create_tool(
         name=name,
         description=description,
@@ -977,6 +1018,7 @@ def create_tool(option_list: list):
 
 
 create_tool_options = [
+    PROJECT_ID_OPTION,
     Option(
         "name",
         ["--name", "-n"],
@@ -1047,6 +1089,7 @@ create_tool_options = [
 
 
 def list_tools(option_list: list):
+    project_id = None
     id = ""
     count = "100"
     access_scope = "public"
@@ -1055,6 +1098,8 @@ def list_tools(option_list: list):
     allow_external = True
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "id":
             id = option_arg
         if option_flag.name == "count":
@@ -1071,7 +1116,7 @@ def list_tools(option_list: list):
     if scope and scope not in VALID_SCOPES:
         raise ValueError(f"Scope must be one of {', '.join(VALID_SCOPES)}.")
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.list_tools(
         id=id,
         count=count,
@@ -1084,6 +1129,7 @@ def list_tools(option_list: list):
 
 
 list_tools_options = [
+    PROJECT_ID_OPTION,
     Option(
         "id",
         ["--id"],
@@ -1124,12 +1170,15 @@ list_tools_options = [
 
 
 def get_tool(option_list: list):
+    project_id = None
     tool_id = None
     revision = 0
     version = 0
     allow_drafts = True
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "tool_id":
             tool_id = option_arg
         if option_flag.name == "revision":
@@ -1142,7 +1191,7 @@ def get_tool(option_list: list):
     if not tool_id:
         raise MissingRequirementException("Tool ID must be specified.")
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.get_tool(
         tool_id=tool_id,
         revision=revision,
@@ -1153,6 +1202,7 @@ def get_tool(option_list: list):
 
 
 get_tool_options = [
+    PROJECT_ID_OPTION,
     Option(
         "tool_id",
         ["--tool-id", "--tid"],
@@ -1181,10 +1231,13 @@ get_tool_options = [
 
 
 def export_tool(option_list: list):
+    project_id = None
     tool_id = None
     file = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "tool_id":
             tool_id = option_arg
         if option_flag.name == "file":
@@ -1194,7 +1247,7 @@ def export_tool(option_list: list):
     if not tool_id:
         raise MissingRequirementException("Tool ID must be specified.")
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.export_tool(
         tool_id=tool_id,
     )
@@ -1212,6 +1265,7 @@ def export_tool(option_list: list):
 
 
 export_tool_options = [
+    PROJECT_ID_OPTION,
     Option(
         "tool_id",
         ["--tool-id", "--tid"],
@@ -1229,10 +1283,13 @@ export_tool_options = [
 
 
 def delete_tool(option_list: list):
+    project_id = None
     tool_id = None
     tool_name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "tool_id":
             tool_id = option_arg
         elif option_flag.name == "tool_name":
@@ -1241,7 +1298,7 @@ def delete_tool(option_list: list):
     if not (tool_id or tool_name):
         raise MissingRequirementException("Either Tool ID or Tool Name must be specified.")
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.delete_tool(
         tool_id=tool_id,
         tool_name=tool_name
@@ -1250,6 +1307,7 @@ def delete_tool(option_list: list):
 
 
 delete_tool_options = [
+    PROJECT_ID_OPTION,
     Option(
         "tool_id",
         ["--tool-id", "--tid"],
@@ -1266,6 +1324,7 @@ delete_tool_options = [
 
 
 def update_tool(option_list: list):
+    project_id = None
     tool_id = None
     name = None
     description = None
@@ -1281,6 +1340,8 @@ def update_tool(option_list: list):
     upsert = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "tool_id":
             tool_id = option_arg
         if option_flag.name == "name":
@@ -1337,7 +1398,7 @@ def update_tool(option_list: list):
 
     tool_parameters = get_tool_parameters(parameters)
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.update_tool(
         tool_id=tool_id,
         name=name,
@@ -1357,6 +1418,7 @@ def update_tool(option_list: list):
 
 
 update_tool_options = [
+    PROJECT_ID_OPTION,
     Option(
         "tool_id",
         ["--tool-id", "--tid"],
@@ -1439,10 +1501,13 @@ update_tool_options = [
 
 
 def publish_tool_revision(option_list: list):
+    project_id = None
     tool_id = None
     revision = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "tool_id":
             tool_id = option_arg
         if option_flag.name == "revision":
@@ -1451,7 +1516,7 @@ def publish_tool_revision(option_list: list):
     if not (tool_id and revision):
         raise MissingRequirementException("Tool ID and revision must be specified.")
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.publish_tool_revision(
         tool_id=tool_id,
         revision=revision
@@ -1460,6 +1525,7 @@ def publish_tool_revision(option_list: list):
 
 
 publish_tool_revision_options = [
+    PROJECT_ID_OPTION,
     Option(
         "tool_id",
         ["--tool-id", "--tid"],
@@ -1476,6 +1542,7 @@ publish_tool_revision_options = [
 
 
 def get_parameter(option_list: list):
+    project_id = None
     tool_id = None
     tool_public_name = None
     allow_drafts = True
@@ -1483,6 +1550,8 @@ def get_parameter(option_list: list):
     version = 0
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "tool_id":
             tool_id = option_arg
         if option_flag.name == "tool_public_name":
@@ -1497,7 +1566,7 @@ def get_parameter(option_list: list):
     if not (tool_public_name or tool_id):
         raise MissingRequirementException("Tool public name or ID must be specified.")
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.get_parameter(
         tool_id=tool_id,
         tool_public_name=tool_public_name,
@@ -1509,6 +1578,7 @@ def get_parameter(option_list: list):
 
 
 get_parameter_options = [
+    PROJECT_ID_OPTION,
     Option(
         "tool_id",
         ["--tool-id", "--tid"],
@@ -1543,11 +1613,14 @@ get_parameter_options = [
 
 
 def set_parameter(option_list: list):
+    project_id = None
     tool_public_name = None
     tool_id = None
     parameters = list()
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "tool_id":
             tool_id = option_arg
         if option_flag.name == "tool_public_name":
@@ -1574,7 +1647,7 @@ def set_parameter(option_list: list):
 
     tool_parameters = get_tool_parameters(parameters)
 
-    client = ToolClient()
+    client = ToolClient(project_id=project_id)
     result = client.set_parameter(
         tool_id=tool_id,
         tool_public_name=tool_public_name,
@@ -1584,6 +1657,7 @@ def set_parameter(option_list: list):
 
 
 set_parameter_options = [
+    PROJECT_ID_OPTION,
     Option(
         "tool_id",
         ["--tool-id", "--tid"],
@@ -1947,6 +2021,7 @@ get_reasoning_strategy_options = [
 
 
 def create_process(option_list: list):
+    project_id = None
     key = None
     name = None
     description = None
@@ -1960,6 +2035,8 @@ def create_process(option_list: list):
     automatic_publish = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "key":
             key = option_arg
         if option_flag.name == "name":
@@ -2052,7 +2129,7 @@ def create_process(option_list: list):
     if not name:
         raise MissingRequirementException("Name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.create_process(
         key=key,
         name=name,
@@ -2070,6 +2147,7 @@ def create_process(option_list: list):
 
 
 create_process_options = [
+    PROJECT_ID_OPTION,
     Option(
         "key",
         ["--key", "-k"],
@@ -2154,6 +2232,7 @@ create_process_options = [
 
 
 def update_process(option_list: list):
+    project_id = None
     process_id = None
     name = None
     key = None
@@ -2169,6 +2248,8 @@ def update_process(option_list: list):
     upsert = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "process_id":
             process_id = option_arg
         if option_flag.name == "name":
@@ -2263,7 +2344,7 @@ def update_process(option_list: list):
     if not (process_id or name):
         raise MissingRequirementException("Either process ID or name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.update_process(
         process_id=process_id,
         name=name,
@@ -2283,6 +2364,7 @@ def update_process(option_list: list):
 
 
 update_process_options = [
+    PROJECT_ID_OPTION,
     Option(
         "process_id",
         ["--process-id", "--pid"],
@@ -2379,6 +2461,7 @@ update_process_options = [
 
 
 def get_process(option_list: list):
+    project_id = None
     process_id = None
     process_name = None
     revision = "0"
@@ -2386,6 +2469,8 @@ def get_process(option_list: list):
     allow_drafts = True
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "process_id":
             process_id = option_arg
         if option_flag.name == "process_name":
@@ -2400,7 +2485,7 @@ def get_process(option_list: list):
     if not (process_id or process_name):
         raise MissingRequirementException("Either process ID or process name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.get_process(
         process_id=process_id,
         process_name=process_name,
@@ -2412,6 +2497,7 @@ def get_process(option_list: list):
 
 
 get_process_options = [
+    PROJECT_ID_OPTION,
     Option(
         "process_id",
         ["--process-id", "--pid"],
@@ -2446,6 +2532,7 @@ get_process_options = [
 
 
 def list_processes(option_list: list):
+    project_id = None
     id = None
     name = None
     status = None
@@ -2454,6 +2541,8 @@ def list_processes(option_list: list):
     allow_draft = True
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "id":
             id = option_arg
         if option_flag.name == "name":
@@ -2467,7 +2556,7 @@ def list_processes(option_list: list):
         if option_flag.name == "allow_draft":
             allow_draft = get_boolean_value(option_arg)
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.list_processes(
         id=id,
         name=name,
@@ -2480,6 +2569,7 @@ def list_processes(option_list: list):
 
 
 list_processes_options = [
+    PROJECT_ID_OPTION,
     Option(
         "id",
         ["--id"],
@@ -2520,12 +2610,15 @@ list_processes_options = [
 
 
 def list_processes_instances(option_list: list):
+    project_id = None
     process_id = None
     is_active = True
     start = "0"
     count = "10"
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "process_id":
             process_id = option_arg
         if option_flag.name == "is_active":
@@ -2538,7 +2631,7 @@ def list_processes_instances(option_list: list):
     if not process_id:
         raise MissingRequirementException("Process ID must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.list_process_instances(
         process_id=process_id,
         is_active=is_active,
@@ -2549,6 +2642,7 @@ def list_processes_instances(option_list: list):
 
 
 list_processes_instances_options = [
+    PROJECT_ID_OPTION,
     Option(
         "process_id",
         ["--process-id", "--pid"],
@@ -2577,10 +2671,13 @@ list_processes_instances_options = [
 
 
 def delete_process(option_list: list):
+    project_id = None
     process_id = None
     process_name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "process_id":
             process_id = option_arg
         if option_flag.name == "process_name":
@@ -2589,7 +2686,7 @@ def delete_process(option_list: list):
     if not (process_id or process_name):
         raise MissingRequirementException("Either process ID or process name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.delete_process(
         process_id=process_id,
         process_name=process_name
@@ -2598,6 +2695,7 @@ def delete_process(option_list: list):
 
 
 delete_process_options = [
+    PROJECT_ID_OPTION,
     Option(
         "process_id",
         ["--process-id", "--pid"],
@@ -2614,11 +2712,14 @@ delete_process_options = [
 
 
 def publish_process_revision(option_list: list):
+    project_id = None
     process_id = None
     process_name = None
     revision = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "process_id":
             process_id = option_arg
         if option_flag.name == "process_name":
@@ -2631,7 +2732,7 @@ def publish_process_revision(option_list: list):
     if not revision:
         raise MissingRequirementException("Revision must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.publish_process_revision(
         process_id=process_id,
         process_name=process_name,
@@ -2641,6 +2742,7 @@ def publish_process_revision(option_list: list):
 
 
 publish_process_revision_options = [
+    PROJECT_ID_OPTION,
     Option(
         "process_id",
         ["--process-id", "--pid"],
@@ -2666,6 +2768,7 @@ publish_process_revision_options = [
 
 
 def create_task(option_list: list):
+    project_id = None
     name = None
     description = None
     title_template = None
@@ -2675,6 +2778,8 @@ def create_task(option_list: list):
     automatic_publish = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "name":
             name = option_arg
         if option_flag.name == "description":
@@ -2703,7 +2808,7 @@ def create_task(option_list: list):
     if not name:
         raise MissingRequirementException("Task name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.create_task(
         name=name,
         description=description,
@@ -2717,6 +2822,7 @@ def create_task(option_list: list):
 
 
 create_task_options = [
+    PROJECT_ID_OPTION,
     Option(
         "name",
         ["--name", "-n"],
@@ -2763,10 +2869,13 @@ create_task_options = [
 
 
 def get_task(option_list: list):
+    project_id = None
     task_id = None
     task_name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "task_id":
             task_id = option_arg
         if option_flag.name == "task_name":
@@ -2775,7 +2884,7 @@ def get_task(option_list: list):
     if not (task_id or task_name):
         raise MissingRequirementException("Either task ID or task name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.get_task(
         task_id=task_id,
         task_name=task_name
@@ -2784,6 +2893,7 @@ def get_task(option_list: list):
 
 
 get_task_options = [
+    PROJECT_ID_OPTION,
     Option(
         "task_id",
         ["--task-id", "--tid"],
@@ -2800,12 +2910,15 @@ get_task_options = [
 
 
 def list_tasks(option_list: list):
+    project_id = None
     id = None
     start = "0"
     count = "100"
     allow_drafts = True
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "id":
             id = option_arg
         if option_flag.name == "start":
@@ -2815,7 +2928,7 @@ def list_tasks(option_list: list):
         if option_flag.name == "allow_drafts":
             allow_drafts = get_boolean_value(option_arg)
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.list_tasks(
         id=id,
         start=start,
@@ -2826,6 +2939,7 @@ def list_tasks(option_list: list):
 
 
 list_tasks_options = [
+    PROJECT_ID_OPTION,
     Option(
         "id",
         ["--id"],
@@ -2854,6 +2968,7 @@ list_tasks_options = [
 
 
 def update_task(option_list: list):
+    project_id = None
     task_id = None
     name = None
     description = None
@@ -2865,6 +2980,8 @@ def update_task(option_list: list):
     upsert = False
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "task_id":
             task_id = option_arg
         if option_flag.name == "name":
@@ -2897,7 +3014,7 @@ def update_task(option_list: list):
     if not (task_id or name):
         raise MissingRequirementException("Either task ID or name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.update_task(
         task_id=task_id,
         name=name,
@@ -2913,6 +3030,7 @@ def update_task(option_list: list):
 
 
 update_task_options = [
+    PROJECT_ID_OPTION,
     Option(
         "task_id",
         ["--task-id", "--tid"],
@@ -2971,10 +3089,13 @@ update_task_options = [
 
 
 def delete_task(option_list: list):
+    project_id = None
     task_id = None
     task_name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "task_id":
             task_id = option_arg
         if option_flag.name == "task_name":
@@ -2983,7 +3104,7 @@ def delete_task(option_list: list):
     if not (task_id or task_name):
         raise MissingRequirementException("Either task ID or task name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.delete_task(
         task_id=task_id,
         task_name = task_name
@@ -2992,6 +3113,7 @@ def delete_task(option_list: list):
 
 
 delete_task_options = [
+    PROJECT_ID_OPTION,
     Option(
         "task_id",
         ["--task-id", "--tid"],
@@ -3008,11 +3130,14 @@ delete_task_options = [
 
 
 def publish_task_revision(option_list: list):
+    project_id = None
     task_id = None
     task_name = None
     revision = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "task_id":
             task_id = option_arg
         if option_flag.name == "task_name":
@@ -3025,7 +3150,7 @@ def publish_task_revision(option_list: list):
     if not revision:
         raise MissingRequirementException("Revision must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.publish_task_revision(
         task_id=task_id,
         task_name=task_name,
@@ -3035,6 +3160,7 @@ def publish_task_revision(option_list: list):
 
 
 publish_task_revision_options = [
+    PROJECT_ID_OPTION,
     Option(
         "task_id",
         ["--task-id", "--tid"],
@@ -3058,11 +3184,14 @@ publish_task_revision_options = [
 
 # RUNTIME - Process instances
 def start_instance(option_list: list):
+    project_id = None
     process_name = None
     subject = None
     variables = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "process_name":
             process_name = option_arg
         if option_flag.name == "subject":
@@ -3080,7 +3209,7 @@ def start_instance(option_list: list):
     if not process_name:
         raise MissingRequirementException("Process name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.start_instance(
         process_name=process_name,
         subject=subject,
@@ -3090,6 +3219,7 @@ def start_instance(option_list: list):
 
 
 start_instance_options = [
+    PROJECT_ID_OPTION,
     Option(
         "process_name",
         ["--process-name", "--pn"],
@@ -3112,16 +3242,19 @@ start_instance_options = [
 
 
 def abort_instance(option_list: list):
+    project_id = None
     instance_id = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "instance_id":
             instance_id = option_arg
 
     if not instance_id:
         raise MissingRequirementException("Instance ID must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.abort_instance(
         instance_id=instance_id
     )
@@ -3129,6 +3262,7 @@ def abort_instance(option_list: list):
 
 
 abort_instance_options = [
+    PROJECT_ID_OPTION,
     Option(
         "instance_id",
         ["--instance-id", "--iid"],
@@ -3139,16 +3273,19 @@ abort_instance_options = [
 
 
 def get_instance(option_list: list):
+    project_id = None
     instance_id = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "instance_id":
             instance_id = option_arg
 
     if not instance_id:
         raise MissingRequirementException("Instance ID must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.get_instance(
         instance_id=instance_id
     )
@@ -3156,6 +3293,7 @@ def get_instance(option_list: list):
 
 
 get_instance_options = [
+    PROJECT_ID_OPTION,
     Option(
         "instance_id",
         ["--instance-id", "--iid"],
@@ -3166,16 +3304,19 @@ get_instance_options = [
 
 
 def get_instance_history(option_list: list):
+    project_id = None
     instance_id = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "instance_id":
             instance_id = option_arg
 
     if not instance_id:
         raise MissingRequirementException("Instance ID must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.get_instance_history(
         instance_id=instance_id
     )
@@ -3183,6 +3324,7 @@ def get_instance_history(option_list: list):
 
 
 get_instance_history_options = [
+    PROJECT_ID_OPTION,
     Option(
         "instance_id",
         ["--instance-id", "--iid"],
@@ -3192,16 +3334,19 @@ get_instance_history_options = [
 ]
 
 def get_thread_information(option_list: list):
+    project_id = None
     thread_id = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "thread_id":
             thread_id = option_arg
 
     if not thread_id:
         raise MissingRequirementException("Thread ID must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.get_thread_information(
         thread_id=thread_id
     )
@@ -3209,6 +3354,7 @@ def get_thread_information(option_list: list):
 
 
 get_thread_information_options = [
+    PROJECT_ID_OPTION,
     Option(
         "thread_id",
         ["--thread-id", "--tid"],
@@ -3219,10 +3365,13 @@ get_thread_information_options = [
 
 
 def send_user_signal(option_list: list):
+    project_id = None
     instance_id = None
     signal_name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "instance_id":
             instance_id = option_arg
         if option_flag.name == "signal_name":
@@ -3233,7 +3382,7 @@ def send_user_signal(option_list: list):
     if not signal_name:
         raise MissingRequirementException("Signal name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.send_user_signal(
         instance_id=instance_id,
         signal_name=signal_name
@@ -3242,6 +3391,7 @@ def send_user_signal(option_list: list):
 
 
 send_user_signal_options = [
+    PROJECT_ID_OPTION,
     Option(
         "instance_id",
         ["--instance-id", "--iid"],
@@ -3260,11 +3410,14 @@ send_user_signal_options = [
 
 
 def create_kb(option_list: list):
+    project_id = None
     name = None
     artifacts = None
     metadata = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "name":
             name = option_arg
         if option_flag.name == "artifacts":
@@ -3289,7 +3442,7 @@ def create_kb(option_list: list):
     if not name:
         raise MissingRequirementException("Name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.create_kb(
         name=name,
         artifacts=artifacts,
@@ -3299,6 +3452,7 @@ def create_kb(option_list: list):
 
 
 create_kb_options = [
+    PROJECT_ID_OPTION,
     Option(
         "name",
         ["--name", "-n"],
@@ -3321,10 +3475,13 @@ create_kb_options = [
 
 
 def get_kb(option_list: list):
+    project_id = None
     kb_id = None
     kb_name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "kb_id":
             kb_id = option_arg
         if option_flag.name == "kb_name":
@@ -3333,7 +3490,7 @@ def get_kb(option_list: list):
     if not (kb_id or kb_name):
         raise MissingRequirementException("Either KB ID or KB name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.get_kb(
         kb_id=kb_id,
         kb_name=kb_name
@@ -3342,6 +3499,7 @@ def get_kb(option_list: list):
 
 
 get_kb_options = [
+    PROJECT_ID_OPTION,
     Option(
         "kb_id",
         ["--kb-id", "--kid"],
@@ -3358,11 +3516,14 @@ get_kb_options = [
 
 
 def list_kbs(option_list: list):
+    project_id = None
     name = None
     start = "0"
     count = "100"
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "name":
             name = option_arg
         if option_flag.name == "start":
@@ -3370,7 +3531,7 @@ def list_kbs(option_list: list):
         if option_flag.name == "count":
             count = option_arg
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.list_kbs(
         name=name,
         start=start,
@@ -3380,6 +3541,7 @@ def list_kbs(option_list: list):
 
 
 list_kbs_options = [
+    PROJECT_ID_OPTION,
     Option(
         "name",
         ["--name", "-n"],
@@ -3402,10 +3564,13 @@ list_kbs_options = [
 
 
 def delete_kb(option_list: list):
+    project_id = None
     kb_id = None
     kb_name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "kb_id":
             kb_id = option_arg
         if option_flag.name == "kb_name":
@@ -3414,7 +3579,7 @@ def delete_kb(option_list: list):
     if not (kb_id or kb_name):
         raise MissingRequirementException("Either KB ID or KB name must be specified.")
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.delete_kb(
         kb_id=kb_id,
         kb_name=kb_name
@@ -3423,6 +3588,7 @@ def delete_kb(option_list: list):
 
 
 delete_kb_options = [
+    PROJECT_ID_OPTION,
     Option(
         "kb_id",
         ["--kb-id", "--kid"],
@@ -3441,6 +3607,7 @@ delete_kb_options = [
 
 
 def list_jobs(option_list: list):
+    project_id = None
     start = "0"
     count = "100"
     topic = None
@@ -3448,6 +3615,8 @@ def list_jobs(option_list: list):
     name = None
 
     for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
         if option_flag.name == "start":
             start = option_arg
         if option_flag.name == "count":
@@ -3459,7 +3628,7 @@ def list_jobs(option_list: list):
         if option_flag.name == "name":
             name = option_arg
 
-    client = AgenticProcessClient()
+    client = AgenticProcessClient(project_id=project_id)
     result = client.list_jobs(
         start=start,
         count=count,
@@ -3471,6 +3640,7 @@ def list_jobs(option_list: list):
 
 
 list_jobs_options = [
+    PROJECT_ID_OPTION,
     Option(
         "start",
         ["--start", "-s"],