pygeai 0.6.0b13__py3-none-any.whl → 0.6.1__py3-none-any.whl

pygeai/cli/commands/lab/ai_lab.py

@@ -26,29 +26,15 @@ def show_help():
 
 
 def list_agents(option_list: list):
-    project_id = None
-    status = ""
-    start = ""
-    count = ""
-    access_scope = "public"
-    allow_drafts = True
-    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":
-            start = option_arg
-        if option_flag.name == "count":
-            count = option_arg
-        if option_flag.name == "access_scope":
-            access_scope = option_arg
-        if option_flag.name == "allow_drafts":
-            allow_drafts = get_boolean_value(option_arg)
-        if option_flag.name == "allow_external":
-            allow_external = get_boolean_value(option_arg)
+    opts = {opt.name: arg for opt, arg in option_list}
+    
+    project_id = opts.get('project_id')
+    status = opts.get('status', '')
+    start = opts.get('start', '')
+    count = opts.get('count', '')
+    access_scope = opts.get('access_scope', 'public')
+    allow_drafts = get_boolean_value(opts['allow_drafts']) if 'allow_drafts' in opts else True
+    allow_external = get_boolean_value(opts['allow_external']) if 'allow_external' in opts else False
 
     client = AgentClient(project_id=project_id)
     result = client.list_agents(
@@ -112,9 +98,9 @@ def create_agent(option_list: list):
     avatar_image = None
     description = None
     agent_data_prompt_instructions = None
-    agent_data_prompt_inputs = list()
-    agent_data_prompt_outputs = list()
-    agent_data_prompt_examples = list()
+    agent_data_prompt_inputs = []
+    agent_data_prompt_outputs = []
+    agent_data_prompt_examples = []
     agent_data_llm_max_tokens = None
     agent_data_llm_timeout = None
     agent_data_llm_temperature = None
@@ -154,7 +140,7 @@ def create_agent(option_list: list):
                         raise ValueError
 
                     agent_data_prompt_inputs = input_json
-                except Exception as e:
+                except Exception:
                     raise WrongArgumentError(
                         "Inputs must be a list of strings: '[\"input_name\", \"another_input\"]'. "
                         "Each element in the list must be a string representing an input name."
@@ -166,7 +152,7 @@ def create_agent(option_list: list):
                     agent_data_prompt_outputs = output_json
                 elif isinstance(output_json, dict):
                     agent_data_prompt_outputs.append(output_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each output must be in JSON format: '{\"key\": \"output_key\", \"description\": \"description of the output\"}' "
                     "It must be a dictionary or a list of dictionaries. Each dictionary must contain 'key' and 'description'."
@@ -179,7 +165,7 @@ def create_agent(option_list: list):
                     agent_data_prompt_examples = examples_json
                 elif isinstance(examples_json, dict):
                     agent_data_prompt_examples.append(examples_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each example must be in JSON format: '{\"inputData\": \"example input\", \"output\": \"expected output in JSON string format\"}' "
                     "It must be a dictionary or a list of dictionaries. Each dictionary must contain 'inputData' and 'output'."
@@ -205,7 +191,7 @@ def create_agent(option_list: list):
                 if not isinstance(pools_json, list):
                     raise ValueError
                 agent_data_resource_pools = pools_json
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Resource pools must be in JSON format: '[{\"name\": \"pool_name\", \"tools\": [{\"name\": \"tool_name\", \"revision\": int}], \"agents\": [{\"name\": \"agent_name\", \"revision\": int}]}]' "
                     "It must be a list of dictionaries. Each dictionary must contain 'name' and optional 'tools' and 'agents' lists."
@@ -213,8 +199,11 @@ def create_agent(option_list: list):
         if option_flag.name == "automatic_publish":
             automatic_publish = get_boolean_value(option_arg)
 
-    if not (name and access_scope and public_name):
-        raise MissingRequirementException("Cannot create assistant without specifying name, access scope and public name")
+    if not name:
+        raise MissingRequirementException("Cannot create assistant without specifying name.")
+
+    if access_scope == 'public' and not public_name:
+        raise MissingRequirementException("If access scope is public, public name must be defined.")
 
     prompt_inputs = get_agent_data_prompt_inputs(agent_data_prompt_inputs)
     prompt_outputs = get_agent_data_prompt_outputs(agent_data_prompt_outputs)
@@ -379,23 +368,13 @@ 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":
-            revision = option_arg
-        if option_flag.name == "version":
-            version = option_arg
-        if option_flag.name == "allow_drafts":
-            allow_drafts = get_boolean_value(option_arg)
+    opts = {opt.name: arg for opt, arg in option_list}
+    
+    project_id = opts.get('project_id')
+    agent_id = opts.get('agent_id')
+    revision = opts.get('revision', 0)
+    version = opts.get('version', 0)
+    allow_drafts = get_boolean_value(opts['allow_drafts']) if 'allow_drafts' in opts else True
 
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
@@ -440,17 +419,11 @@ 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":
-            file = option_arg
+    opts = {opt.name: arg for opt, arg in option_list}
+    
+    project_id = opts.get('project_id')
+    agent_id = opts.get('agent_id')
+    file = opts.get('file')
 
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
@@ -467,7 +440,7 @@ def export_agent(option_list: list):
             Console.write_stdout(f"Result from API saved to {file}.")
         except json.JSONDecodeError as e:
             logger.error(f"Result from API endpoint is not in JSON format: {e}")
-            Console.write_stderr(f"Result from API endpoint is not in JSON format.")
+            Console.write_stderr("Result from API endpoint is not in JSON format.")
     else:
         Console.write_stdout(f"Agent spec: \n{result}")
 
@@ -490,15 +463,10 @@ 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
-
+    opts = {opt.name: arg for opt, arg in option_list}
+    
+    project_id = opts.get('project_id')
+    file = opts.get('file')
 
     if not file:
         raise MissingRequirementException("File path to spec must be specified.")
@@ -508,7 +476,7 @@ def import_agent(option_list: list):
             agent_data = json.load(f)
     except json.JSONDecodeError as e:
         logger.error(f"File is not in JSON format: {e}")
-        Console.write_stderr(f"File is not in JSON format.")
+        Console.write_stderr("File is not in JSON format.")
 
     client = AgentClient(project_id=project_id)
     result = client.import_agent(
@@ -531,14 +499,10 @@ 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
+    opts = {opt.name: arg for opt, arg in option_list}
+    
+    project_id = opts.get('project_id')
+    agent_id = opts.get('agent_id')
 
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
@@ -562,17 +526,11 @@ 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":
-            revision = option_arg
+    opts = {opt.name: arg for opt, arg in option_list}
+    
+    project_id = opts.get('project_id')
+    agent_id = opts.get('agent_id')
+    revision = opts.get('revision')
 
     if not (agent_id and revision):
         raise MissingRequirementException("Agent ID and revision must be specified.")
@@ -603,14 +561,10 @@ 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
+    opts = {opt.name: arg for opt, arg in option_list}
+    
+    project_id = opts.get('project_id')
+    agent_id = opts.get('agent_id')
 
     if not agent_id:
         raise MissingRequirementException("Agent ID must be specified.")
@@ -643,9 +597,9 @@ def update_agent(option_list: list):
     avatar_image = None
     description = None
     agent_data_prompt_instructions = None
-    agent_data_prompt_inputs = list()
-    agent_data_prompt_outputs = list()
-    agent_data_prompt_examples = list()
+    agent_data_prompt_inputs = []
+    agent_data_prompt_outputs = []
+    agent_data_prompt_examples = []
     agent_data_llm_max_tokens = None
     agent_data_llm_timeout = None
     agent_data_llm_temperature = None
@@ -688,7 +642,7 @@ def update_agent(option_list: list):
                         raise ValueError
 
                     agent_data_prompt_inputs = input_json
-                except Exception as e:
+                except Exception:
                     raise WrongArgumentError(
                         "Inputs must be a list of strings: '[\"input_name\", \"another_input\"]'. "
                         "Each element in the list must be a string representing an input name."
@@ -700,7 +654,7 @@ def update_agent(option_list: list):
                     agent_data_prompt_outputs = output_json
                 elif isinstance(output_json, dict):
                     agent_data_prompt_outputs.append(output_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each output must be in JSON format: '{\"key\": \"output_key\", \"description\": \"description of the output\"}' "
                     "It must be a dictionary or a list of dictionaries. Each dictionary must contain 'key' and 'description'."
@@ -713,7 +667,7 @@ def update_agent(option_list: list):
                     agent_data_prompt_examples = examples_json
                 elif isinstance(examples_json, dict):
                     agent_data_prompt_examples.append(examples_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each example must be in JSON format: '{\"inputData\": \"example input\", \"output\": \"expected output in JSON string format\"}' "
                     "It must be a dictionary or a list of dictionaries. Each dictionary must contain 'inputData' and 'output'."
@@ -739,7 +693,7 @@ def update_agent(option_list: list):
                 if not isinstance(pools_json, list):
                     raise ValueError
                 agent_data_resource_pools = pools_json
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Resource pools must be in JSON format: '[{\"name\": \"pool_name\", \"tools\": [{\"name\": \"tool_name\", \"revision\": int}], \"agents\": [{\"name\": \"agent_name\", \"revision\": int}]}]' "
                     "It must be a list of dictionaries. Each dictionary must contain 'name' and optional 'tools' and 'agents' lists."
@@ -940,7 +894,7 @@ def create_tool(option_list: list):
     open_api = None
     open_api_json = None
     report_events = "None"
-    parameters = list()
+    parameters = []
     automatic_publish = False
 
     for option_flag, option_arg in option_list:
@@ -965,7 +919,7 @@ def create_tool(option_list: list):
                 open_api_json = json.loads(option_arg)
                 if not isinstance(open_api_json, dict):
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "open_api_json must be a valid JSON object (e.g., '{\"openapi\": \"3.0.0\", \"info\": {\"title\": \"example\", \"version\": \"1.0.0\"}, ...}')"
                 )
@@ -977,7 +931,7 @@ def create_tool(option_list: list):
                 if not isinstance(param_json, dict):
                     raise ValueError
                 parameters.append(param_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each parameter must be in JSON format (e.g., "
                     "'{\"key\": \"param_name\", \"description\": \"param description\", \"isRequired\": true, \"type\": \"app\"}' "
@@ -1257,7 +1211,7 @@ def export_tool(option_list: list):
             Console.write_stdout(f"Result from API saved to {file}.")
         except json.JSONDecodeError as e:
             logger.error(f"Result from API endpoint is not in JSON format: {e}")
-            Console.write_stderr(f"Result from API endpoint is not in JSON format.")
+            Console.write_stderr("Result from API endpoint is not in JSON format.")
 
 
 
@@ -1332,7 +1286,7 @@ def update_tool(option_list: list):
     open_api = None
     open_api_json = None
     report_events = "None"
-    parameters = list()
+    parameters = []
     automatic_publish = False
     upsert = False
 
@@ -1360,7 +1314,7 @@ def update_tool(option_list: list):
                 open_api_json = json.loads(option_arg)
                 if not isinstance(open_api_json, dict):
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "open_api_json must be a valid JSON object (e.g., '{\"openapi\": \"3.0.0\", \"info\": {\"title\": \"example\", \"version\": \"1.0.0\"}, ...}')"
                 )
@@ -1372,7 +1326,7 @@ def update_tool(option_list: list):
                 if not isinstance(param_json, dict):
                     raise ValueError
                 parameters.append(param_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each parameter must be in JSON format (e.g., "
                     "'{\"key\": \"param_name\", \"description\": \"param description\", \"isRequired\": true, \"type\": \"app\"}' "
@@ -1613,7 +1567,7 @@ def set_parameter(option_list: list):
     project_id = None
     tool_public_name = None
     tool_id = None
-    parameters = list()
+    parameters = []
 
     for option_flag, option_arg in option_list:
         if option_flag.name == "project_id":
@@ -1628,7 +1582,7 @@ def set_parameter(option_list: list):
                 if not isinstance(param_json, dict):
                     raise ValueError
                 parameters.append(param_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each parameter must be in JSON format: "
                     "'{\"key\": \"param_name\", \"dataType\": \"String\", \"description\": \"param description\", \"isRequired\": true}' "
@@ -1757,7 +1711,7 @@ def create_reasoning_strategy(option_list: list):
     system_prompt = None
     access_scope = "public"
     strategy_type = "addendum"
-    localized_descriptions = list()
+    localized_descriptions = []
     automatic_publish = False
 
     for option_flag, option_arg in option_list:
@@ -1775,7 +1729,7 @@ def create_reasoning_strategy(option_list: list):
                 if not isinstance(desc_json, dict) or "language" not in desc_json or "description" not in desc_json:
                     raise ValueError
                 localized_descriptions.append(desc_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each localized description must be in JSON format: "
                     "'{\"language\": \"english\", \"description\": \"description text\"}'"
@@ -1860,7 +1814,7 @@ def update_reasoning_strategy(option_list: list):
     system_prompt = None
     access_scope = None
     strategy_type = None
-    localized_descriptions = list()
+    localized_descriptions = []
     automatic_publish = False
     upsert = False
 
@@ -1881,7 +1835,7 @@ def update_reasoning_strategy(option_list: list):
                 if not isinstance(desc_json, dict) or "language" not in desc_json or "description" not in desc_json:
                     raise ValueError
                 localized_descriptions.append(desc_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each localized description must be in JSON format: "
                     "'{\"language\": \"english\", \"description\": \"description text\"}'"
@@ -2023,12 +1977,12 @@ def create_process(option_list: list):
     name = None
     description = None
     kb = None
-    agentic_activities = list()
-    artifact_signals = list()
-    user_signals = list()
+    agentic_activities = []
+    artifact_signals = []
+    user_signals = []
     start_event = None
     end_event = None
-    sequence_flows = list()
+    sequence_flows = []
     automatic_publish = False
 
     for option_flag, option_arg in option_list:
@@ -2045,7 +1999,7 @@ def create_process(option_list: list):
                 kb = json.loads(option_arg)
                 if not isinstance(kb, dict) or "name" not in kb or "artifactTypeName" not in kb:
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "KB must be in JSON format: "
                     "'{\"name\": \"basic-sample\", \"artifactTypeName\": [\"sample-artifact\"]}'"
@@ -2058,7 +2012,7 @@ def create_process(option_list: list):
                 elif not isinstance(activity_json, dict) or "key" not in activity_json or "name" not in activity_json:
                     raise ValueError
                 agentic_activities.append(activity_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each agentic activity must be in JSON format: "
                     "'{\"key\": \"activityOne\", \"name\": \"First Step\", \"taskName\": \"basic-task\", "
@@ -2070,7 +2024,7 @@ def create_process(option_list: list):
                 if not isinstance(signal_json, dict) or "key" not in signal_json or "name" not in signal_json:
                     raise ValueError
                 artifact_signals.append(signal_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each artifact signal must be in JSON format: "
                     "'{\"key\": \"artifact.upload.1\", \"name\": \"artifact.upload\", \"handlingType\": \"C\", "
@@ -2082,7 +2036,7 @@ def create_process(option_list: list):
                 if not isinstance(signal_json, dict) or "key" not in signal_json or "name" not in signal_json:
                     raise ValueError
                 user_signals.append(signal_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each user signal must be in JSON format: "
                     "'{\"key\": \"signal_done\", \"name\": \"process-completed\"}'"
@@ -2092,7 +2046,7 @@ def create_process(option_list: list):
                 start_event = json.loads(option_arg)
                 if not isinstance(start_event, dict) or "key" not in start_event or "name" not in start_event:
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Start event must be in JSON format: "
                     "'{\"key\": \"artifact.upload.1\", \"name\": \"artifact.upload\"}'"
@@ -2102,7 +2056,7 @@ def create_process(option_list: list):
                 end_event = json.loads(option_arg)
                 if not isinstance(end_event, dict) or "key" not in end_event or "name" not in end_event:
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "End event must be in JSON format: "
                     "'{\"key\": \"end\", \"name\": \"Done\"}'"
@@ -2113,7 +2067,7 @@ def create_process(option_list: list):
                 if not isinstance(flow_json, dict) or "key" not in flow_json or "sourceKey" not in flow_json or "targetKey" not in flow_json:
                     raise ValueError
                 sequence_flows.append(flow_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each sequence flow must be in JSON format: "
                     "'{\"key\": \"step1\", \"sourceKey\": \"artifact.upload.1\", \"targetKey\": \"activityOne\"}'"
@@ -2235,12 +2189,12 @@ def update_process(option_list: list):
     key = None
     description = None
     kb = None
-    agentic_activities = list()
-    artifact_signals = list()
-    user_signals = list()
+    agentic_activities = []
+    artifact_signals = []
+    user_signals = []
     start_event = None
     end_event = None
-    sequence_flows = list()
+    sequence_flows = []
     automatic_publish = False
     upsert = False
 
@@ -2260,7 +2214,7 @@ def update_process(option_list: list):
                 kb = json.loads(option_arg)
                 if not isinstance(kb, dict) or "name" not in kb or "artifactTypeName" not in kb:
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "KB must be in JSON format: "
                     "'{\"name\": \"basic-sample\", \"artifactTypeName\": [\"sample-artifact\"]}'"
@@ -2273,7 +2227,7 @@ def update_process(option_list: list):
                 elif not isinstance(activity_json, dict) or "key" not in activity_json or "name" not in activity_json:
                     raise ValueError
                 agentic_activities.append(activity_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each agentic activity must be in JSON format: "
                     "'{\"key\": \"activityOne\", \"name\": \"First Step\", \"taskName\": \"basic-task\", "
@@ -2285,7 +2239,7 @@ def update_process(option_list: list):
                 if not isinstance(signal_json, dict) or "key" not in signal_json or "name" not in signal_json:
                     raise ValueError
                 artifact_signals.append(signal_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each artifact signal must be in JSON format: "
                     "'{\"key\": \"artifact.upload.1\", \"name\": \"artifact.upload\", \"handlingType\": \"C\", "
@@ -2297,7 +2251,7 @@ def update_process(option_list: list):
                 if not isinstance(signal_json, dict) or "key" not in signal_json or "name" not in signal_json:
                     raise ValueError
                 user_signals.append(signal_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each user signal must be in JSON format: "
                     "'{\"key\": \"signal_done\", \"name\": \"process-completed\"}'"
@@ -2307,7 +2261,7 @@ def update_process(option_list: list):
                 start_event = json.loads(option_arg)
                 if not isinstance(start_event, dict) or "key" not in start_event or "name" not in start_event:
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Start event must be in JSON format: "
                     "'{\"key\": \"artifact.upload.1\", \"name\": \"artifact.upload\"}'"
@@ -2317,7 +2271,7 @@ def update_process(option_list: list):
                 end_event = json.loads(option_arg)
                 if not isinstance(end_event, dict) or "key" not in end_event or "name" not in end_event:
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "End event must be in JSON format: "
                     "'{\"key\": \"end\", \"name\": \"Done\"}'"
@@ -2328,7 +2282,7 @@ def update_process(option_list: list):
                 if not isinstance(flow_json, dict) or "key" not in flow_json or "sourceKey" not in flow_json or "targetKey" not in flow_json:
                     raise ValueError
                 sequence_flows.append(flow_json)
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Each sequence flow must be in JSON format: "
                     "'{\"key\": \"step1\", \"sourceKey\": \"artifact.upload.1\", \"targetKey\": \"activityOne\"}'"
@@ -3422,7 +3376,7 @@ def create_kb(option_list: list):
                 artifacts = json.loads(option_arg)
                 if not isinstance(artifacts, list):
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Artifacts must be a JSON list of strings: '[\"artifact1\", \"artifact2\"]'"
                 )
@@ -3431,7 +3385,7 @@ def create_kb(option_list: list):
                 metadata = json.loads(option_arg)
                 if not isinstance(metadata, list):
                     raise ValueError
-            except Exception as e:
+            except Exception:
                 raise WrongArgumentError(
                     "Metadata must be a JSON list of strings: '[\"meta1\", \"meta2\"]'"
                 )

pygeai/cli/commands/lab/common.py

@@ -123,7 +123,7 @@ def get_tool_parameters(parameter_list: list) -> list:
 
             parameters.append(validated_param)
 
-    except (KeyError, TypeError) as e:
+    except (KeyError, TypeError):
         raise WrongArgumentError(
             "Each parameter must be in JSON format: "
             "'{\"key\": \"param_name\", \"dataType\": \"String\", \"description\": \"param description\", \"isRequired\": true}' "

pygeai/cli/commands/lab/spec.py

@@ -19,14 +19,9 @@ def show_help():
 
 
 def load_agent(option_list: list):
-    file = None
-    automatic_publish = False
-
-    for option_flag, option_arg in option_list:
-        if option_flag.name == "file":
-            file = option_arg
-        if option_flag.name == "automatic_publish":
-            automatic_publish = option_arg
+    opts = {opt.name: arg for opt, arg in option_list}
+    file = opts.get("file")
+    automatic_publish = opts.get("automatic_publish", False)
 
     if not file:
         raise MissingRequirementException("Cannot load agent definition without specifying path to JSON file.")
@@ -71,14 +66,9 @@ load_agent_options = [
 
 
 def load_tool(option_list: list):
-    file = None
-    automatic_publish = False
-
-    for option_flag, option_arg in option_list:
-        if option_flag.name == "file":
-            file = option_arg
-        if option_flag.name == "automatic_publish":
-            automatic_publish = option_arg
+    opts = {opt.name: arg for opt, arg in option_list}
+    file = opts.get("file")
+    automatic_publish = opts.get("automatic_publish", False)
 
     if not file:
         raise MissingRequirementException("Cannot load tool definition without specifying path to JSON file.")
@@ -122,14 +112,9 @@ load_tool_options = [
 
 
 def load_task(option_list: list):
-    file = None
-    automatic_publish = False
-
-    for option_flag, option_arg in option_list:
-        if option_flag.name == "file":
-            file = option_arg
-        if option_flag.name == "automatic_publish":
-            automatic_publish = option_arg
+    opts = {opt.name: arg for opt, arg in option_list}
+    file = opts.get("file")
+    automatic_publish = opts.get("automatic_publish", False)
 
     if not file:
         raise MissingRequirementException("Cannot load task definition without specifying path to JSON file.")
@@ -173,14 +158,9 @@ load_task_options = [
 
 
 def load_agentic_process(option_list: list):
-    file = None
-    automatic_publish = False
-
-    for option_flag, option_arg in option_list:
-        if option_flag.name == "file":
-            file = option_arg
-        if option_flag.name == "automatic_publish":
-            automatic_publish = option_arg
+    opts = {opt.name: arg for opt, arg in option_list}
+    file = opts.get("file")
+    automatic_publish = opts.get("automatic_publish", False)
 
 
     if not file: