glaip-sdk 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

glaip_sdk/cli/utils.py

@@ -147,9 +147,7 @@ def format_command_hint(
 
     highlighted = f"[{HINT_COMMAND_STYLE}]{command}[/]"
     if description:
-        highlighted += (
-            f"  [{HINT_DESCRIPTION_COLOR}]{description}[/{HINT_DESCRIPTION_COLOR}]"
-        )
+        highlighted += f"  [{HINT_DESCRIPTION_COLOR}]{description}[/{HINT_DESCRIPTION_COLOR}]"
     return highlighted
 
 
@@ -234,7 +232,7 @@ _spinner_stop = stop_spinner
 def get_client(ctx: Any) -> Client:  # pragma: no cover
     """Get configured client from context, env, and config file (ctx > env > file)."""
     module = importlib.import_module("glaip_sdk")
-    client_class = cast("type[Client]", getattr(module, "Client"))
+    client_class = cast("type[Client]", module.Client)
     file_config = load_config() or {}
     context_config_obj = getattr(ctx, "obj", None)
     context_config = context_config_obj or {}
@@ -419,9 +417,7 @@ def _prompt_with_auto_select(
             reserve_space_for_menu=8,
         )
     except Exception as exc:  # pragma: no cover - depends on prompt_toolkit
-        logger.debug(
-            "PromptSession init failed (%s); falling back to basic prompt.", exc
-        )
+        logger.debug("PromptSession init failed (%s); falling back to basic prompt.", exc)
         return _basic_prompt(message, completer)
 
     buffer = session.default_buffer
@@ -447,9 +443,7 @@ def _prompt_with_auto_select(
     except (KeyboardInterrupt, EOFError):
         return None
     except Exception as exc:  # pragma: no cover - defensive
-        logger.debug(
-            "PromptSession prompt failed (%s); falling back to basic prompt.", exc
-        )
+        logger.debug("PromptSession prompt failed (%s); falling back to basic prompt.", exc)
         return _basic_prompt(message, completer)
     finally:
         if handler_attached:
@@ -465,9 +459,7 @@ class _FuzzyCompleter:
     def __init__(self, words: list[str]) -> None:
         self.words = words
 
-    def get_completions(
-        self, document: Any, _complete_event: Any
-    ) -> Any:  # pragma: no cover
+    def get_completions(self, document: Any, _complete_event: Any) -> Any:  # pragma: no cover
         word = document.get_word_before_cursor()
         if not word:
             return
@@ -492,9 +484,7 @@ class _FuzzyCompleter:
         return False
 
 
-def _perform_fuzzy_search(
-    answer: str, labels: list[str], by_label: dict[str, dict[str, Any]]
-) -> dict[str, Any] | None:
+def _perform_fuzzy_search(answer: str, labels: list[str], by_label: dict[str, dict[str, Any]]) -> dict[str, Any] | None:
     """Perform fuzzy search fallback and return best match."""
     # Exact label match
     if answer in by_label:
@@ -615,7 +605,7 @@ def _coerce_result_payload(result: Any) -> Any:
 
 
 def _ensure_displayable(payload: Any) -> Any:
-    if isinstance(payload, dict | list | str | int | float | bool) or payload is None:
+    if isinstance(payload, (dict, list, str, int, float, bool)) or payload is None:
         return payload
 
     if hasattr(payload, "__dict__"):
@@ -686,9 +676,7 @@ def output_result(
 # _PICK_THRESHOLD = int(os.getenv("AIP_PICK_THRESHOLD", "5") or "5")
 
 
-def _normalise_rows(
-    items: list[Any], transform_func: Callable[[Any], dict[str, Any]] | None
-) -> list[dict[str, Any]]:
+def _normalise_rows(items: list[Any], transform_func: Callable[[Any], dict[str, Any]] | None) -> list[dict[str, Any]]:
     try:
         rows: list[dict[str, Any]] = []
         for item in items:
@@ -707,9 +695,7 @@ def _normalise_rows(
         return []
 
 
-def _render_plain_list(
-    rows: list[dict[str, Any]], title: str, columns: list[tuple]
-) -> None:
+def _render_plain_list(rows: list[dict[str, Any]], title: str, columns: list[tuple]) -> None:
     if not rows:
         click.echo(f"No {title.lower()} found.")
         return
@@ -718,9 +704,7 @@ def _render_plain_list(
         click.echo(row_str)
 
 
-def _render_markdown_list(
-    rows: list[dict[str, Any]], title: str, columns: list[tuple]
-) -> None:
+def _render_markdown_list(rows: list[dict[str, Any]], title: str, columns: list[tuple]) -> None:
     if not rows:
         click.echo(f"No {title.lower()} found.")
         return
@@ -748,9 +732,7 @@ def _create_table(columns: list[tuple[str, str, str, int | None]], title: str) -
     return table
 
 
-def _build_table_group(
-    rows: list[dict[str, Any]], columns: list[tuple], title: str
-) -> Group:
+def _build_table_group(rows: list[dict[str, Any]], columns: list[tuple], title: str) -> Group:
     table = _create_table(columns, title)
     for row in rows:
         table.add_row(*[str(row.get(key, "N/A")) for key, _, _, _ in columns])
@@ -760,24 +742,16 @@ def _build_table_group(
 
 def _handle_json_output(items: list[Any], rows: list[dict[str, Any]]) -> None:
     """Handle JSON output format."""
-    data = (
-        rows
-        if rows
-        else [it.to_dict() if hasattr(it, "to_dict") else it for it in items]
-    )
+    data = rows if rows else [it.to_dict() if hasattr(it, "to_dict") else it for it in items]
     click.echo(json.dumps(data, indent=2, default=str))
 
 
-def _handle_plain_output(
-    rows: list[dict[str, Any]], title: str, columns: list[tuple]
-) -> None:
+def _handle_plain_output(rows: list[dict[str, Any]], title: str, columns: list[tuple]) -> None:
     """Handle plain text output format."""
     _render_plain_list(rows, title, columns)
 
 
-def _handle_markdown_output(
-    rows: list[dict[str, Any]], title: str, columns: list[tuple]
-) -> None:
+def _handle_markdown_output(rows: list[dict[str, Any]], title: str, columns: list[tuple]) -> None:
     """Handle markdown output format."""
     _render_markdown_list(rows, title, columns)
 
@@ -792,9 +766,7 @@ def _should_use_fuzzy_picker() -> bool:
     return console.is_terminal and os.isatty(1)
 
 
-def _try_fuzzy_pick(
-    rows: list[dict[str, Any]], columns: list[tuple], title: str
-) -> dict[str, Any] | None:
+def _try_fuzzy_pick(rows: list[dict[str, Any]], columns: list[tuple], title: str) -> dict[str, Any] | None:
     """Best-effort fuzzy selection; returns None if the picker fails."""
     if not _should_use_fuzzy_picker():
         return None
@@ -825,14 +797,10 @@ def _print_selection_tip(title: str) -> None:
     """Print the contextual follow-up tip after a fuzzy selection."""
     tip_cmd = _resource_tip_command(title)
     if tip_cmd:
-        console.print(
-            markup_text(f"\n[dim]Tip: use `{tip_cmd} <ID>` for details[/dim]")
-        )
+        console.print(markup_text(f"\n[dim]Tip: use `{tip_cmd} <ID>` for details[/dim]"))
 
 
-def _handle_fuzzy_pick_selection(
-    rows: list[dict[str, Any]], columns: list[tuple], title: str
-) -> bool:
+def _handle_fuzzy_pick_selection(rows: list[dict[str, Any]], columns: list[tuple], title: str) -> bool:
     """Handle fuzzy picker selection, returns True if selection was made."""
     picked = _try_fuzzy_pick(rows, columns, title)
     if picked is None:
@@ -855,9 +823,7 @@ def _handle_table_output(
     """Handle table output with paging."""
     content = _build_table_group(rows, columns, title)
     should_page = (
-        pager._should_page_output(len(rows), console.is_terminal and os.isatty(1))
-        if use_pager is None
-        else use_pager
+        pager._should_page_output(len(rows), console.is_terminal and os.isatty(1)) if use_pager is None else use_pager
     )
 
     if should_page:
@@ -989,9 +955,7 @@ def build_renderer(
 
     # Create the renderer instance
     renderer = RichStreamRenderer(
-        working_console.original_console
-        if isinstance(working_console, CapturingConsole)
-        else working_console,
+        working_console.original_console if isinstance(working_console, CapturingConsole) else working_console,
         cfg=renderer_cfg,
         verbose=verbose,
     )
@@ -1079,22 +1043,14 @@ def _resolve_by_name_multiple_with_select(matches: list[Any], select: int) -> An
     return matches[idx]
 
 
-def _resolve_by_name_multiple_fuzzy(
-    ctx: Any, ref: str, matches: list[Any], label: str
-) -> Any:
+def _resolve_by_name_multiple_fuzzy(ctx: Any, ref: str, matches: list[Any], label: str) -> Any:
     """Resolve multiple matches preferring the fuzzy picker interface."""
-    return handle_ambiguous_resource(
-        ctx, label.lower(), ref, matches, interface_preference="fuzzy"
-    )
+    return handle_ambiguous_resource(ctx, label.lower(), ref, matches, interface_preference="fuzzy")
 
 
-def _resolve_by_name_multiple_questionary(
-    ctx: Any, ref: str, matches: list[Any], label: str
-) -> Any:
+def _resolve_by_name_multiple_questionary(ctx: Any, ref: str, matches: list[Any], label: str) -> Any:
     """Resolve multiple matches preferring the questionary interface."""
-    return handle_ambiguous_resource(
-        ctx, label.lower(), ref, matches, interface_preference="questionary"
-    )
+    return handle_ambiguous_resource(ctx, label.lower(), ref, matches, interface_preference="questionary")
 
 
 def resolve_resource(
@@ -1139,9 +1095,7 @@ def resolve_resource(
         raise click.ClickException(f"{label} '{ref}' not found")
 
     # Find resources by name
-    _spinner_update(
-        spinner, f"[bold blue]Searching {label}s matching '{ref}'…[/bold blue]"
-    )
+    _spinner_update(spinner, f"[bold blue]Searching {label}s matching '{ref}'…[/bold blue]")
     matches = find_by_name(name=ref)
     if not matches:
         _spinner_stop(spinner)
@@ -1172,9 +1126,7 @@ def _handle_json_view_ambiguity(matches: list[Any]) -> Any:
     return matches[0]
 
 
-def _handle_questionary_ambiguity(
-    resource_type: str, ref: str, matches: list[Any]
-) -> Any:
+def _handle_questionary_ambiguity(resource_type: str, ref: str, matches: list[Any]) -> Any:
     """Handle ambiguity using questionary interactive interface."""
     if not (questionary and os.getenv("TERM") and os.isatty(0) and os.isatty(1)):
         raise click.ClickException("Interactive selection not available")
@@ -1187,7 +1139,10 @@ def _handle_questionary_ambiguity(
         f"Multiple {safe_resource_type}s match '{safe_ref}'. Pick one:",
         choices=[
             questionary.Choice(
-                title=f"{getattr(m, 'name', '—').replace('{', '{{').replace('}', '}}')} — {getattr(m, 'id', '').replace('{', '{{').replace('}', '}}')}",
+                title=(
+                    f"{getattr(m, 'name', '—').replace('{', '{{').replace('}', '}}')} — "
+                    f"{getattr(m, 'id', '').replace('{', '{{').replace('}', '}}')}"
+                ),
                 value=i,
             )
             for i, m in enumerate(matches)
@@ -1201,19 +1156,13 @@ def _handle_questionary_ambiguity(
     return matches[picked_idx]
 
 
-def _handle_fallback_numeric_ambiguity(
-    resource_type: str, ref: str, matches: list[Any]
-) -> Any:
+def _handle_fallback_numeric_ambiguity(resource_type: str, ref: str, matches: list[Any]) -> Any:
     """Handle ambiguity using numeric prompt fallback."""
     # Escape special characters for display
     safe_resource_type = resource_type.replace("{", "{{").replace("}", "}}")
     safe_ref = ref.replace("{", "{{").replace("}", "}}")
 
-    console.print(
-        markup_text(
-            f"[{WARNING_STYLE}]Multiple {safe_resource_type}s found matching '{safe_ref}':[/]"
-        )
-    )
+    console.print(markup_text(f"[{WARNING_STYLE}]Multiple {safe_resource_type}s found matching '{safe_ref}':[/]"))
     table = AIPTable(
         title=f"Select {safe_resource_type.title()}",
     )
@@ -1228,8 +1177,8 @@ def _handle_fallback_numeric_ambiguity(
     )
     try:
         choice = int(choice_str)
-    except ValueError:
-        raise click.ClickException("Invalid selection")
+    except ValueError as err:
+        raise click.ClickException("Invalid selection") from err
     if 1 <= choice <= len(matches):
         return matches[choice - 1]
     raise click.ClickException("Invalid selection")

glaip_sdk/cli/validators.py

@@ -42,7 +42,7 @@ def validate_agent_name_cli(name: str) -> str:
     try:
         return validate_agent_name(name)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_agent_instruction_cli(instruction: str) -> str:
@@ -60,7 +60,7 @@ def validate_agent_instruction_cli(instruction: str) -> str:
     try:
         return validate_agent_instruction(instruction)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_timeout_cli(timeout: int) -> int:
@@ -78,7 +78,7 @@ def validate_timeout_cli(timeout: int) -> int:
     try:
         return validate_timeout(timeout)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_tool_name_cli(name: str) -> str:
@@ -96,7 +96,7 @@ def validate_tool_name_cli(name: str) -> str:
     try:
         return validate_tool_name(name)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_mcp_name_cli(name: str) -> str:
@@ -114,7 +114,7 @@ def validate_mcp_name_cli(name: str) -> str:
     try:
         return validate_mcp_name(name)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_file_path_cli(file_path: str | Path, must_exist: bool = True) -> Path:
@@ -133,7 +133,7 @@ def validate_file_path_cli(file_path: str | Path, must_exist: bool = True) -> Pa
     try:
         return validate_file_path(file_path, must_exist)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_directory_path_cli(dir_path: str | Path, must_exist: bool = True) -> Path:
@@ -152,7 +152,7 @@ def validate_directory_path_cli(dir_path: str | Path, must_exist: bool = True) -
     try:
         return validate_directory_path(dir_path, must_exist)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_url_cli(url: str) -> str:
@@ -170,7 +170,7 @@ def validate_url_cli(url: str) -> str:
     try:
         return validate_url(url)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_api_key_cli(api_key: str) -> str:
@@ -188,7 +188,7 @@ def validate_api_key_cli(api_key: str) -> str:
     try:
         return validate_api_key(api_key)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def coerce_timeout_cli(value: int | float | str) -> int:
@@ -206,7 +206,7 @@ def coerce_timeout_cli(value: int | float | str) -> int:
     try:
         return coerce_timeout(value)
     except ValueError as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def validate_name_uniqueness_cli(
@@ -230,8 +230,7 @@ def validate_name_uniqueness_cli(
         existing = finder_func(name=name)
         if existing:
             raise click.ClickException(
-                f"A {resource_type.lower()} named '{name}' already exists. "
-                "Please choose a unique name."
+                f"A {resource_type.lower()} named '{name}' already exists. Please choose a unique name."
             )
     except click.ClickException:
         raise

glaip_sdk/client/_agent_payloads.py

@@ -348,9 +348,7 @@ class AgentCreateRequest:
             payload["tool_configs"] = _copy_structure(self.tool_configs)
 
         effective_agent_config = _sanitize_agent_config(self.agent_config)
-        effective_agent_config = _merge_execution_timeout(
-            effective_agent_config, self.timeout
-        )
+        effective_agent_config = _merge_execution_timeout(effective_agent_config, self.timeout)
         if effective_agent_config:
             payload["agent_config"] = effective_agent_config
 
@@ -407,31 +405,19 @@ __all__ = [
 ]
 
 
-def _build_base_update_payload(
-    request: AgentUpdateRequest, current_agent: Any
-) -> dict[str, Any]:
+def _build_base_update_payload(request: AgentUpdateRequest, current_agent: Any) -> dict[str, Any]:
     return {
-        "name": request.name.strip()
-        if request.name is not None
-        else getattr(current_agent, "name", None),
+        "name": request.name.strip() if request.name is not None else getattr(current_agent, "name", None),
         "instruction": request.instruction.strip()
         if request.instruction is not None
         else getattr(current_agent, "instruction", None),
-        "type": request.agent_type
-        or getattr(current_agent, "type", None)
-        or DEFAULT_AGENT_TYPE,
-        "framework": request.framework
-        or getattr(current_agent, "framework", None)
-        or DEFAULT_AGENT_FRAMEWORK,
-        "version": request.version
-        or getattr(current_agent, "version", None)
-        or DEFAULT_AGENT_VERSION,
+        "type": request.agent_type or getattr(current_agent, "type", None) or DEFAULT_AGENT_TYPE,
+        "framework": request.framework or getattr(current_agent, "framework", None) or DEFAULT_AGENT_FRAMEWORK,
+        "version": request.version or getattr(current_agent, "version", None) or DEFAULT_AGENT_VERSION,
     }
 
 
-def _resolve_update_language_model_fields(
-    request: AgentUpdateRequest, current_agent: Any
-) -> dict[str, Any]:
+def _resolve_update_language_model_fields(request: AgentUpdateRequest, current_agent: Any) -> dict[str, Any]:
     fields = resolve_language_model_fields(
         model=request.model,
         language_model_id=request.language_model_id,
@@ -443,9 +429,7 @@ def _resolve_update_language_model_fields(
     return fields
 
 
-def _collect_optional_update_fields(
-    request: AgentUpdateRequest, current_agent: Any
-) -> dict[str, Any]:
+def _collect_optional_update_fields(request: AgentUpdateRequest, current_agent: Any) -> dict[str, Any]:
     result: dict[str, Any] = {}
 
     for field_name, value in (
@@ -483,9 +467,7 @@ def _collect_optional_update_fields(
     return result
 
 
-def _collect_relationship_fields(
-    request: AgentUpdateRequest, current_agent: Any
-) -> dict[str, Any]:
+def _collect_relationship_fields(request: AgentUpdateRequest, current_agent: Any) -> dict[str, Any]:
     return {
         "tools": _resolve_relation_ids(request.tools, current_agent, "tools"),
         "agents": _resolve_relation_ids(request.agents, current_agent, "agents"),
@@ -493,9 +475,7 @@ def _collect_relationship_fields(
     }
 
 
-def _resolve_agent_config_update(
-    request: AgentUpdateRequest, current_agent: Any
-) -> dict[str, Any] | None:
+def _resolve_agent_config_update(request: AgentUpdateRequest, current_agent: Any) -> dict[str, Any] | None:
     effective_agent_config = _sanitize_agent_config(request.agent_config)
     if effective_agent_config is not None:
         return effective_agent_config