olbrain-python-sdk 0.5.0__tar.gz → 0.5.1__tar.gz

{olbrain_python_sdk-0.5.0/olbrain_python_sdk.egg-info → olbrain_python_sdk-0.5.1}/PKG-INFO

@@ -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.0 → olbrain_python_sdk-0.5.1}/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_python_sdk-0.5.0 → olbrain_python_sdk-0.5.1}/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_python_sdk-0.5.0 → olbrain_python_sdk-0.5.1}/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_python_sdk-0.5.0 → olbrain_python_sdk-0.5.1}/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 → olbrain_python_sdk-0.5.1/olbrain_python_sdk.egg-info}/PKG-INFO

@@ -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.0 → olbrain_python_sdk-0.5.1}/olbrain_python_sdk.egg-info/SOURCES.txt

@@ -26,6 +26,7 @@ olbrain_python_sdk.egg-info/not-zip-safe
 olbrain_python_sdk.egg-info/requires.txt
 olbrain_python_sdk.egg-info/top_level.txt
 tests/__init__.py
+tests/conftest.py
 tests/test_client.py
 tests/test_exceptions.py
 tests/test_olbrain.py

{olbrain_python_sdk-0.5.0 → olbrain_python_sdk-0.5.1}/setup.py

@@ -15,7 +15,7 @@ if os.path.exists(readme_path):
         long_description = f.read()
 
 # Read version from the package
-version = "0.5.0"
+version = "0.5.1"
 
 setup(
     name="olbrain-python-sdk",

olbrain_python_sdk-0.5.1/tests/conftest.py

@@ -0,0 +1,7 @@
+import pytest
+
+
+@pytest.fixture(autouse=True)
+def no_url_from_env(monkeypatch):
+    # A developer's own OLBRAIN_AGENT_URL must not leak into tests
+    monkeypatch.delenv("OLBRAIN_AGENT_URL", raising=False)

{olbrain_python_sdk-0.5.0 → olbrain_python_sdk-0.5.1}/tests/test_client.py

@@ -26,6 +26,15 @@ class TestAgentClient:
         client = AgentClient(agent_id="a1", api_key="ak_test", agent_url="https://custom.example/")
         assert client.agent_url == "https://custom.example"
 
+    def test_env_url(self, monkeypatch):
+        monkeypatch.setenv("OLBRAIN_AGENT_URL", "http://localhost:8080/")
+        assert AgentClient(agent_id="a1", api_key="ak_test").agent_url == "http://localhost:8080"
+
+    def test_argument_beats_env_url(self, monkeypatch):
+        monkeypatch.setenv("OLBRAIN_AGENT_URL", "http://localhost:8080")
+        client = AgentClient(agent_id="a1", api_key="ak_test", agent_url="https://custom.example")
+        assert client.agent_url == "https://custom.example"
+
     def test_chat_goes_to_the_agent_webhook(self, client, requests_mock):
         requests_mock.post(CHAT, json={"session_id": "s1", "response": "Hi", "success": True})
         assert client.create_session(title="Hello") == "s1"

{olbrain_python_sdk-0.5.0 → olbrain_python_sdk-0.5.1}/tests/test_olbrain.py

@@ -36,6 +36,11 @@ class TestConfig:
         with pytest.raises(ValidationError, match="ak_"):
             Olbrain(api_key="sk_live_x")
 
+    def test_env_url(self, monkeypatch):
+        monkeypatch.setenv("OLBRAIN_AGENT_URL", "http://localhost:8080/")
+        assert Olbrain(api_key="ak_test").base_url == "http://localhost:8080"
+        assert Olbrain(api_key="ak_test", base_url="https://custom.example").base_url == "https://custom.example"
+
     def test_asks_for_a_fresh_sign_in_token_on_every_call(self, requests_mock):
         tokens = iter(["t1", "t2"])
         olbrain = Olbrain(get_id_token=lambda: next(tokens))