olbrain-python-sdk 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

olbrain/AGENTS.md

@@ -56,7 +56,8 @@ with AgentClient(agent_id="your-agent-id", api_key=os.environ["OLBRAIN_API_KEY"]
 - `agent_id`: required. An empty value raises `ValueError`.
 - `api_key`: must start with `ak_`, `sk_live_`, `org_live_`, `sk_` or `org_`,
   or the constructor raises `ValueError`.
-- `agent_url`: optional. Defaults to `https://webhook.olbrain.com`.
+- `agent_url`: optional. Defaults to the `OLBRAIN_AGENT_URL` environment
+  variable if set, otherwise `https://webhook.olbrain.com`.
 
 Chat calls are `POST {agent_url}/api/agents/webhook/{agent_id}` with
 `Authorization: Bearer <api_key>`. Session reads are
@@ -146,7 +147,8 @@ olbrain = Olbrain(api_key=os.environ["OLBRAIN_API_KEY"])           # server / sc
 
 `Olbrain(api_key=None, get_id_token=None, base_url=None, timeout=30)`:
 `api_key` must start `ak_`; `get_id_token` is a callable returning the
-token and is called before **every** request; `base_url` defaults to
+token and is called before **every** request; `base_url` defaults to the
+`OLBRAIN_AGENT_URL` environment variable if set, otherwise
 `https://webhook.olbrain.com`; `timeout` is in seconds. Give **exactly one** of
 `api_key` or `get_id_token`, or the constructor raises `ValidationError`.
 

olbrain/__init__.py

@@ -39,7 +39,7 @@ from .exceptions import (
     BillingError,
 )
 
-__version__ = "0.5.0"
+__version__ = "0.5.1"
 __author__ = "Olbrain Team"
 __email__ = "support@olbrain.com"
 

olbrain/client.py

@@ -9,6 +9,7 @@ FIXES:
 - send_and_wait() now works correctly
 """
 
+import os
 import requests
 import logging
 import time
@@ -150,7 +151,8 @@ class AgentClient:
         Args:
             agent_id: Agent identifier
             api_key: Access key (ak_…); 'sk_live_', 'org_live_', 'sk_' and 'org_' are also accepted
-            agent_url: Optional agent URL (auto-constructed if not provided)
+            agent_url: Optional API base URL. Falls back to the OLBRAIN_AGENT_URL
+                environment variable, then https://webhook.olbrain.com
 
         Raises:
             ValueError: If agent_id or api_key is invalid
@@ -167,12 +169,9 @@ class AgentClient:
         self.agent_id = agent_id
         self.api_key = api_key
 
-        # Auto-construct agent URL if not provided
-        if agent_url:
-            self.agent_url = agent_url.rstrip('/')
-        else:
-            # The Olbrain API. The old run.app host this SDK defaulted to no longer exists.
-            self.agent_url = "https://webhook.olbrain.com"
+        # Explicit argument, then OLBRAIN_AGENT_URL, then the Olbrain API
+        agent_url = agent_url or os.environ.get("OLBRAIN_AGENT_URL") or "https://webhook.olbrain.com"
+        self.agent_url = agent_url.rstrip('/')
 
         self._streams = {}  # session_id -> MessageStream
         self._running = False

olbrain/olbrain.py

@@ -5,6 +5,7 @@ Olbrain sign-in (get_id_token). The same API as the JavaScript SDK's `Olbrain`.
 """
 
 import json
+import os
 from types import SimpleNamespace
 from typing import Any, Callable, Dict, Iterator, Optional
 from urllib.parse import urlencode
@@ -70,7 +71,7 @@ class Olbrain:
             raise ValidationError("api_key must be an Olbrain access key (ak_…)")
         self._api_key = api_key
         self._get_id_token = get_id_token
-        self.base_url = (base_url or DEFAULT_BASE_URL).rstrip("/")
+        self.base_url = (base_url or os.environ.get("OLBRAIN_AGENT_URL") or DEFAULT_BASE_URL).rstrip("/")
         self.timeout = timeout
 
         def runs(agent_id):

{olbrain_python_sdk-0.5.0.dist-info → olbrain_python_sdk-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: olbrain-python-sdk
-Version: 0.5.0
+Version: 0.5.1
 Summary: Official Python SDK for Olbrain AI agents
 Home-page: https://github.com/Olbrain/olbrain-sdk
 Author: Olbrain Team

olbrain_python_sdk-0.5.1.dist-info/RECORD

@@ -0,0 +1,14 @@
+olbrain/AGENTS.md,sha256=LCUONEQy-OW3xQzZjBYwTQErK2d75QHpp__1DGoBDqA,13276
+olbrain/__init__.py,sha256=damIVzm9Ju78DskFULa-GRUwllS3NSyguLWpYq2MDxM,1728
+olbrain/client.py,sha256=GetL1pFa4KZ_7gtLs5ILsGtOQpAck-eCaDUy5kwh5z8,22093
+olbrain/exceptions.py,sha256=fY8B4K8dts-NRwc8RROeFyXP18nUgJnvrrOvmzH555E,2243
+olbrain/olbrain.py,sha256=MzdjTKv-1w7r94CLJ-hqLMWnpEhmZssP0yiP-Om8UkQ,8255
+olbrain/research.py,sha256=DWUUWFkXOqicH2QkVf5g96n6UDyP-Mi8OmYFLbK6zWs,7599
+olbrain/session.py,sha256=0Off9pW0dTKYL9hzFYwH35zeNT5R20M9UfOKlCybYuQ,6662
+olbrain/streaming.py,sha256=YQDLStIkiZusA2du5JLtx2lwdCMAVWuOW9jsAZoRWxY,3532
+olbrain/utils.py,sha256=NrOB61Zub7i-ITlfXPfwUFunRR9YXJIS3MYgqVvqigM,4262
+olbrain_python_sdk-0.5.1.dist-info/licenses/LICENSE,sha256=_4Q6zZ9zlqqRkRm_ZNiGTkmrMeTJc88agU6c_O-c5sI,1068
+olbrain_python_sdk-0.5.1.dist-info/METADATA,sha256=RmdZMmb2E8XdmdJPbXEpqXWDK5u_VrGIPrlWsjadTgQ,16369
+olbrain_python_sdk-0.5.1.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+olbrain_python_sdk-0.5.1.dist-info/top_level.txt,sha256=iAqbGCG4mlzbE-zLwW4AgQv5_CI9enWxgc7yA9PttvM,8
+olbrain_python_sdk-0.5.1.dist-info/RECORD,,

olbrain_python_sdk-0.5.0.dist-info/RECORD

@@ -1,14 +0,0 @@
-olbrain/AGENTS.md,sha256=I2OIYFMmRbHOoORIb9JC-_d7BzW8Ai6kPtZH3-jnemI,13148
-olbrain/__init__.py,sha256=wGSzsv1hZ3mCn9iCoE6ZgOzjdofDNAhyaIqtfpzms7M,1728
-olbrain/client.py,sha256=YkmlthZ90V7lI_j4pj2SbancDnww0eoVDepAPJPYyKk,22073
-olbrain/exceptions.py,sha256=fY8B4K8dts-NRwc8RROeFyXP18nUgJnvrrOvmzH555E,2243
-olbrain/olbrain.py,sha256=_y2Qtq6NIQ_DkJW6PHAGiRjQz3__u-bfGeRCJS3-g1A,8206
-olbrain/research.py,sha256=DWUUWFkXOqicH2QkVf5g96n6UDyP-Mi8OmYFLbK6zWs,7599
-olbrain/session.py,sha256=0Off9pW0dTKYL9hzFYwH35zeNT5R20M9UfOKlCybYuQ,6662
-olbrain/streaming.py,sha256=YQDLStIkiZusA2du5JLtx2lwdCMAVWuOW9jsAZoRWxY,3532
-olbrain/utils.py,sha256=NrOB61Zub7i-ITlfXPfwUFunRR9YXJIS3MYgqVvqigM,4262
-olbrain_python_sdk-0.5.0.dist-info/licenses/LICENSE,sha256=_4Q6zZ9zlqqRkRm_ZNiGTkmrMeTJc88agU6c_O-c5sI,1068
-olbrain_python_sdk-0.5.0.dist-info/METADATA,sha256=MGS_Ev7Jn_c_jHjKxitEjzhVuvglPdsIJOi3mk_o76k,16369
-olbrain_python_sdk-0.5.0.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
-olbrain_python_sdk-0.5.0.dist-info/top_level.txt,sha256=iAqbGCG4mlzbE-zLwW4AgQv5_CI9enWxgc7yA9PttvM,8
-olbrain_python_sdk-0.5.0.dist-info/RECORD,,