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

glaip_sdk/_version.py

@@ -47,9 +47,7 @@ def _try_get_dev_version() -> str | None:
             return None
 
         data = _toml.loads(pyproject.read_text(encoding="utf-8"))
-        ver = data.get("project", {}).get("version") or data.get("tool", {}).get(
-            "poetry", {}
-        ).get("version")
+        ver = data.get("project", {}).get("version") or data.get("tool", {}).get("poetry", {}).get("version")
         if isinstance(ver, str) and ver:
             return ver
     except Exception:

glaip_sdk/branding.py

@@ -104,9 +104,7 @@ GDP Labs AI Agents Package
     @staticmethod
     def _make_console() -> Console:
         # Respect NO_COLOR/AIP_NO_COLOR environment variables
-        no_color_env = (
-            os.getenv("NO_COLOR") is not None or os.getenv("AIP_NO_COLOR") is not None
-        )
+        no_color_env = os.getenv("NO_COLOR") is not None or os.getenv("AIP_NO_COLOR") is not None
         if no_color_env:
             color_system = None
             no_color = True
@@ -187,9 +185,7 @@ GDP Labs AI Agents Package
         self.console.print(banner)
 
     @classmethod
-    def create_from_sdk(
-        cls, sdk_version: str | None = None, package_name: str | None = None
-    ) -> AIPBranding:
+    def create_from_sdk(cls, sdk_version: str | None = None, package_name: str | None = None) -> AIPBranding:
         """Create AIPBranding instance from SDK package information.
 
         Args:

glaip_sdk/cli/agent_config.py

@@ -16,9 +16,7 @@ from glaip_sdk.utils.agent_config import (
 from glaip_sdk.utils.import_export import merge_import_with_cli_args
 
 
-def merge_agent_config_with_cli_args(
-    import_data: dict[str, Any], cli_args: dict[str, Any]
-) -> dict[str, Any]:
+def merge_agent_config_with_cli_args(import_data: dict[str, Any], cli_args: dict[str, Any]) -> dict[str, Any]:
     """Merge imported agent data with CLI arguments, preferring CLI args.
 
     This is a CLI-specific wrapper that handles agent-specific merging logic.
@@ -35,9 +33,7 @@ def merge_agent_config_with_cli_args(
         - Handles agent-specific fields like tools and agents arrays
         - Preserves agent configuration structure
     """
-    return merge_import_with_cli_args(
-        import_data, cli_args, array_fields=["tools", "agents"]
-    )
+    return merge_import_with_cli_args(import_data, cli_args, array_fields=["tools", "agents"])
 
 
 def resolve_agent_language_model_selection(

glaip_sdk/cli/auth.py

@@ -65,9 +65,7 @@ def prepare_authentication_export(
 
     # Handle bearer-token authentication
     if auth_type == "bearer-token":
-        return _prepare_bearer_token_auth(
-            auth, prompt_for_secrets, placeholder, console
-        )
+        return _prepare_bearer_token_auth(auth, prompt_for_secrets, placeholder, console)
 
     # Handle api-key authentication
     if auth_type == "api-key":
@@ -75,9 +73,7 @@ def prepare_authentication_export(
 
     # Handle custom-header authentication
     if auth_type == "custom-header":
-        return _prepare_custom_header_auth(
-            auth, prompt_for_secrets, placeholder, console
-        )
+        return _prepare_custom_header_auth(auth, prompt_for_secrets, placeholder, console)
 
     # Unknown auth type - return as-is but strip helper metadata
     result = auth.copy()
@@ -85,9 +81,7 @@ def prepare_authentication_export(
     return result
 
 
-def _get_token_value(
-    prompt_for_secrets: bool, placeholder: str, console: Console
-) -> str:
+def _get_token_value(prompt_for_secrets: bool, placeholder: str, console: Console) -> str:
     """Get bearer token value either by prompting or using a placeholder.
 
     Args:
@@ -109,10 +103,7 @@ def _get_token_value(
         )
 
     if not click.get_text_stream("stdin").isatty():
-        console.print(
-            f"[{WARNING_STYLE}]⚠️  Non-interactive mode: "
-            "using placeholder for bearer token[/]"
-        )
+        console.print(f"[{WARNING_STYLE}]⚠️  Non-interactive mode: using placeholder for bearer token[/]")
     return placeholder
 
 
@@ -220,16 +211,11 @@ def _get_api_key_value(
         )
 
     if not click.get_text_stream("stdin").isatty():
-        console.print(
-            f"[{WARNING_STYLE}]⚠️  Non-interactive mode: "
-            f"using placeholder for API key '{key_name}'[/]"
-        )
+        console.print(f"[{WARNING_STYLE}]⚠️  Non-interactive mode: using placeholder for API key '{key_name}'[/]")
     return placeholder
 
 
-def _build_api_key_headers(
-    auth: dict[str, Any], key_name: str | None, key_value: str
-) -> dict[str, str]:
+def _build_api_key_headers(auth: dict[str, Any], key_name: str | None, key_value: str) -> dict[str, str]:
     """Build headers for API key authentication.
 
     Args:
@@ -270,9 +256,7 @@ def _prepare_api_key_auth(
 
     # Capture or use placeholder for value
     if not has_valid_value:
-        key_value = _get_api_key_value(
-            key_name, prompt_for_secrets, placeholder, console
-        )
+        key_value = _get_api_key_value(key_name, prompt_for_secrets, placeholder, console)
 
     # Check if original had headers structure
     if "headers" in auth or "header_keys" in auth:
@@ -317,9 +301,7 @@ def _prepare_custom_header_auth(
     return {"type": "custom-header", "headers": headers}
 
 
-def _extract_header_names(
-    existing_headers: Mapping[str, Any] | None, header_keys: Iterable[str] | None
-) -> list[str]:
+def _extract_header_names(existing_headers: Mapping[str, Any] | None, header_keys: Iterable[str] | None) -> list[str]:
     """Extract the list of header names to process.
 
     Args:
@@ -376,9 +358,7 @@ def _prompt_or_placeholder(
         )
 
     if not click.get_text_stream("stdin").isatty():
-        console.print(
-            f"[{WARNING_STYLE}]⚠️  Non-interactive mode: using placeholder for header '{name}'[/]"
-        )
+        console.print(f"[{WARNING_STYLE}]⚠️  Non-interactive mode: using placeholder for header '{name}'[/]")
     return placeholder
 
 
@@ -450,7 +430,8 @@ def _prompt_secret_with_placeholder(
     tip = command_hint(tip_cli_command, tip_slash_command)
     if tip:
         console.print(
-            f"[{HINT_PREFIX_STYLE}]Tip:[/] use {format_command_hint(tip) or tip} later if you want to update these credentials."
+            f"[{HINT_PREFIX_STYLE}]Tip:[/] use {format_command_hint(tip) or tip} later "
+            "if you want to update these credentials."
         )
 
     attempts = 0

glaip_sdk/cli/commands/agents.py

@@ -247,9 +247,7 @@ def _resolve_resources_by_name(
         if not matches:
             raise click.ClickException(f"{label} not found: {ref}")
         if len(matches) > 1:
-            raise click.ClickException(
-                f"Multiple {resource_type}s named '{ref}'. Use ID instead."
-            )
+            raise click.ClickException(f"Multiple {resource_type}s named '{ref}'. Use ID instead.")
         out.append(str(matches[0].id))
     return out
 
@@ -395,15 +393,9 @@ def _resolve_agent(
 
 
 @agents_group.command(name="list")
-@click.option(
-    "--simple", is_flag=True, help="Show simple table without interactive picker"
-)
-@click.option(
-    "--type", "agent_type", help="Filter by agent type (config, code, a2a, langflow)"
-)
-@click.option(
-    "--framework", help="Filter by framework (langchain, langgraph, google_adk)"
-)
+@click.option("--simple", is_flag=True, help="Show simple table without interactive picker")
+@click.option("--type", "agent_type", help="Filter by agent type (config, code, a2a, langflow)")
+@click.option("--framework", help="Filter by framework (langchain, langgraph, google_adk)")
 @click.option("--name", help="Filter by partial name match (case-insensitive)")
 @click.option("--version", help="Filter by exact version match")
 @click.option(
@@ -482,15 +474,12 @@ def list_agents(
             f"{ICON_AGENT} Available Agents",
             columns,
             transform_agent,
-            skip_picker=simple
-            or any(
-                param is not None for param in (agent_type, framework, name, version)
-            ),
+            skip_picker=simple or any(param is not None for param in (agent_type, framework, name, version)),
             use_pager=False,
         )
 
     except Exception as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 @agents_group.command()
@@ -515,9 +504,7 @@ def get(ctx: Any, agent_ref: str, select: int | None, export: str | None) -> Non
         client = get_client(ctx)
 
         # Resolve agent with ambiguity handling - use questionary interface for traditional UX
-        agent = _resolve_agent(
-            ctx, client, agent_ref, select, interface_preference="questionary"
-        )
+        agent = _resolve_agent(ctx, client, agent_ref, select, interface_preference="questionary")
 
         # Handle export option
         if export:
@@ -536,22 +523,19 @@ def get(ctx: Any, agent_ref: str, select: int | None, export: str | None) -> Non
             except Exception as e:
                 handle_rich_output(
                     ctx,
-                    markup_text(
-                        f"[{WARNING_STYLE}]⚠️  Could not fetch full agent details: {e}[/]"
-                    ),
+                    markup_text(f"[{WARNING_STYLE}]⚠️  Could not fetch full agent details: {e}[/]"),
                 )
                 handle_rich_output(
                     ctx,
-                    markup_text(
-                        f"[{WARNING_STYLE}]⚠️  Proceeding with available data[/]"
-                    ),
+                    markup_text(f"[{WARNING_STYLE}]⚠️  Proceeding with available data[/]"),
                 )
 
             export_resource_to_file(agent, export_path, detected_format)
             handle_rich_output(
                 ctx,
                 markup_text(
-                    f"[{SUCCESS_STYLE}]✅ Complete agent configuration exported to: {export_path} (format: {detected_format})[/]"
+                    f"[{SUCCESS_STYLE}]✅ Complete agent configuration exported to: {export_path} "
+                    f"(format: {detected_format})[/]"
                 ),
             )
 
@@ -562,7 +546,7 @@ def get(ctx: Any, agent_ref: str, select: int | None, export: str | None) -> Non
         handle_rich_output(ctx, display_agent_run_suggestions(agent))
 
     except Exception as e:
-        raise click.ClickException(str(e))
+        raise click.ClickException(str(e)) from e
 
 
 def _validate_run_input(input_option: str | None, input_text: str | None) -> str:
@@ -570,9 +554,7 @@ def _validate_run_input(input_option: str | None, input_text: str | None) -> str
     final_input_text = input_option if input_option else input_text
 
     if not final_input_text:
-        raise click.ClickException(
-            "Input text is required. Use either positional argument or --input option."
-        )
+        raise click.ClickException("Input text is required. Use either positional argument or --input option.")
 
     return final_input_text
 
@@ -584,8 +566,8 @@ def _parse_chat_history(chat_history: str | None) -> list[dict[str, Any]] | None
 
     try:
         return json.loads(chat_history)
-    except json.JSONDecodeError:
-        raise click.ClickException("Invalid JSON in chat history")
+    except json.JSONDecodeError as err:
+        raise click.ClickException("Invalid JSON in chat history") from err
 
 
 def _setup_run_renderer(ctx: Any, save: str | None, verbose: bool) -> Any:
@@ -665,9 +647,7 @@ def _save_run_transcript(save: str | None, result: Any, working_console: Any) ->
     if ext == "json":
         save_data = {
             "output": result or "",
-            "full_debug_output": getattr(
-                working_console, "get_captured_output", lambda: ""
-            )(),
+            "full_debug_output": getattr(working_console, "get_captured_output", lambda: "")(),
             "timestamp": "captured during agent execution",
         }
         content = json.dumps(save_data, indent=2)
@@ -680,9 +660,7 @@ def _save_run_transcript(save: str | None, result: Any, working_console: Any) ->
 
     with open(save, "w", encoding="utf-8") as f:
         f.write(content)
-    print_markup(
-        f"[{SUCCESS_STYLE}]Full debug output saved to: {save}[/]", console=console
-    )
+    print_markup(f"[{SUCCESS_STYLE}]Full debug output saved to: {save}[/]", console=console)
 
 
 @agents_group.command()
@@ -745,9 +723,7 @@ def run(
 
     try:
         client = get_client(ctx)
-        agent = _resolve_agent(
-            ctx, client, agent_ref, select, interface_preference="fuzzy"
-        )
+        agent = _resolve_agent(ctx, client, agent_ref, select, interface_preference="fuzzy")
 
         parsed_chat_history = _parse_chat_history(chat_history)
         renderer, working_console = _setup_run_renderer(ctx, save, verbose)
@@ -796,7 +772,7 @@ def run(
     except AgentTimeoutError as e:
         error_msg = str(e)
         handle_json_output(ctx, error=Exception(error_msg))
-        raise click.ClickException(error_msg)
+        raise click.ClickException(error_msg) from e
     except Exception as e:
         _handle_command_exception(ctx, e)
 
@@ -885,16 +861,12 @@ def _extract_and_validate_fields(
     if not name:
         raise click.ClickException("Agent name is required (--name or --import)")
     if not instruction:
-        raise click.ClickException(
-            "Agent instruction is required (--instruction or --import)"
-        )
+        raise click.ClickException("Agent instruction is required (--instruction or --import)")
 
     return name, instruction, model, tools, agents, mcps, timeout
 
 
-def _validate_and_coerce_fields(
-    name: str, instruction: str, timeout: Any
-) -> tuple[str, str, Any]:
+def _validate_and_coerce_fields(name: str, instruction: str, timeout: Any) -> tuple[str, str, Any]:
     """Validate and coerce field values."""
     name = validate_agent_name(name)
     instruction = validate_agent_instruction(instruction)
@@ -905,19 +877,11 @@ def _validate_and_coerce_fields(
     return name, instruction, timeout
 
 
-def _resolve_resources(
-    client: Any, tools: tuple, agents: tuple, mcps: tuple
-) -> tuple[list, list, list]:
+def _resolve_resources(client: Any, tools: tuple, agents: tuple, mcps: tuple) -> tuple[list, list, list]:
     """Resolve tool, agent, and MCP references."""
-    resolved_tools = _resolve_resources_by_name(
-        client, tools, "tool", client.find_tools, "Tool"
-    )
-    resolved_agents = _resolve_resources_by_name(
-        client, agents, "agent", client.find_agents, "Agent"
-    )
-    resolved_mcps = _resolve_resources_by_name(
-        client, mcps, "mcp", client.find_mcps, "MCP"
-    )
+    resolved_tools = _resolve_resources_by_name(client, tools, "tool", client.find_tools, "Tool")
+    resolved_agents = _resolve_resources_by_name(client, agents, "agent", client.find_agents, "Agent")
+    resolved_mcps = _resolve_resources_by_name(client, mcps, "mcp", client.find_mcps, "MCP")
 
     return resolved_tools, resolved_agents, resolved_mcps
 
@@ -944,16 +908,12 @@ def _build_create_kwargs(
     }
 
     # Handle language model selection
-    lm_selection_dict, should_strip_lm_identity = resolve_language_model_selection(
-        merged_data, model
-    )
+    lm_selection_dict, should_strip_lm_identity = resolve_language_model_selection(merged_data, model)
     create_kwargs.update(lm_selection_dict)
 
     # Handle import file specific logic
     if import_file:
-        _add_import_file_attributes(
-            create_kwargs, merged_data, should_strip_lm_identity
-        )
+        _add_import_file_attributes(create_kwargs, merged_data, should_strip_lm_identity)
 
     return create_kwargs
 
@@ -999,12 +959,7 @@ def _get_language_model_display_name(agent: Any, model: str | None) -> str:
     lm_display = getattr(agent, "model", None)
     if not lm_display:
         cfg = getattr(agent, "agent_config", {}) or {}
-        lm_display = (
-            cfg.get("lm_name")
-            or cfg.get("model")
-            or model
-            or f"{DEFAULT_MODEL} (backend default)"
-        )
+        lm_display = cfg.get("lm_name") or cfg.get("model") or model or f"{DEFAULT_MODEL} (backend default)"
     return lm_display
 
 
@@ -1039,7 +994,7 @@ def _handle_command_exception(ctx: Any, e: Exception) -> None:
     handle_json_output(ctx, error=e)
     if get_ctx_value(ctx, "view") != "json":
         print_api_error(e)
-    raise click.ClickException(str(e))
+    raise click.ClickException(str(e)) from e
 
 
 def _handle_creation_exception(ctx: Any, e: Exception) -> None:
@@ -1093,13 +1048,9 @@ def create(
 
         # Handle import file or CLI args
         if import_file:
-            merged_data = _handle_import_file_logic(
-                import_file, model, name, instruction, tools, agents, mcps, timeout
-            )
+            merged_data = _handle_import_file_logic(import_file, model, name, instruction, tools, agents, mcps, timeout)
         else:
-            merged_data = _build_cli_args_data(
-                name, instruction, model, tools, agents, mcps, timeout
-            )
+            merged_data = _build_cli_args_data(name, instruction, model, tools, agents, mcps, timeout)
 
         # Extract and validate fields
         (
@@ -1111,14 +1062,10 @@ def create(
             mcps,
             timeout,
         ) = _extract_and_validate_fields(merged_data)
-        name, instruction, timeout = _validate_and_coerce_fields(
-            name, instruction, timeout
-        )
+        name, instruction, timeout = _validate_and_coerce_fields(name, instruction, timeout)
 
         # Resolve resources
-        resolved_tools, resolved_agents, resolved_mcps = _resolve_resources(
-            client, tools, agents, mcps
-        )
+        resolved_tools, resolved_agents, resolved_mcps = _resolve_resources(client, tools, agents, mcps)
 
         # Build create kwargs
         create_kwargs = _build_create_kwargs(
@@ -1148,7 +1095,7 @@ def _get_agent_for_update(client: Any, agent_id: str) -> Any:
     try:
         return client.agents.get_agent_by_id(agent_id)
     except Exception as e:
-        raise click.ClickException(f"Agent with ID '{agent_id}' not found: {e}")
+        raise click.ClickException(f"Agent with ID '{agent_id}' not found: {e}") from e
 
 
 def _handle_update_import_file(
@@ -1230,16 +1177,12 @@ def _handle_update_import_config(
     if not import_file:
         return
 
-    lm_selection, should_strip_lm_identity = resolve_language_model_selection(
-        merged_data, None
-    )
+    lm_selection, should_strip_lm_identity = resolve_language_model_selection(merged_data, None)
     update_data.update(lm_selection)
 
     raw_cfg = merged_data.get("agent_config") if isinstance(merged_data, dict) else None
     if isinstance(raw_cfg, dict):
-        update_data["agent_config"] = sanitize_agent_config(
-            raw_cfg, strip_lm_identity=should_strip_lm_identity
-        )
+        update_data["agent_config"] = sanitize_agent_config(raw_cfg, strip_lm_identity=should_strip_lm_identity)
 
     excluded_fields = {
         "name",
@@ -1308,13 +1251,9 @@ def update(
             agents,
             mcps,
             timeout,
-        ) = _handle_update_import_file(
-            import_file, name, instruction, tools, agents, mcps, timeout
-        )
+        ) = _handle_update_import_file(import_file, name, instruction, tools, agents, mcps, timeout)
 
-        update_data = _build_update_data(
-            name, instruction, tools, agents, mcps, timeout
-        )
+        update_data = _build_update_data(name, instruction, tools, agents, mcps, timeout)
 
         if merged_data:
             _handle_update_import_config(import_file, merged_data, update_data)
@@ -1352,7 +1291,7 @@ def delete(ctx: Any, agent_id: str, yes: bool) -> None:
         try:
             agent = client.agents.get_agent_by_id(agent_id)
         except Exception as e:
-            raise click.ClickException(f"Agent with ID '{agent_id}' not found: {e}")
+            raise click.ClickException(f"Agent with ID '{agent_id}' not found: {e}") from e
 
         # Confirm deletion when not forced
         if not yes and not display_confirmation_prompt("Agent", agent.name):
@@ -1384,9 +1323,7 @@ def delete(ctx: Any, agent_id: str, yes: bool) -> None:
     "--base-url",
     help="Custom LangFlow server base URL (overrides LANGFLOW_BASE_URL env var)",
 )
-@click.option(
-    "--api-key", help="Custom LangFlow API key (overrides LANGFLOW_API_KEY env var)"
-)
+@click.option("--api-key", help="Custom LangFlow API key (overrides LANGFLOW_API_KEY env var)")
 @output_flags()
 @click.pass_context
 def sync_langflow(ctx: Any, base_url: str | None, api_key: str | None) -> None:
@@ -1415,15 +1352,16 @@ def sync_langflow(ctx: Any, base_url: str | None, api_key: str | None) -> None:
         # Show success message for non-JSON output
         if get_ctx_value(ctx, "view") != "json":
             # Extract some useful info from the result
-            success_count = result.get("data", {}).get("created_count", 0) + result.get(
-                "data", {}
-            ).get("updated_count", 0)
+            success_count = result.get("data", {}).get("created_count", 0) + result.get("data", {}).get(
+                "updated_count", 0
+            )
             total_count = result.get("data", {}).get("total_processed", 0)
 
             handle_rich_output(
                 ctx,
                 markup_text(
-                    f"[{SUCCESS_STYLE}]✅ Successfully synced {success_count} LangFlow agents ({total_count} total processed)[/]"
+                    f"[{SUCCESS_STYLE}]✅ Successfully synced {success_count} LangFlow agents "
+                    f"({total_count} total processed)[/]"
                 ),
             )
 

glaip_sdk/cli/commands/configure.py

@@ -57,9 +57,7 @@ def set_config(key: str, value: str) -> None:
     valid_keys = ["api_url", "api_key"]
 
     if key not in valid_keys:
-        console.print(
-            f"[{ERROR_STYLE}]Error: Invalid key '{key}'. Valid keys are: {', '.join(valid_keys)}[/]"
-        )
+        console.print(f"[{ERROR_STYLE}]Error: Invalid key '{key}'. Valid keys are: {', '.join(valid_keys)}[/]")
         raise click.ClickException(f"Invalid configuration key: {key}")
 
     config = load_config()
@@ -67,9 +65,7 @@ def set_config(key: str, value: str) -> None:
     save_config(config)
 
     if key == "api_key":
-        console.print(
-            Text(f"✅ Set {key} = {_mask_api_key(value)}", style=SUCCESS_STYLE)
-        )
+        console.print(Text(f"✅ Set {key} = {_mask_api_key(value)}", style=SUCCESS_STYLE))
     else:
         console.print(Text(f"✅ Set {key} = {value}", style=SUCCESS_STYLE))
 
@@ -81,9 +77,7 @@ def get_config(key: str) -> None:
     config = load_config()
 
     if key not in config:
-        console.print(
-            markup_text(f"[{WARNING_STYLE}]Configuration key '{key}' not found.[/]")
-        )
+        console.print(markup_text(f"[{WARNING_STYLE}]Configuration key '{key}' not found.[/]"))
         raise click.ClickException(f"Configuration key not found: {key}")
 
     value = config[key]
@@ -101,9 +95,7 @@ def unset_config(key: str) -> None:
     config = load_config()
 
     if key not in config:
-        console.print(
-            markup_text(f"[{WARNING_STYLE}]Configuration key '{key}' not found.[/]")
-        )
+        console.print(markup_text(f"[{WARNING_STYLE}]Configuration key '{key}' not found.[/]"))
         return
 
     del config[key]
@@ -128,9 +120,7 @@ def reset_config(force: bool) -> None:
 
     if not file_exists and not config_data:
         console.print(f"[{WARNING_STYLE}]No configuration found to reset.[/]")
-        console.print(
-            Text("✅ Configuration reset (nothing to remove).", style=SUCCESS_STYLE)
-        )
+        console.print(Text("✅ Configuration reset (nothing to remove).", style=SUCCESS_STYLE))
         return
 
     if file_exists:
@@ -188,9 +178,7 @@ def _mask_api_key(value: str | None) -> str:
 def _print_missing_config_hint() -> None:
     hint = command_hint("config configure", slash_command="login")
     if hint:
-        console.print(
-            f"[{WARNING_STYLE}]No configuration found.[/] Run {format_command_hint(hint) or hint} to set up."
-        )
+        console.print(f"[{WARNING_STYLE}]No configuration found.[/] Run {format_command_hint(hint) or hint} to set up.")
     else:
         console.print(f"[{WARNING_STYLE}]No configuration found.[/]")
 
@@ -208,9 +196,7 @@ def _render_config_table(config: dict[str, str]) -> None:
 
 
 def _render_configuration_header() -> None:
-    branding = AIPBranding.create_from_sdk(
-        sdk_version=_SDK_VERSION, package_name="glaip-sdk"
-    )
+    branding = AIPBranding.create_from_sdk(sdk_version=_SDK_VERSION, package_name="glaip-sdk")
     heading = "[bold]>_ GDP Labs AI Agents Package (AIP CLI)[/bold]"
     console.print(heading)
     console.print()
@@ -249,9 +235,7 @@ def _prompt_api_key(config: dict[str, str]) -> None:
 
 def _save_configuration(config: dict[str, str]) -> None:
     save_config(config)
-    console.print(
-        Text(f"\n✅ Configuration saved to: {CONFIG_FILE}", style=SUCCESS_STYLE)
-    )
+    console.print(Text(f"\n✅ Configuration saved to: {CONFIG_FILE}", style=SUCCESS_STYLE))
 
 
 def _test_and_report_connection(config: dict[str, str]) -> None:
@@ -274,17 +258,13 @@ def _test_and_report_connection(config: dict[str, str]) -> None:
                     style=WARNING_STYLE,
                 )
             )
-            console.print(
-                "   You may need to check your API permissions or network access"
-            )
+            console.print("   You may need to check your API permissions or network access")
     except Exception as exc:
         console.print(Text(f"❌ Connection failed: {exc}"))
         console.print("   Please check your API URL and key")
         hint_status = command_hint("status", slash_command="status")
         if hint_status:
-            console.print(
-                f"   You can run {format_command_hint(hint_status) or hint_status} later to test again"
-            )
+            console.print(f"   You can run {format_command_hint(hint_status) or hint_status} later to test again")
     finally:
         if client is not None:
             client.close()
@@ -294,11 +274,7 @@ def _print_post_configuration_hints() -> None:
     console.print("\n💡 You can now use AIP CLI commands!")
     hint_status = command_hint("status", slash_command="status")
     if hint_status:
-        console.print(
-            f"   • Run {format_command_hint(hint_status) or hint_status} to check connection"
-        )
+        console.print(f"   • Run {format_command_hint(hint_status) or hint_status} to check connection")
     hint_agents = command_hint("agents list", slash_command="agents")
     if hint_agents:
-        console.print(
-            f"   • Run {format_command_hint(hint_agents) or hint_agents} to see your agents"
-        )
+        console.print(f"   • Run {format_command_hint(hint_agents) or hint_agents} to see your agents")