baalebos-ai 1.0.0__tar.gz

baalebos_ai-1.0.0/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: baalebos-ai
+Version: 1.0.0
+Summary: Official Python client and terminal CLI for the Baalebos AI Gateway
+Home-page: https://github.com/baalebos-cloud/nexus-ai-gateway
+Author: Baalebos AI Team
+License: MIT
+Project-URL: Source, https://github.com/baalebos-cloud/nexus-ai-gateway
+Project-URL: Bug Tracker, https://github.com/baalebos-cloud/nexus-ai-gateway/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests
+Requires-Dist: httpx
+Dynamic: home-page
+Dynamic: requires-python

baalebos_ai-1.0.0/baalebos_ai/__init__.py

@@ -0,0 +1,19 @@
+from baalebos_ai.client import BaalebosAI, AsyncBaalebosAI
+from baalebos_ai.exceptions import (
+    BaalebosError,
+    BaalebosConfigError,
+    BaalebosConnectionError,
+    BaalebosAPIError,
+    BaalebosAuthError,
+)
+
+__all__ = [
+    "BaalebosAI",
+    "AsyncBaalebosAI",
+    "BaalebosError",
+    "BaalebosConfigError",
+    "BaalebosConnectionError",
+    "BaalebosAPIError",
+    "BaalebosAuthError",
+]
+__version__ = "1.0.0"

baalebos_ai-1.0.0/baalebos_ai/cli.py

@@ -0,0 +1,27 @@
+import sys
+import argparse
+from baalebos_ai.client import BaalebosAI
+
+
+def main():
+    parser = argparse.ArgumentParser(description='Baalebos AI Gateway Terminal CLI')
+    parser.add_argument('prompt', nargs='*', help='The prompt to send to the AI gateway')
+    parser.add_argument('--mode', default='auto', help='Execution mode (auto, fast, smart)')
+    args = parser.parse_args()
+
+    if not args.prompt:
+        print('Usage: ai "Your prompt here"')
+        sys.exit(1)
+
+    full_prompt = ' '.join(args.prompt)
+    try:
+        client = BaalebosAI()
+        response = client.chat(prompt=full_prompt, mode=args.mode)
+        print('\n' + response + '\n')
+    except Exception as e:
+        print(f'Error: {e}', file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == '__main__':
+    main()

baalebos_ai-1.0.0/baalebos_ai/client.py

@@ -0,0 +1,98 @@
+import os
+import requests
+import httpx
+
+from baalebos_ai.exceptions import (
+    BaalebosConfigError,
+    BaalebosConnectionError,
+    BaalebosAPIError,
+    BaalebosAuthError,
+)
+
+
+def _resolve_config(api_url: str, api_key: str):
+    api_url = api_url or os.getenv('BAALEBOS_API_URL')
+    api_key = api_key or os.getenv('BAALEBOS_API_KEY')
+
+    if not api_url:
+        raise BaalebosConfigError(
+            'BAALEBOS_API_URL is missing. Set environment variable or pass to constructor.'
+        )
+    if not api_key:
+        raise BaalebosConfigError(
+            'BAALEBOS_API_KEY is missing. Set environment variable or pass to constructor.'
+        )
+    return api_url, api_key
+
+
+def _build_request(api_key: str, prompt: str, mode: str, temperature: float):
+    headers = {
+        'Content-Type': 'application/json',
+        'x-api-key': api_key,
+    }
+    payload = {
+        'message': prompt,
+        'mode': mode,
+        'temperature': temperature,
+    }
+    return headers, payload
+
+
+def _parse_response(status_code: int, reason: str, data: dict) -> str:
+    if status_code == 401:
+        raise BaalebosAuthError(
+            'Unauthorized - check that BAALEBOS_API_KEY is correct.',
+            status_code=401,
+        )
+    if status_code >= 400:
+        raise BaalebosAPIError(
+            f'Gateway returned {status_code} {reason}',
+            status_code=status_code,
+        )
+
+    if 'data' in data and 'choices' in data['data']:
+        return data['data']['choices'][0]['message']['content']
+    elif 'output' in data:
+        return data['output']
+    return str(data)
+
+
+class BaalebosAI:
+    """Synchronous client - use this in regular scripts and the CLI."""
+
+    def __init__(self, api_url: str = None, api_key: str = None):
+        self.api_url, self.api_key = _resolve_config(api_url, api_key)
+
+    def chat(self, prompt: str, mode: str = 'auto', temperature: float = 0.7) -> str:
+        headers, payload = _build_request(self.api_key, prompt, mode, temperature)
+
+        try:
+            response = requests.post(self.api_url, json=payload, headers=headers, timeout=60)
+        except requests.exceptions.Timeout:
+            raise BaalebosConnectionError(f'Request to {self.api_url} timed out after 60s.')
+        except requests.exceptions.ConnectionError as e:
+            raise BaalebosConnectionError(f'Could not reach {self.api_url}: {e}')
+
+        return _parse_response(response.status_code, response.reason, response.json())
+
+
+class AsyncBaalebosAI:
+    """Async client - use this with `await` in async codebases (e.g. FastAPI,
+    asyncio scripts). Same behavior and same error types as BaalebosAI, just
+    non-blocking. Requires the `httpx` package."""
+
+    def __init__(self, api_url: str = None, api_key: str = None):
+        self.api_url, self.api_key = _resolve_config(api_url, api_key)
+
+    async def chat(self, prompt: str, mode: str = 'auto', temperature: float = 0.7) -> str:
+        headers, payload = _build_request(self.api_key, prompt, mode, temperature)
+
+        try:
+            async with httpx.AsyncClient(timeout=60) as client:
+                response = await client.post(self.api_url, json=payload, headers=headers)
+        except httpx.TimeoutException:
+            raise BaalebosConnectionError(f'Request to {self.api_url} timed out after 60s.')
+        except httpx.ConnectError as e:
+            raise BaalebosConnectionError(f'Could not reach {self.api_url}: {e}')
+
+        return _parse_response(response.status_code, response.reason_phrase, response.json())

baalebos_ai-1.0.0/baalebos_ai/exceptions.py

@@ -0,0 +1,28 @@
+class BaalebosError(Exception):
+    """Base class for all errors raised by this SDK. Catch this to catch anything we raise."""
+    pass
+
+
+class BaalebosConfigError(BaalebosError):
+    """Raised when required configuration (api_url / api_key) is missing."""
+    pass
+
+
+class BaalebosConnectionError(BaalebosError):
+    """Raised when the request never reached the gateway at all
+    (DNS failure, connection refused, timeout)."""
+    pass
+
+
+class BaalebosAPIError(BaalebosError):
+    """Raised when the gateway responded, but with an error status
+    (401 Unauthorized, 429 rate-limited, 5xx server error, etc.)."""
+
+    def __init__(self, message: str, status_code: int = None):
+        super().__init__(message)
+        self.status_code = status_code
+
+
+class BaalebosAuthError(BaalebosAPIError):
+    """Raised specifically for 401 Unauthorized - a bad or missing x-api-key."""
+    pass

baalebos_ai-1.0.0/baalebos_ai.egg-info/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: baalebos-ai
+Version: 1.0.0
+Summary: Official Python client and terminal CLI for the Baalebos AI Gateway
+Home-page: https://github.com/baalebos-cloud/nexus-ai-gateway
+Author: Baalebos AI Team
+License: MIT
+Project-URL: Source, https://github.com/baalebos-cloud/nexus-ai-gateway
+Project-URL: Bug Tracker, https://github.com/baalebos-cloud/nexus-ai-gateway/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests
+Requires-Dist: httpx
+Dynamic: home-page
+Dynamic: requires-python

baalebos_ai-1.0.0/baalebos_ai.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+pyproject.toml
+setup.py
+baalebos_ai/__init__.py
+baalebos_ai/cli.py
+baalebos_ai/client.py
+baalebos_ai/exceptions.py
+baalebos_ai.egg-info/PKG-INFO
+baalebos_ai.egg-info/SOURCES.txt
+baalebos_ai.egg-info/dependency_links.txt
+baalebos_ai.egg-info/entry_points.txt
+baalebos_ai.egg-info/requires.txt
+baalebos_ai.egg-info/top_level.txt

baalebos_ai-1.0.0/baalebos_ai.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

baalebos_ai-1.0.0/baalebos_ai.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ai = baalebos_ai.cli:main

baalebos_ai-1.0.0/baalebos_ai.egg-info/requires.txt

@@ -0,0 +1,2 @@
+requests
+httpx

baalebos_ai-1.0.0/baalebos_ai.egg-info/top_level.txt

@@ -0,0 +1 @@
+baalebos_ai

baalebos_ai-1.0.0/pyproject.toml

@@ -0,0 +1,30 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "baalebos-ai"
+version = "1.0.0"
+description = "Official Python client and terminal CLI for the Baalebos AI Gateway"
+readme = "README.md"
+authors = [
+    { name = "Baalebos AI Team" }
+]
+license = { text = "MIT" }
+requires-python = ">=3.8"
+dependencies = [
+    "requests",
+    "httpx",
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[project.urls]
+Source = "https://github.com/baalebos-cloud/nexus-ai-gateway"
+"Bug Tracker" = "https://github.com/baalebos-cloud/nexus-ai-gateway/issues"
+
+[project.scripts]
+ai = "baalebos_ai.cli:main"

baalebos_ai-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

baalebos_ai-1.0.0/setup.py

@@ -0,0 +1,33 @@
+from setuptools import setup, find_packages
+from pathlib import Path
+
+this_dir = Path(__file__).parent
+long_description = (this_dir / "README.md").read_text(encoding="utf-8") if (this_dir / "README.md").exists() else ""
+
+setup(
+    name='baalebos-ai',
+    version='1.0.0',
+    description='Official Python client and terminal CLI for the Baalebos AI Gateway',
+    long_description=long_description,
+    long_description_content_type='text/markdown',
+    author='Baalebos AI Team',
+    url='https://github.com/baalebos-cloud/nexus-ai-gateway',
+    project_urls={
+        'Source': 'https://github.com/baalebos-cloud/nexus-ai-gateway',
+        'Bug Tracker': 'https://github.com/baalebos-cloud/nexus-ai-gateway/issues',
+    },
+    packages=find_packages(),
+    install_requires=['requests', 'httpx'],
+    python_requires='>=3.8',
+    classifiers=[
+        'Programming Language :: Python :: 3',
+        'License :: OSI Approved :: MIT License',
+        'Operating System :: OS Independent',
+    ],
+    license='MIT',
+    entry_points={
+        'console_scripts': [
+            'ai = baalebos_ai.cli:main',
+        ],
+    },
+)