glaip-sdk 0.5.3__py3-none-any.whl → 0.6.0__py3-none-any.whl

glaip_sdk/utils/sync.py

@@ -0,0 +1,142 @@
+"""Agent and tool synchronization (create/update) operations.
+
+This module provides convenience functions for tool classes that need bundling.
+
+For direct upsert operations, use the client methods:
+    - client.agents.upsert_agent(identifier, **kwargs)
+    - client.tools.upsert_tool(identifier, code, **kwargs)
+    - client.mcps.upsert_mcp(identifier, **kwargs)
+
+Authors:
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from glaip_sdk.utils.bundler import ToolBundler
+from glaip_sdk.utils.import_resolver import load_class
+from gllm_core.utils import LoggerManager
+
+if TYPE_CHECKING:
+    from glaip_sdk.models import Agent, Tool
+
+logger = LoggerManager().get_logger(__name__)
+
+
+def _extract_tool_name(tool_class: Any) -> str:
+    """Extract tool name from a class, handling Pydantic v2 models."""
+    # Direct attribute access (works for non-Pydantic classes)
+    if hasattr(tool_class, "name"):
+        name = getattr(tool_class, "name", None)
+        if isinstance(name, str):
+            return name
+
+    # Pydantic v2 model - check model_fields
+    if hasattr(tool_class, "model_fields"):
+        model_fields = getattr(tool_class, "model_fields", {})
+        if "name" in model_fields:
+            field_info = model_fields["name"]
+            if hasattr(field_info, "default") and isinstance(field_info.default, str):
+                return field_info.default
+
+    raise ValueError(f"Cannot extract name from tool class: {tool_class}")
+
+
+def _extract_tool_description(tool_class: Any) -> str:
+    """Extract tool description from a class, handling Pydantic v2 models."""
+    # Direct attribute access
+    if hasattr(tool_class, "description"):
+        desc = getattr(tool_class, "description", None)
+        if isinstance(desc, str):
+            return desc
+
+    # Pydantic v2 model - check model_fields
+    if hasattr(tool_class, "model_fields"):
+        model_fields = getattr(tool_class, "model_fields", {})
+        if "description" in model_fields:
+            field_info = model_fields["description"]
+            if hasattr(field_info, "default") and isinstance(field_info.default, str):
+                return field_info.default
+
+    return ""
+
+
+def update_or_create_tool(tool_ref: Any) -> Tool:
+    """Create or update a tool from a tool class with bundled source code.
+
+    This function takes a tool class (LangChain BaseTool), bundles its source
+    code with inlined imports, and creates/updates it in the backend.
+
+    Args:
+        tool_ref: A tool class (LangChain BaseTool subclass) or import path string.
+
+    Returns:
+        The created or updated tool.
+
+    Example:
+        >>> from glaip_sdk.utils.sync import update_or_create_tool
+        >>> from my_tools import WeatherAPITool
+        >>> tool = update_or_create_tool(WeatherAPITool)
+    """
+    from glaip_sdk.utils.client import get_client  # noqa: PLC0415
+
+    client = get_client()
+
+    # Handle string import path
+    if isinstance(tool_ref, str):
+        tool_class = load_class(tool_ref)
+    else:
+        tool_class = tool_ref
+
+    # Get tool info - handle Pydantic v2 model classes
+    tool_name = _extract_tool_name(tool_class)
+    tool_description = _extract_tool_description(tool_class)
+
+    # Bundle source code
+    bundler = ToolBundler(tool_class)
+    bundled_source = bundler.bundle()
+
+    logger.info("Tool info: name='%s', description='%s...'", tool_name, tool_description[:50])
+    logger.info("Bundled source code: %d characters", len(bundled_source))
+
+    # Use client's upsert method
+    return client.tools.upsert_tool(
+        tool_name,
+        code=bundled_source,
+        description=tool_description,
+    )
+
+
+def update_or_create_agent(agent_config: dict[str, Any]) -> Agent:
+    """Create or update an agent from configuration.
+
+    Args:
+        agent_config: Agent configuration dictionary containing:
+            - name (str): Agent name (required)
+            - description (str): Agent description
+            - instruction (str): Agent instruction
+            - tools (list, optional): List of tool IDs
+            - agents (list, optional): List of sub-agent IDs
+            - metadata (dict, optional): Additional metadata
+
+    Returns:
+        The created or updated agent.
+
+    Example:
+        >>> from glaip_sdk.utils.sync import update_or_create_agent
+        >>> config = {
+        ...     "name": "weather_reporter",
+        ...     "description": "Weather reporting agent",
+        ...     "instruction": "You are a weather reporter.",
+        ... }
+        >>> agent = update_or_create_agent(config)
+    """
+    from glaip_sdk.utils.client import get_client  # noqa: PLC0415
+
+    client = get_client()
+    agent_name = agent_config.pop("name")
+
+    # Use client's upsert method
+    return client.agents.upsert_agent(agent_name, **agent_config)

{glaip_sdk-0.5.3.dist-info → glaip_sdk-0.6.0.dist-info}/METADATA

@@ -1,18 +1,20 @@
 Metadata-Version: 2.1
 Name: glaip-sdk
-Version: 0.5.3
+Version: 0.6.0
 Summary: Python SDK for GL AIP (GDP Labs AI Agent Package) - Simplified CLI Design
 License: MIT
 Author: Raymond Christopher
 Author-email: raymond.christopher@gdplabs.id
-Requires-Python: >=3.10,<4.0
+Requires-Python: >=3.11,<3.14
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: click (>=8.2.0,<8.3.0)
+Requires-Dist: gllm-core-binary (>=0.1.0)
+Requires-Dist: gllm-tools-binary (>=0.1.3)
 Requires-Dist: httpx (>=0.28.1)
+Requires-Dist: langchain-core (>=0.3.0)
 Requires-Dist: packaging (>=23.2)
 Requires-Dist: pydantic (>=2.0.0)
 Requires-Dist: python-dotenv (>=1.1.1,<2.0.0)

{glaip_sdk-0.5.3.dist-info → glaip_sdk-0.6.0.dist-info}/RECORD

@@ -1,18 +1,20 @@
-glaip_sdk/__init__.py,sha256=7ICOkPeiwQGkadqxIgJXOYPHcJOHYRIF_GU0xkjOtSE,366
+glaip_sdk/__init__.py,sha256=0PAFfodqAEdggIiV1Es_JryDokZrhYLFFIXosqdguJU,420
 glaip_sdk/_version.py,sha256=5CHGCxx_36fgmMWuEx6jJ2CzzM-i9eBFyQWFwBi23XE,2259
+glaip_sdk/agents/__init__.py,sha256=VfYov56edbWuySXFEbWJ_jLXgwnFzPk1KB-9-mfsUCc,776
+glaip_sdk/agents/base.py,sha256=PLl1B8guGxNLeDvObrgCP-Lk12lFrwFMcK-mJl9rjrk,33577
 glaip_sdk/branding.py,sha256=tLqYCIHMkUf8p2smpuAGNptwaKUN38G4mlh0A0DOl_w,7823
 glaip_sdk/cli/__init__.py,sha256=xCCfuF1Yc7mpCDcfhHZTX0vizvtrDSLeT8MJ3V7m5A0,156
 glaip_sdk/cli/account_store.py,sha256=NXuAVPaJS_32Aw1VTaZCNwIID-gADw4F_UMieoWmg3g,17336
 glaip_sdk/cli/agent_config.py,sha256=YAbFKrTNTRqNA6b0i0Q3pH-01rhHDRi5v8dxSFwGSwM,2401
 glaip_sdk/cli/auth.py,sha256=9hfjZyd4cx2_mImqykJ1sWQsuVTR2gy6D4hFqAQNKL4,24129
 glaip_sdk/cli/commands/__init__.py,sha256=6Z3ASXDut0lAbUX_umBFtxPzzFyqoiZfVeTahThFu1A,219
-glaip_sdk/cli/commands/accounts.py,sha256=B5itsUzqoH_hBRYOVd2m4nPoIuBbPDIoK974zKMm9NE,18635
+glaip_sdk/cli/commands/accounts.py,sha256=NTexdyiv9Qp3xZMxmtwOWeCkRDHegyk06O9J3UWyXHQ,24644
 glaip_sdk/cli/commands/agents.py,sha256=WCOzllyh_Znwlju5camT4vE6OeRJbsAmjWwcyiAqWs4,48429
 glaip_sdk/cli/commands/common_config.py,sha256=IY13gPkeifXxSdpzRFUvfRin8J7s38p6Y7TYjdGw7w4,2474
 glaip_sdk/cli/commands/configure.py,sha256=95PQiJnpvsdH02v_tLVANd64qAJJnZKlhNe4tpfWIS4,30262
 glaip_sdk/cli/commands/mcps.py,sha256=tttqQnfM89iI9Pm94u8YRhiHMQNYNouecFX0brsT4cQ,42551
 glaip_sdk/cli/commands/models.py,sha256=vfcGprK5CHprQ0CNpNzQlNNTELvdgKC7JxTG_ijOwmE,2009
-glaip_sdk/cli/commands/tools.py,sha256=7_RMTuTI1Guu7psClovbyt2umfk4rkp7jSW19GXKA44,18440
+glaip_sdk/cli/commands/tools.py,sha256=_VBqG-vIjnn-gqvDlSTvcU7_F4N3ANGGKEECcQVR-BM,18430
 glaip_sdk/cli/commands/transcripts.py,sha256=ofxZLus1xLB061NxrLo1J6LPEb2VIxJTjmz7hLKgPmc,26377
 glaip_sdk/cli/commands/update.py,sha256=rIZo_x-tvpvcwpQLpwYwso1ix6qTHuNNTL4egmn5fEM,1812
 glaip_sdk/cli/config.py,sha256=_UU7pljAhgzwKLx7Yqp2_ibiInR-dHuykC_UWL51BHc,2310
@@ -47,33 +49,50 @@ glaip_sdk/cli/update_notifier.py,sha256=FnTjzS8YT94RmP6c5aU_XNIyRi7FRHvAskMy-VJi
 glaip_sdk/cli/utils.py,sha256=fV6PZlQ7K5zckpFWvwh3yLmETGrVylK9AXtN7zKBp-A,57374
 glaip_sdk/cli/validators.py,sha256=d-kq4y7HWMo6Gc7wLXWUsCt8JwFvJX_roZqRm1Nko1I,5622
 glaip_sdk/client/__init__.py,sha256=F-eE_dRSzA0cc1it06oi0tZetZBHmSUjWSHGhJMLCls,263
-glaip_sdk/client/_agent_payloads.py,sha256=VfBHoijuoqUOixGBf2SA2vlQIXQmBsjB3sXHZhXYiec,17681
+glaip_sdk/client/_agent_payloads.py,sha256=cH7CvNRn0JvudwKLr072E7W2QGWO9r-4xDxWMvXoPKE,17865
 glaip_sdk/client/agent_runs.py,sha256=tZSFEZZ3Yx0uYRgnwkLe-X0TlmgKJQ-ivzb6SrVnxY8,4862
-glaip_sdk/client/agents.py,sha256=tKIjkU9-lMEehf1lJe8Bl35nNQntIiyvq02b4fsVN5c,43332
-glaip_sdk/client/base.py,sha256=ikW33raz2M6rXzo3JmhttfXXuVdMv5zBRKEZkU1F-4I,18176
-glaip_sdk/client/main.py,sha256=-grOD3mMNbjOPbx2s1ZIF0z0KpEaCaNuXLHX9dqcZgQ,8711
-glaip_sdk/client/mcps.py,sha256=410HY1aC8hzaCDnEGnhrhKs8vC2xQ50PHvUy1PuuIIc,9364
+glaip_sdk/client/agents.py,sha256=QSeXH93ysyNae67FCStIpDfDr33t4_d2gXpJD07GV54,45866
+glaip_sdk/client/base.py,sha256=BhNaC2TJJ2jVWRTYmfxD3WjYgAyIuWNz9YURdNXXjJo,18245
+glaip_sdk/client/main.py,sha256=CRan8BCTH84d7MwJ-unAkpRlT5_oJkPnz9M7qe-fAl0,9034
+glaip_sdk/client/mcps.py,sha256=gFRuLOGeh6ieIhR4PeD6yNVT6NhvUMTqPq9iuu1vkAY,13019
 glaip_sdk/client/run_rendering.py,sha256=ubBO-NzyZoYRELNwxVvrQFRGQVJCuLfqqJNiXrBZDoQ,14223
 glaip_sdk/client/shared.py,sha256=esHlsR0LEfL-pFDaWebQjKKOLl09jsRY-2pllBUn4nU,522
-glaip_sdk/client/tools.py,sha256=LP7ZRMMlaolBpsu6M14KUifiCM2MWrQoWwX_DYgNxhE,17293
+glaip_sdk/client/tools.py,sha256=RR7345wECqPtofDXPW12VOnLtus554tXleL0YHQy82U,22435
 glaip_sdk/client/validators.py,sha256=ioF9VCs-LG2yLkaRDd7Hff74lojDZZ0_Q3CiLbdm1RY,8381
 glaip_sdk/config/constants.py,sha256=Y03c6op0e7K0jTQ8bmWXhWAqsnjWxkAhWniq8Z0iEKY,1081
 glaip_sdk/exceptions.py,sha256=iAChFClkytXRBLP0vZq1_YjoZxA9i4m4bW1gDLiGR1g,2321
 glaip_sdk/icons.py,sha256=J5THz0ReAmDwIiIooh1_G3Le-mwTJyEjhJDdJ13KRxM,524
-glaip_sdk/models/__init__.py,sha256=DtFft8zH3eJjeedm6jov1z54wTLTcb6dyOKRwM9ZGbk,1756
-glaip_sdk/models/agent_runs.py,sha256=oGUxS4UOq8C2QE4ilp0IQZ7ur1m0DPh6ybzTAOg2tcc,3808
-glaip_sdk/models.py,sha256=OyaGkh1OoWJeVAbIjnkPC8XhtuxJ7aylDvwsKC8LPqs,8504
+glaip_sdk/mcps/__init__.py,sha256=4jYrt8K__oxrxexHRcmnRBXt-W_tbJN61H9Kf2lVh4Q,551
+glaip_sdk/mcps/base.py,sha256=jWwHjDF67_mtDGRp9p5SolANjVeB8jt1PSwPBtX876M,11654
+glaip_sdk/models/__init__.py,sha256=-qO4Yr1-fkyaYC9RcT3nYhplDjoXATrIFZr4JrqflHI,2577
+glaip_sdk/models/agent.py,sha256=vtmUSDrrib1hvm0xnofIuOebqlrs-gIaLsny8hzg1iE,1523
+glaip_sdk/models/agent_runs.py,sha256=MYgab07k8MfExjvI5_o6HxsksJZ3tl2TDZefVxzGc1g,3807
+glaip_sdk/models/common.py,sha256=O30MEGO2nKcGhKbnPNkoGzwNvDVUBjM-uU-Tpigaz5Y,1180
+glaip_sdk/models/mcp.py,sha256=ti_8MUf4k7qbR1gPs9JhqhybMcLUhZxEELtHQrTv2-U,944
+glaip_sdk/models/tool.py,sha256=w3nL2DqyCtGgDPCd40Asi9obRGghQjLlC9Vt_p32Mpc,951
 glaip_sdk/payload_schemas/__init__.py,sha256=nTJmzwn2BbEpzZdq-8U24eVHQHxqYO3_-SABMV9lS_Q,142
 glaip_sdk/payload_schemas/agent.py,sha256=Nap68mI2Ba8eNGOhk79mGrYUoYUahcUJLof3DLWtVO4,3198
+glaip_sdk/registry/__init__.py,sha256=5y4zDT0fNnti0yhd2gv9f8rppgq9Cx6dyFaTV1tL3hc,1647
+glaip_sdk/registry/agent.py,sha256=F0axW4BIUODqnttIOzxnoS5AqQkLZ1i48FTeZNnYkhA,5203
+glaip_sdk/registry/base.py,sha256=e3KUD7WuKypvbu_h-roouD-ynIp49nDZavgpbgSdPbk,4151
+glaip_sdk/registry/mcp.py,sha256=bCpO-2ybaeQyKIQfdmz6BKmdPYb-_MG9jYEJNFykpQg,7764
+glaip_sdk/registry/tool.py,sha256=Y3mubiT84TOGByxOab1gF5ROBzcak_6o__JEGiv7uyM,7871
 glaip_sdk/rich_components.py,sha256=44Z0V1ZQleVh9gUDGwRR5mriiYFnVGOhm7fFxZYbP8c,4052
-glaip_sdk/utils/__init__.py,sha256=Fy11SCrasMRNnP-qbZ-Rr9JMhxgHHcSGG2v7jd7PP20,1087
+glaip_sdk/tools/__init__.py,sha256=rhGzEqQFCzeMrxmikBuNrMz4PyYczwic28boDKVmoHs,585
+glaip_sdk/tools/base.py,sha256=bvumLJ-DiQTmuYKgq2yCnlwrTZ9nYXpOwWU0e1vWR5g,15185
+glaip_sdk/utils/__init__.py,sha256=cAvQ3V0nocJxAOJ3_cablcGAsZkWzypDf9bm_cfM1dg,2750
 glaip_sdk/utils/agent_config.py,sha256=RhcHsSOVwOaSC2ggnPuHn36Aa0keGJhs8KGb2InvzRk,7262
-glaip_sdk/utils/client_utils.py,sha256=huMq2FWS4YnTTjWT23gaZABVdqRGRFLpcyhnQR8bs2c,14328
+glaip_sdk/utils/bundler.py,sha256=fQWAv0e5qNjF1Lah-FGoA9W5Q59YaHbQfX_4ZB84YRw,9120
+glaip_sdk/utils/client.py,sha256=otPUOIDvLCCsvFBNR8YMZFtRrORggmvvlFjl3YeeTqQ,3121
+glaip_sdk/utils/client_utils.py,sha256=hzHxxNuM37mK4HhgIdS0qg4AqjAA5ai2irPO6Nr1Uzo,15350
 glaip_sdk/utils/datetime_helpers.py,sha256=QLknNLEAY56628-MTRKnCXAffATkF33erOqBubKmU98,1544
+glaip_sdk/utils/discovery.py,sha256=DbnPuCXuS5mwTZ9fMfsPHQDJltFV99Wf8Em0YttktVU,1994
 glaip_sdk/utils/display.py,sha256=zu3SYqxj9hPyEN8G1vIXv_yXBkV8jLLCXEg2rs8NlzM,4485
 glaip_sdk/utils/export.py,sha256=1NxxE3wGsA1auzecG5oJw5ELB4VmPljoeIkGhrGOh1I,5006
 glaip_sdk/utils/general.py,sha256=3HSVIopUsIymPaim-kP2lqLX75TkkdIVLe6g3UKabZ0,1507
 glaip_sdk/utils/import_export.py,sha256=RCvoydm_6_L7_J1igcE6IYDunqgS5mQUbWT4VGrytMw,5510
+glaip_sdk/utils/import_resolver.py,sha256=PAM9-3KgxVJD4TO0qsf7vocI1OTpQ6GjUEvtInBTvVQ,16560
+glaip_sdk/utils/instructions.py,sha256=MTk93lsq3I8aRnvnRMSXXNMzcpnaIM_Pm3Aiiiq3GBc,2997
 glaip_sdk/utils/rendering/__init__.py,sha256=cJhhBEf46RnmUGJ1fivGkFuCoOn2pkJkSuRWoo1xlhE,3608
 glaip_sdk/utils/rendering/formatting.py,sha256=tP-CKkKFDhiAHS1vpJQ3D6NmPVl0TX1ZOwBOoxia2eE,8009
 glaip_sdk/utils/rendering/layout/__init__.py,sha256=Lz8eLXDO28wyK36BhKJ6o9YmsRjmQZrNhvZ2wwSjvPw,1609
@@ -106,8 +125,9 @@ glaip_sdk/utils/rendering/viewer/presenter.py,sha256=mlLMTjnyeyPVtsyrAbz1BJu9lFG
 glaip_sdk/utils/resource_refs.py,sha256=vF34kyAtFBLnaKnQVrsr2st1JiSxVbIZ4yq0DelJvCI,5966
 glaip_sdk/utils/run_renderer.py,sha256=d_VMI6LbvHPUUeRmGqh5wK_lHqDEIAcym2iqpbtDad0,1365
 glaip_sdk/utils/serialization.py,sha256=z-qpvWLSBrGK3wbUclcA1UIKLXJedTnMSwPdq-FF4lo,13308
+glaip_sdk/utils/sync.py,sha256=3VKqs1UfNGWSobgRXohBKP7mMMzdUW3SU0bJQ1uxOgw,4872
 glaip_sdk/utils/validation.py,sha256=Vt8oSnn7OM6ns5vjOl5FwGIMWBPb0yI6RD5XL_L5_4M,6826
-glaip_sdk-0.5.3.dist-info/METADATA,sha256=RwJBNhUEnTWDIPr-X4oYiLk-8GtG2IzBoV41jxGkad0,7053
-glaip_sdk-0.5.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-glaip_sdk-0.5.3.dist-info/entry_points.txt,sha256=EGs8NO8J1fdFMWA3CsF7sKBEvtHb_fujdCoNPhfMouE,47
-glaip_sdk-0.5.3.dist-info/RECORD,,
+glaip_sdk-0.6.0.dist-info/METADATA,sha256=kFvWJW0jzG-EdvGTRJoaZBaXqDNk66EpZN_Q9tq0OUM,7128
+glaip_sdk-0.6.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+glaip_sdk-0.6.0.dist-info/entry_points.txt,sha256=EGs8NO8J1fdFMWA3CsF7sKBEvtHb_fujdCoNPhfMouE,47
+glaip_sdk-0.6.0.dist-info/RECORD,,

glaip_sdk/models.py

@@ -1,241 +0,0 @@
-#!/usr/bin/env python3
-"""Data models for AIP SDK.
-
-Authors:
-    Raymond Christopher (raymond.christopher@gdplabs.id)
-"""
-
-from collections.abc import AsyncGenerator
-from datetime import datetime
-from typing import Any
-
-from pydantic import BaseModel
-
-from glaip_sdk.config.constants import DEFAULT_AGENT_RUN_TIMEOUT
-
-_AGENT_CLIENT_REQUIRED_MSG = "No client available. Use client.get_agent_by_id() to get a client-connected agent."
-_MCP_CLIENT_REQUIRED_MSG = "No client available. Use client.get_mcp_by_id() to get a client-connected MCP."
-
-
-class Agent(BaseModel):
-    """Agent model for API responses."""
-
-    id: str
-    name: str
-    instruction: str | None = None
-    description: str | None = None
-    type: str | None = None
-    framework: str | None = None
-    version: str | None = None
-    tools: list[dict[str, Any]] | None = None
-    agents: list[dict[str, Any]] | None = None
-    mcps: list[dict[str, Any]] | None = None
-    tool_configs: dict[str, Any] | None = None
-    mcp_configs: dict[str, Any] | None = None
-    agent_config: dict[str, Any] | None = None
-    timeout: int = DEFAULT_AGENT_RUN_TIMEOUT
-    metadata: dict[str, Any] | None = None
-    language_model_id: str | None = None
-    a2a_profile: dict[str, Any] | None = None
-    created_at: datetime | None = None
-    updated_at: datetime | None = None
-    _client: Any = None
-
-    def _set_client(self, client: Any) -> "Agent":
-        """Set the client reference for this resource."""
-        self._client = client
-        return self
-
-    def run(self, message: str, verbose: bool = False, **kwargs) -> str:
-        """Run the agent with a message.
-
-        Args:
-            message: The message to send to the agent
-            verbose: Enable verbose output and event JSON logging
-            **kwargs: Additional arguments passed to run_agent
-        """
-        if not self._client:
-            raise RuntimeError(_AGENT_CLIENT_REQUIRED_MSG)
-        # Automatically pass the agent name for better renderer display
-        kwargs.setdefault("agent_name", self.name)
-        # Pass the agent's configured timeout if not explicitly overridden
-        if "timeout" not in kwargs:
-            kwargs["timeout"] = self.timeout
-        # Pass verbose flag through to enable event JSON output
-        return self._client.run_agent(self.id, message, verbose=verbose, **kwargs)
-
-    async def arun(self, message: str, **kwargs) -> AsyncGenerator[dict, None]:
-        """Async run the agent with a message, yielding streaming JSON chunks.
-
-        Args:
-            message: The message to send to the agent
-            **kwargs: Additional arguments passed to arun_agent
-
-        Yields:
-            Dictionary containing parsed JSON chunks from the streaming response
-
-        Raises:
-            RuntimeError: When no client is available
-            AgentTimeoutError: When agent execution times out
-            Exception: For other unexpected errors
-        """
-        if not self._client:
-            raise RuntimeError(_AGENT_CLIENT_REQUIRED_MSG)
-        # Automatically pass the agent name for better context
-        kwargs.setdefault("agent_name", self.name)
-        # Pass the agent's configured timeout if not explicitly overridden
-        if "timeout" not in kwargs:
-            kwargs["timeout"] = self.timeout
-
-        async for chunk in self._client.arun_agent(self.id, message, **kwargs):
-            yield chunk
-
-    def update(self, **kwargs) -> "Agent":
-        """Update agent attributes."""
-        if not self._client:
-            raise RuntimeError(_AGENT_CLIENT_REQUIRED_MSG)
-        updated_agent = self._client.update_agent(self.id, **kwargs)
-        # Update current instance with new data
-        for key, value in updated_agent.model_dump().items():
-            if hasattr(self, key):
-                setattr(self, key, value)
-        return self
-
-    def delete(self) -> None:
-        """Delete the agent."""
-        if not self._client:
-            raise RuntimeError(_AGENT_CLIENT_REQUIRED_MSG)
-        self._client.delete_agent(self.id)
-
-
-class Tool(BaseModel):
-    """Tool model for API responses."""
-
-    id: str
-    name: str
-    tool_type: str | None = None
-    description: str | None = None
-    framework: str | None = None
-    version: str | None = None
-    tool_script: str | None = None
-    tool_file: str | None = None
-    tags: str | list[str] | None = None
-    _client: Any = None  # Will hold client reference
-
-    def _set_client(self, client: Any) -> "Tool":
-        """Set the client reference for this resource."""
-        self._client = client
-        return self
-
-    def get_script(self) -> str:
-        """Get the tool script content."""
-        if self.tool_script:
-            return self.tool_script
-        elif self.tool_file:
-            return f"Script content from file: {self.tool_file}"
-        else:
-            return "No script content available"
-
-    def update(self, **kwargs) -> "Tool":
-        """Update tool attributes.
-
-        Supports both metadata updates and file uploads.
-        Pass 'file' parameter to update tool code via file upload.
-        """
-        if not self._client:
-            raise RuntimeError("No client available. Use client.get_tool_by_id() to get a client-connected tool.")
-
-        # Check if file upload is requested
-        if "file" in kwargs:
-            file_path = kwargs.pop("file")  # Remove file from kwargs for metadata
-            updated_tool = self._client.tools.update_tool_via_file(self.id, file_path, **kwargs)
-        else:
-            # Regular metadata update
-            updated_tool = self._client.tools.update_tool(self.id, **kwargs)
-
-        # Update current instance with new data
-        for key, value in updated_tool.model_dump().items():
-            if hasattr(self, key):
-                setattr(self, key, value)
-        return self
-
-    def delete(self) -> None:
-        """Delete the tool."""
-        if not self._client:
-            raise RuntimeError("No client available. Use client.get_tool_by_id() to get a client-connected tool.")
-        self._client.delete_tool(self.id)
-
-
-class MCP(BaseModel):
-    """MCP model for API responses."""
-
-    id: str
-    name: str
-    description: str | None = None
-    config: dict[str, Any] | None = None
-    transport: str | None = None  # "sse" or "http"
-    authentication: dict[str, Any] | None = None
-    metadata: dict[str, Any] | None = None
-    _client: Any = None  # Will hold client reference
-
-    def _set_client(self, client: Any) -> "MCP":
-        """Set the client reference for this resource."""
-        self._client = client
-        return self
-
-    def get_tools(self) -> list[dict[str, Any]]:
-        """Get tools available from this MCP."""
-        if not self._client:
-            raise RuntimeError(_MCP_CLIENT_REQUIRED_MSG)
-        # This would delegate to the client's MCP tools endpoint
-        # For now, return empty list as placeholder
-        return []
-
-    def update(self, **kwargs) -> "MCP":
-        """Update MCP attributes."""
-        if not self._client:
-            raise RuntimeError(_MCP_CLIENT_REQUIRED_MSG)
-        updated_mcp = self._client.update_mcp(self.id, **kwargs)
-        # Update current instance with new data
-        for key, value in updated_mcp.model_dump().items():
-            if hasattr(self, key):
-                setattr(self, key, value)
-        return self
-
-    def delete(self) -> None:
-        """Delete the MCP."""
-        if not self._client:
-            raise RuntimeError("No client available. Use client.get_mcp_by_id() to get a client-connected MCP.")
-        self._client.delete_mcp(self.id)
-
-
-class LanguageModelResponse(BaseModel):
-    """Language model response model."""
-
-    name: str
-    provider: str
-    description: str | None = None
-    capabilities: list[str] | None = None
-    max_tokens: int | None = None
-    supports_streaming: bool = False
-
-
-class TTYRenderer:
-    """Simple TTY renderer for non-Rich environments."""
-
-    def __init__(self, use_color: bool = True):
-        """Initialize the TTY renderer.
-
-        Args:
-            use_color: Whether to use color output
-        """
-        self.use_color = use_color
-
-    def render_message(self, message: str, event_type: str = "message") -> None:
-        """Render a message with optional color."""
-        if event_type == "error":
-            print(f"ERROR: {message}", flush=True)
-        elif event_type == "done":
-            print(f"\n✅ {message}", flush=True)
-        else:
-            print(message, flush=True)