olbrain-python-sdk 0.3.1__tar.gz → 0.3.2__tar.gz

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2}/MANIFEST.in

@@ -3,7 +3,7 @@ include README.md
 include LICENSE
 include requirements.txt
 include pyproject.toml
-recursive-include olbrain *.py
+recursive-include olbrain *.py *.md
 recursive-exclude * __pycache__
 recursive-exclude * *.py[co]
 global-exclude .DS_Store

{olbrain_python_sdk-0.3.1/olbrain_python_sdk.egg-info → olbrain_python_sdk-0.3.2}/PKG-INFO

@@ -1,14 +1,14 @@
 Metadata-Version: 2.4
 Name: olbrain-python-sdk
-Version: 0.3.1
+Version: 0.3.2
 Summary: Official Python SDK for Olbrain AI agents
-Home-page: https://github.com/Olbrain/olbrain-python-sdk
+Home-page: https://github.com/Olbrain/olbrain-sdk
 Author: Olbrain Team
 Author-email: Olbrain Team <support@olbrain.com>
 License: MIT
-Project-URL: Homepage, https://github.com/Olbrain/olbrain-python-sdk
-Project-URL: Bug Reports, https://github.com/Olbrain/olbrain-python-sdk/issues
-Project-URL: Source, https://github.com/Olbrain/olbrain-python-sdk
+Project-URL: Homepage, https://github.com/Olbrain/olbrain-sdk/tree/main/packages/python
+Project-URL: Bug Reports, https://github.com/Olbrain/olbrain-sdk/issues
+Project-URL: Source, https://github.com/Olbrain/olbrain-sdk
 Project-URL: Documentation, https://docs.olbrain.com/python-sdk
 Keywords: olbrain,ai,agents,chatbot,nlp,artificial intelligence,sdk,api,conversation,streaming
 Classifier: Development Status :: 4 - Beta
@@ -54,6 +54,8 @@ Dynamic: requires-python
 
 Official Python SDK for integrating Olbrain AI agents into your applications.
 
+> **Coding assistants:** the complete API reference ships in the package as [`olbrain/AGENTS.md`](olbrain/AGENTS.md) (`site-packages/olbrain/AGENTS.md`).
+
 ## Installation
 
 ```bash
@@ -66,8 +68,8 @@ For local development and testing:
 
 ```bash
 # Clone the repository
-git clone https://github.com/Olbrain/olbrain-python-sdk.git
-cd olbrain-python-sdk
+git clone https://github.com/Olbrain/olbrain-sdk.git
+cd olbrain-sdk/packages/python
 
 # Create virtual environment
 python3 -m venv venv
@@ -315,7 +317,7 @@ The SDK validates API keys. Valid prefixes are:
 # Valid API key
 client = AgentClient(
     agent_id="your-agent-id",
-    api_key="sk_live_24c2859a0cf359b8aebbe3d8e9c60f0341d111b30004a56b428726edb354bec5"
+    api_key="sk_live_your_api_key_here"
 )
 ```
 
@@ -509,7 +511,7 @@ python examples/basic_usage.py
 
 ## Support
 
-- **Issues**: [GitHub Issues](https://github.com/Olbrain/olbrain-python-sdk/issues)
+- **Issues**: [GitHub Issues](https://github.com/Olbrain/olbrain-sdk/issues)
 - **Documentation**: See [examples/](examples/) directory
 - **API Status**: Check [status page](https://olbrain.com/status)
 
@@ -524,6 +526,6 @@ Current version: **0.2.0**
 ## Links
 
 - [PyPI Package](https://pypi.org/project/olbrain-python-sdk/)
-- [GitHub Repository](https://github.com/Olbrain/olbrain-python-sdk)
-- [Report Issues](https://github.com/Olbrain/olbrain-python-sdk/issues)
+- [GitHub Repository](https://github.com/Olbrain/olbrain-sdk)
+- [Report Issues](https://github.com/Olbrain/olbrain-sdk/issues)
 - [Olbrain Website](https://olbrain.com)

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2}/README.md

@@ -6,6 +6,8 @@
 
 Official Python SDK for integrating Olbrain AI agents into your applications.
 
+> **Coding assistants:** the complete API reference ships in the package as [`olbrain/AGENTS.md`](olbrain/AGENTS.md) (`site-packages/olbrain/AGENTS.md`).
+
 ## Installation
 
 ```bash
@@ -18,8 +20,8 @@ For local development and testing:
 
 ```bash
 # Clone the repository
-git clone https://github.com/Olbrain/olbrain-python-sdk.git
-cd olbrain-python-sdk
+git clone https://github.com/Olbrain/olbrain-sdk.git
+cd olbrain-sdk/packages/python
 
 # Create virtual environment
 python3 -m venv venv
@@ -267,7 +269,7 @@ The SDK validates API keys. Valid prefixes are:
 # Valid API key
 client = AgentClient(
     agent_id="your-agent-id",
-    api_key="sk_live_24c2859a0cf359b8aebbe3d8e9c60f0341d111b30004a56b428726edb354bec5"
+    api_key="sk_live_your_api_key_here"
 )
 ```
 
@@ -461,7 +463,7 @@ python examples/basic_usage.py
 
 ## Support
 
-- **Issues**: [GitHub Issues](https://github.com/Olbrain/olbrain-python-sdk/issues)
+- **Issues**: [GitHub Issues](https://github.com/Olbrain/olbrain-sdk/issues)
 - **Documentation**: See [examples/](examples/) directory
 - **API Status**: Check [status page](https://olbrain.com/status)
 
@@ -476,6 +478,6 @@ Current version: **0.2.0**
 ## Links
 
 - [PyPI Package](https://pypi.org/project/olbrain-python-sdk/)
-- [GitHub Repository](https://github.com/Olbrain/olbrain-python-sdk)
-- [Report Issues](https://github.com/Olbrain/olbrain-python-sdk/issues)
+- [GitHub Repository](https://github.com/Olbrain/olbrain-sdk)
+- [Report Issues](https://github.com/Olbrain/olbrain-sdk/issues)
 - [Olbrain Website](https://olbrain.com)

olbrain_python_sdk-0.3.2/olbrain/AGENTS.md

@@ -0,0 +1,167 @@
+# olbrain-python-sdk — reference for coding assistants
+
+This file ships inside the installed package (`site-packages/olbrain/AGENTS.md`).
+It is the complete, offline reference for the public API. Everything here is
+taken from the source. If it disagrees with the web docs, this file is right.
+
+Find it from Python with:
+`python -c "import olbrain, os; print(os.path.join(os.path.dirname(olbrain.__file__), 'AGENTS.md'))"`
+
+The Python SDK covers **conversational agents** only (`AgentClient`). Workflow
+runs, approvals and research agents are available only in the JavaScript SDK
+(`@olbrain/js-sdk`, class `Olbrain`).
+
+## Install and import
+
+```bash
+pip install olbrain-python-sdk   # import name: olbrain. Requires Python 3.7+ and requests.
+```
+
+```python
+from olbrain import AgentClient, ChatResponse, OlbrainError
+```
+
+## Quick start
+
+```python
+import os
+from olbrain import AgentClient
+
+with AgentClient(agent_id="your-agent-id", api_key=os.environ["OLBRAIN_API_KEY"]) as client:
+    session_id = client.create_session(title="Support chat", user_id="u_42")
+    reply = client.send_and_wait(session_id, "Hello!")
+    if reply.success:
+        print(reply.text)
+    else:
+        print("agent error:", reply.error)
+```
+
+The client is synchronous and uses `requests`.
+
+## `AgentClient(agent_id, api_key, agent_url=None)`
+
+- `agent_id`: required. An empty value raises `ValueError`.
+- `api_key`: must start with `sk_live_`, `org_live_`, `sk_` or `org_`, or the
+  constructor raises `ValueError`. `ak_` keys are **not** accepted here.
+- `agent_url`: optional. Defaults to `https://olbrain-agent-cloud-768934887465.us-central1.run.app`.
+
+Every call is a `POST {agent_url}/api/agent/webhook` with headers
+`Authorization: Bearer <api_key>` and `X-Agent-ID: <agent_id>`. The client is a
+context manager, and `__exit__` calls `close()`.
+
+### Methods
+
+| Method | Returns | Notes |
+|---|---|---|
+| `create_session(on_message=None, title=None, user_id=None, metadata=None, mode="production", description=None)` | `str` session id | Sends `title` (default `"New Session"`) to the agent **as the first message**. `mode` must be `"development"`, `"testing"` or `"production"`, otherwise `ValueError`. `description` is stored as `metadata["session_description"]`. If you pass `on_message`, it also calls `listen()`. Timeout 30 s. |
+| `send_and_wait(session_id, message, user_id=None, metadata=None, model=None, timeout=120)` | `ChatResponse` | Waits for the reply. `timeout` is in **seconds**. `model` overrides the agent's model. Empty or whitespace-only messages raise `ValueError`. |
+| `send(session_id, message, user_id=None, metadata=None, model=None, mode="production")` | `dict` (raw JSON) | Same request, but returns the raw response dict (`response`, `token_usage`, `success`, …). It still blocks, with a 120 s timeout. |
+| `listen(session_id, on_message)` | `None` | Starts a background daemon thread that reads SSE from `GET {agent_url}/sessions/{id}/stream`. It reconnects with backoff from 5 s up to 60 s. Calling it again for the same session does nothing. |
+| `run()` | `None` | Blocks until Ctrl+C, then calls `close()`. Use it to keep a script alive for `listen` callbacks. |
+| `close()` | `None` | Stops all streams. |
+| `get_session(session_id)` | `SessionInfo` | Needs backend support (webhook `action: get_session_info`). |
+| `update_session(session_id, title=None, metadata=None, status=None)` | `SessionInfo` | `status` must be `"active"` or `"archived"`. |
+| `delete_session(session_id)` | `dict` | Also stops that session's stream. |
+| `get_session_stats(session_id)` | `dict` | |
+| `get_messages(session_id, limit=20, offset=0)` | `dict` | The messages list and pagination info, as the backend returns them. |
+
+The five session-management methods (`get_session` through `get_messages`)
+depend on webhook `action`s that not every backend implements. They raise if
+the backend refuses. Only `create_session`, `send_and_wait` and `send` are
+guaranteed to work.
+
+### Streaming callback
+
+`on_message` gets the **raw dict** from each SSE frame that has a `content`
+key. Frames whose `type` is `ping` or `connected` are skipped. The callback
+runs on the stream thread, so keep it fast and don't raise from it.
+
+```python
+def on_message(msg: dict):
+    print(f"{msg.get('role')}: {msg['content']}")
+
+client = AgentClient(agent_id="…", api_key="sk_live_…")
+sid = client.create_session(on_message=on_message, title="Streaming chat")
+client.send(sid, "Tell me a story")
+client.run()  # blocks; Ctrl+C to stop
+```
+
+## Data classes (`olbrain.session`, re-exported from `olbrain`)
+
+```python
+@dataclass
+class ChatResponse:
+    text: str
+    session_id: str
+    success: bool = True
+    token_usage: Optional[TokenUsage] = None   # None if the server sent none
+    model_used: Optional[str] = None
+    response_time_ms: Optional[int] = None
+    metadata: Optional[dict] = None
+    error: Optional[str] = None
+    # ChatResponse.from_dict(data, session_id=""), .to_dict()
+
+@dataclass
+class TokenUsage:
+    prompt_tokens: int = 0
+    completion_tokens: int = 0
+    total_tokens: int = 0
+    cost: float = 0.0
+
+@dataclass
+class SessionInfo:
+    session_id: str
+    title: str = ""
+    status: str = "active"        # "active" | "archived"
+    created_at: str = ""
+    updated_at: str = ""
+    message_count: int = 0
+    user_id: Optional[str] = None
+    channel: str = "api"
+    metadata: dict = {}
+```
+
+`token_usage` can be `None`, so guard before reading it:
+`reply.token_usage.total_tokens if reply.token_usage else 0`. Always check
+`reply.success` too. A 200 response can carry `success=False` and an `error`.
+
+Avoid the exported `Message` and `Session` classes. Nothing in the client
+returns them, and `Session.get_stats()` always falls back to local data.
+
+## Errors (`olbrain.exceptions`)
+
+All of them subclass `OlbrainError`, which has `.message`, `.error_code` and
+`.details`.
+
+| Condition | Raised |
+|---|---|
+| Bad constructor args, empty message, bad `mode`/`status` | `ValueError` (**not** `ValidationError`) |
+| HTTP 401 | `AuthenticationError` |
+| HTTP 404 whose detail mentions "session" | `SessionNotFoundError` (a subclass of `SessionError`) |
+| Other HTTP 404 | `OlbrainError` ("Endpoint not found (404) …") |
+| HTTP 429 | `RateLimitError` (`.retry_after` from `detail.retry_after`, may be `None`) |
+| Any other non-200 (403, 5xx…) | `OlbrainError("Failed to <operation>: <status> - <detail>")` |
+| Connection error or timeout | `NetworkError` |
+
+```python
+from olbrain import AgentClient, AuthenticationError, RateLimitError, NetworkError, OlbrainError
+import time
+
+def ask(client, sid, text, tries=3):
+    for i in range(tries):
+        try:
+            return client.send_and_wait(sid, text)
+        except RateLimitError as e:
+            time.sleep(e.retry_after or 2 ** i)
+        except NetworkError:
+            time.sleep(2 ** i)
+    raise OlbrainError("gave up")
+```
+
+`ValidationError` and `StreamingError` are exported, but the client never
+raises them.
+
+## Logging
+
+The SDK logs through `logging.getLogger("olbrain")`. Turn on
+`logging.basicConfig(level=logging.DEBUG)` to see requests and stream events.

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2}/olbrain/__init__.py

@@ -2,6 +2,8 @@
 Olbrain Python SDK
 Simple client for Olbrain AI agents with real-time message streaming
 
+Full offline API reference: AGENTS.md next to this file.
+
 Example:
     >>> from olbrain import AgentClient
     >>>
@@ -33,7 +35,7 @@ from .exceptions import (
     StreamingError
 )
 
-__version__ = "0.3.1"
+__version__ = "0.3.2"
 __author__ = "Olbrain Team"
 __email__ = "support@olbrain.com"
 

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2/olbrain_python_sdk.egg-info}/PKG-INFO

@@ -1,14 +1,14 @@
 Metadata-Version: 2.4
 Name: olbrain-python-sdk
-Version: 0.3.1
+Version: 0.3.2
 Summary: Official Python SDK for Olbrain AI agents
-Home-page: https://github.com/Olbrain/olbrain-python-sdk
+Home-page: https://github.com/Olbrain/olbrain-sdk
 Author: Olbrain Team
 Author-email: Olbrain Team <support@olbrain.com>
 License: MIT
-Project-URL: Homepage, https://github.com/Olbrain/olbrain-python-sdk
-Project-URL: Bug Reports, https://github.com/Olbrain/olbrain-python-sdk/issues
-Project-URL: Source, https://github.com/Olbrain/olbrain-python-sdk
+Project-URL: Homepage, https://github.com/Olbrain/olbrain-sdk/tree/main/packages/python
+Project-URL: Bug Reports, https://github.com/Olbrain/olbrain-sdk/issues
+Project-URL: Source, https://github.com/Olbrain/olbrain-sdk
 Project-URL: Documentation, https://docs.olbrain.com/python-sdk
 Keywords: olbrain,ai,agents,chatbot,nlp,artificial intelligence,sdk,api,conversation,streaming
 Classifier: Development Status :: 4 - Beta
@@ -54,6 +54,8 @@ Dynamic: requires-python
 
 Official Python SDK for integrating Olbrain AI agents into your applications.
 
+> **Coding assistants:** the complete API reference ships in the package as [`olbrain/AGENTS.md`](olbrain/AGENTS.md) (`site-packages/olbrain/AGENTS.md`).
+
 ## Installation
 
 ```bash
@@ -66,8 +68,8 @@ For local development and testing:
 
 ```bash
 # Clone the repository
-git clone https://github.com/Olbrain/olbrain-python-sdk.git
-cd olbrain-python-sdk
+git clone https://github.com/Olbrain/olbrain-sdk.git
+cd olbrain-sdk/packages/python
 
 # Create virtual environment
 python3 -m venv venv
@@ -315,7 +317,7 @@ The SDK validates API keys. Valid prefixes are:
 # Valid API key
 client = AgentClient(
     agent_id="your-agent-id",
-    api_key="sk_live_24c2859a0cf359b8aebbe3d8e9c60f0341d111b30004a56b428726edb354bec5"
+    api_key="sk_live_your_api_key_here"
 )
 ```
 
@@ -509,7 +511,7 @@ python examples/basic_usage.py
 
 ## Support
 
-- **Issues**: [GitHub Issues](https://github.com/Olbrain/olbrain-python-sdk/issues)
+- **Issues**: [GitHub Issues](https://github.com/Olbrain/olbrain-sdk/issues)
 - **Documentation**: See [examples/](examples/) directory
 - **API Status**: Check [status page](https://olbrain.com/status)
 
@@ -524,6 +526,6 @@ Current version: **0.2.0**
 ## Links
 
 - [PyPI Package](https://pypi.org/project/olbrain-python-sdk/)
-- [GitHub Repository](https://github.com/Olbrain/olbrain-python-sdk)
-- [Report Issues](https://github.com/Olbrain/olbrain-python-sdk/issues)
+- [GitHub Repository](https://github.com/Olbrain/olbrain-sdk)
+- [Report Issues](https://github.com/Olbrain/olbrain-sdk/issues)
 - [Olbrain Website](https://olbrain.com)

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2}/olbrain_python_sdk.egg-info/SOURCES.txt

@@ -5,12 +5,12 @@ README.md
 pyproject.toml
 requirements.txt
 setup.py
-test_agent.py
 examples/advanced_features.py
 examples/basic_usage.py
 examples/error_handling.py
 examples/session_management.py
 examples/streaming_responses.py
+olbrain/AGENTS.md
 olbrain/__init__.py
 olbrain/client.py
 olbrain/exceptions.py

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2}/olbrain_python_sdk.egg-info/top_level.txt

@@ -1,2 +1 @@
 olbrain
-tests

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2}/pyproject.toml

@@ -61,11 +61,14 @@ async = [
 ]
 
 [project.urls]
-"Homepage" = "https://github.com/Olbrain/olbrain-python-sdk"
-"Bug Reports" = "https://github.com/Olbrain/olbrain-python-sdk/issues"
-"Source" = "https://github.com/Olbrain/olbrain-python-sdk"
+"Homepage" = "https://github.com/Olbrain/olbrain-sdk/tree/main/packages/python"
+"Bug Reports" = "https://github.com/Olbrain/olbrain-sdk/issues"
+"Source" = "https://github.com/Olbrain/olbrain-sdk"
 "Documentation" = "https://docs.olbrain.com/python-sdk"
 
+[tool.setuptools.package-data]
+olbrain = ["AGENTS.md"]
+
 [tool.setuptools_scm]
 write_to = "olbrain/_version.py"
 

{olbrain_python_sdk-0.3.1 → olbrain_python_sdk-0.3.2}/setup.py

@@ -15,7 +15,7 @@ if os.path.exists(readme_path):
         long_description = f.read()
 
 # Read version from the package
-version = "0.3.1"
+version = "0.3.2"
 
 setup(
     name="olbrain-python-sdk",
@@ -25,8 +25,9 @@ setup(
     description="Official Python SDK for Olbrain AI agents",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    url="https://github.com/Olbrain/olbrain-python-sdk",
-    packages=find_packages(),
+    url="https://github.com/Olbrain/olbrain-sdk",
+    packages=find_packages(exclude=["tests", "tests.*"]),
+    package_data={"olbrain": ["AGENTS.md"]},
     classifiers=[
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Developers",
@@ -76,8 +77,8 @@ setup(
         "streaming"
     ],
     project_urls={
-        "Bug Reports": "https://github.com/Olbrain/olbrain-python-sdk/issues",
-        "Source": "https://github.com/Olbrain/olbrain-python-sdk",
+        "Bug Reports": "https://github.com/Olbrain/olbrain-sdk/issues",
+        "Source": "https://github.com/Olbrain/olbrain-sdk/tree/main/packages/python",
         "Documentation": "https://docs.olbrain.com/python-sdk",
     },
     include_package_data=True,

olbrain_python_sdk-0.3.1/test_agent.py

@@ -1,143 +0,0 @@
-#!/usr/bin/env python3
-"""
-Test script for Olbrain Python SDK
-Tests agent eQG36P3MaS3T6craxbET with real-time streaming
-"""
-
-import sys
-import os
-import logging
-import time
-import threading
-
-# Add current directory to path to use local SDK
-sys.path.insert(0, os.path.dirname(__file__))
-
-from olbrain import AgentClient
-
-# Setup logging
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-)
-
-def main():
-    """Main test function"""
-
-    # Agent configuration
-    AGENT_ID = "Ock2O6ZkG7bN1bPoAYSS"
-
-    print("\n" + "=" * 70)
-    print("🧪 OLBRAIN SDK TEST")
-    print("=" * 70)
-    print(f"\n🎯 Target Agent: {AGENT_ID}")
-
-    # Get API key
-    API_KEY = "sk_live_24c2859a0cf359b8aebbe3d8e9c60f0341d111b30004a56b428726edb354bec5"
-
-    print("\n" + "=" * 70)
-    print("🚀 STARTING TEST")
-    print("=" * 70)
-
-    # Message handler with nice formatting
-    def on_message(msg):
-        """Handle incoming messages"""
-        role = msg.get('role', 'unknown')
-        content = msg.get('content', '')
-        tokens = msg.get('token_usage', {})
-        total_tokens = tokens.get('total', 0)
-        cost = tokens.get('cost', 0)
-
-        # Format output
-        if role == 'user':
-            emoji = "👤"
-            color_code = "\033[94m"  # Blue
-        else:
-            emoji = "🤖"
-            color_code = "\033[92m"  # Green
-
-        reset_code = "\033[0m"
-
-        print(f"\n{color_code}{emoji} [{role.upper()}]{reset_code}")
-        print(f"   {content}")
-
-        if total_tokens > 0:
-            print(f"   💰 Tokens: {total_tokens} | Cost: ${cost:.6f}")
-
-    # Create client
-    try:
-        print("\n📦 Initializing SDK client...")
-        client = AgentClient(agent_id=AGENT_ID, api_key=API_KEY)
-        print(f"✅ Client initialized: {client}")
-
-    except Exception as e:
-        print(f"\n❌ Failed to initialize client: {e}")
-        return 1
-
-    try:
-        # Create session
-        print("\n📝 Creating session...")
-        session = client.create_session(on_message=on_message)
-        print(f"✅ Session created: {session}")
-        print("   📡 Real-time message stream active")
-
-        print("\n" + "=" * 70)
-        print("💬 INTERACTIVE MODE")
-        print("=" * 70)
-        print("\n   ✨ Type your messages below")
-        print("   📨 All responses will appear in real-time")
-        print("   ⏰ Scheduled messages will arrive automatically")
-        print("   🚪 Type 'exit' or 'quit' to stop")
-        print("   ⏹️  Press Ctrl+C to force exit\n")
-
-        # Main thread handles input (SSE stream runs in background)
-        while True:
-            try:
-                print("USER > ", end="", flush=True)
-                user_input = input()
-
-                if user_input.strip().lower() in ['exit', 'quit']:
-                    print("\n👋 Exiting...")
-                    break
-
-                if user_input.strip():
-                    client.send(session, user_input)
-
-            except EOFError:
-                # Ctrl+D pressed
-                print("\n👋 Exiting...")
-                break
-            except Exception as e:
-                print(f"\n⚠️  Error: {e}")
-                break
-
-    except KeyboardInterrupt:
-        print("\n\n" + "=" * 70)
-        print("👋 GRACEFUL SHUTDOWN")
-        print("=" * 70)
-        print("\n✅ Test completed successfully!")
-
-    except Exception as e:
-        print("\n\n" + "=" * 70)
-        print("❌ ERROR OCCURRED")
-        print("=" * 70)
-        print(f"\n💥 Error: {e}")
-        print("\n📋 Traceback:")
-        import traceback
-        traceback.print_exc()
-        return 1
-
-    finally:
-        print("\n🧹 Cleaning up...")
-        client.close()
-        print("✅ Cleanup complete")
-        print("\n" + "=" * 70)
-        print("🏁 TEST FINISHED")
-        print("=" * 70 + "\n")
-
-    return 0
-
-
-if __name__ == "__main__":
-    exit_code = main()
-    sys.exit(exit_code)