bos-ai 0.1.0__tar.gz

bos_ai-0.1.0/.gitignore

@@ -0,0 +1,218 @@
+.local/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[codz]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#   Usually these files are written by a python script from a template
+#   before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py.cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+# Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+# uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+# poetry.lock
+# poetry.toml
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#   pdm recommends including project-wide configuration in pdm.toml, but excluding .pdm-python.
+#   https://pdm-project.org/en/latest/usage/project/#working-with-version-control
+# pdm.lock
+# pdm.toml
+.pdm-python
+.pdm-build/
+
+# pixi
+#   Similar to Pipfile.lock, it is generally recommended to include pixi.lock in version control.
+# pixi.lock
+#   Pixi creates a virtual environment in the .pixi directory, just like venv module creates one
+#   in the .local directory. It is recommended not to include this directory in version control.
+.pixi
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# Redis
+*.rdb
+*.aof
+*.pid
+
+# RabbitMQ
+mnesia/
+rabbitmq/
+rabbitmq-data/
+
+# ActiveMQ
+activemq-data/
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.local
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#   JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#   be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#   and can be added to the global gitignore or merged into this file.  For a more nuclear
+#   option (not recommended) you can uncomment the following to ignore the entire idea folder.
+# .idea/
+
+# Abstra
+#   Abstra is an AI-powered process automation framework.
+#   Ignore directories containing user credentials, local state, and settings.
+#   Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#   Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore
+#   that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#   and can be added to the global gitignore or merged into this file. However, if you prefer,
+#   you could uncomment the following to ignore the entire vscode folder
+# .vscode/
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Marimo
+marimo/_static/
+marimo/_lsp/
+__marimo__/
+
+# Streamlit
+.streamlit/secrets.toml

bos_ai-0.1.0/PKG-INFO

@@ -0,0 +1,126 @@
+Metadata-Version: 2.4
+Name: bos-ai
+Version: 0.1.0
+Summary: Lightweight single-file agent framework
+Requires-Python: >=3.13
+Requires-Dist: beautifulsoup4>=4.14.3
+Requires-Dist: click>=8.1.7
+Requires-Dist: filelock>=3.19.1
+Requires-Dist: litellm==1.83.0
+Requires-Dist: python-dotenv>=1.2.2
+Requires-Dist: rich>=13.0.0
+Requires-Dist: textual>=1.0.0
+Requires-Dist: tomli>=2.0.0; python_version < '3.11'
+Provides-Extra: all
+Requires-Dist: oauth-cli-kit>=0.1.3; extra == 'all'
+Provides-Extra: providers
+Requires-Dist: oauth-cli-kit>=0.1.3; extra == 'providers'
+Description-Content-Type: text/markdown
+
+# BOS AI
+
+**Lightweight single-file agent framework**
+
+`bos-ai` is an extensible, actor-based Python framework for building and running autonomous AI agents. Built around a remarkably lean core, it provides everything you need to run, extend, and orchestrate LLM-driven agents with local tool execution, memory, and message passing.
+
+---
+
+## 🚀 Easy Start
+
+1. **Install the package:**
+   ```bash
+   pip install bos-ai
+   ```
+
+2. **Initialize a Workspace:**
+   Navigate into your project directory and initialize the BOS AI workspace:
+   ```bash
+   bos init
+   ```
+   This command creates a `.bos` directory and a `config.toml` file to hold your project's agent configurations.
+
+3. **Start Chatting:**
+   Use the chat command to interact with your agent:
+   ```bash
+   bos chat
+   ```
+
+---
+
+## 💻 CLI Intro
+
+BOS AI ships with a `bos` CLI with lazy-loaded commands to keep startup times incredibly fast:
+
+- **`bos init`**: Bootstraps a new workspace. It creates the `.bos/config.toml` file and provisions necessary data directories.
+- **`bos auth`**: Set up authentication for various LLM providers and utilities.
+- **`bos chat`**: Drops you into an interactive chat application to talk to the agents defined in your configuration. You can also use it in "oneshot" mode.
+
+**LLM Providers via Auth:**
+```bash
+# Chat with the Antigravity provider (requires `bos auth antigravity`)
+bos chat -M "hello" -m antigravity/gemini-3.1-pro-low -a main
+
+# Chat with the Gemini CLI provider (requires `bos auth gemini-cli`)
+bos chat -M "hello" -m gemini-cli/gemini-2.5-flash -a main
+
+# Chat with the Codex provider (requires `bos auth codex`)
+bos chat -M "hello" -m codex/gpt-5.3-codex -a main
+```
+
+**Global Options:**
+- `-w`, `--workspace`: Path to the workspace directory (defaults to `.`).
+
+---
+
+## 🧠 Principles
+
+`bos-ai` is built on a few core design principles:
+
+- **Lightweight & Embeddable**: The heart of the framework lives in a single, comprehensive file (`core.py`), ensuring that complexity is minimized while keeping the abstractions powerfully dense.
+- **Extensible at the Core**: Every significant layer—from LLM providers, to message persistence, to memory and tools—is powered by an internal Extension System.
+- **Agent as an Actor**: Agents communicate via asynchronous message passing (the `Mailbox` protocol). This enables robust multi-agent orchestration without tightly coupled code.
+- **Harness-Managed Lifecycle**: An `AgentHarness` is used to bootstrap, maintain, and gracefully tear down shared resources (like databases or API connections) across all agents in the workspace.
+
+---
+
+## 🔌 Extension Framework
+
+BOS AI utilizes an `ExtensionPoint` pattern for its modular capabilities. You can seamlessly inject your own logic or override defaults decorators.
+
+The framework provides named extension points such as:
+- `@ep_provider`: Connect new LLM backends (OpenAI, Anthropic, Gemini, etc.).
+- `@ep_tool`: Add new conversational tools that the LLM can invoke.
+- `@ep_memory_store`: Connect alternative vector databases or key-value stores.
+- `@ep_message_store`: Custom persistence logic for conversation history.
+- `@ep_mailbox`: Implement distributed message-passing interfaces like Redis or RabbitMQ.
+- `@ep_react_interceptor`: Hooks to orchestrate the internal ReAct loop of an agent.
+
+Example registering a custom tool:
+
+```python
+from bos.core import ep_tool
+
+@ep_tool(
+    name="my_custom_tool",
+    description="Does something awesome.",
+    # ... additional structured metadata ...
+)
+async def my_custom_tool(arg1: str):
+    return f"Processed {arg1}"
+```
+
+To load your extensions, simply add their module paths to the `platform.extensions` array in your workspace's `config.toml`.
+
+---
+
+## ⚙️ Configuration System
+
+BOS AI uses a hierarchical, TOML-based configuration pattern that is specifically designed for isolation per-workspace.
+
+When you run a command, `bos` searches upwards from the current directory to find a `.bos/config.toml` file, eventually falling back to a global `~/.bos/config.toml` if none is found.
+
+### Config Structure (`.bos/config.toml`)
+- **`[platform]`**: Define environment variables, `.env` file locations, and structural extensions loading rules.
+- **`[[platform.agents]]`**: Array of dictionaries defining your agents. You can configure `system_prompt`, limits (`max_tokens`), required `tools`, `skills`, memory definitions, and even `subagents`.
+- **`[harness]`**: Define the overarching services all agents in the environment share. For example, configure the active `memory_store` directory or hook up an interceptor chain.
+- **`[cli]`**: Directives and default options for the command-line application (like specifying a default agent target).

bos_ai-0.1.0/README.md

@@ -0,0 +1,107 @@
+# BOS AI
+
+**Lightweight single-file agent framework**
+
+`bos-ai` is an extensible, actor-based Python framework for building and running autonomous AI agents. Built around a remarkably lean core, it provides everything you need to run, extend, and orchestrate LLM-driven agents with local tool execution, memory, and message passing.
+
+---
+
+## 🚀 Easy Start
+
+1. **Install the package:**
+   ```bash
+   pip install bos-ai
+   ```
+
+2. **Initialize a Workspace:**
+   Navigate into your project directory and initialize the BOS AI workspace:
+   ```bash
+   bos init
+   ```
+   This command creates a `.bos` directory and a `config.toml` file to hold your project's agent configurations.
+
+3. **Start Chatting:**
+   Use the chat command to interact with your agent:
+   ```bash
+   bos chat
+   ```
+
+---
+
+## 💻 CLI Intro
+
+BOS AI ships with a `bos` CLI with lazy-loaded commands to keep startup times incredibly fast:
+
+- **`bos init`**: Bootstraps a new workspace. It creates the `.bos/config.toml` file and provisions necessary data directories.
+- **`bos auth`**: Set up authentication for various LLM providers and utilities.
+- **`bos chat`**: Drops you into an interactive chat application to talk to the agents defined in your configuration. You can also use it in "oneshot" mode.
+
+**LLM Providers via Auth:**
+```bash
+# Chat with the Antigravity provider (requires `bos auth antigravity`)
+bos chat -M "hello" -m antigravity/gemini-3.1-pro-low -a main
+
+# Chat with the Gemini CLI provider (requires `bos auth gemini-cli`)
+bos chat -M "hello" -m gemini-cli/gemini-2.5-flash -a main
+
+# Chat with the Codex provider (requires `bos auth codex`)
+bos chat -M "hello" -m codex/gpt-5.3-codex -a main
+```
+
+**Global Options:**
+- `-w`, `--workspace`: Path to the workspace directory (defaults to `.`).
+
+---
+
+## 🧠 Principles
+
+`bos-ai` is built on a few core design principles:
+
+- **Lightweight & Embeddable**: The heart of the framework lives in a single, comprehensive file (`core.py`), ensuring that complexity is minimized while keeping the abstractions powerfully dense.
+- **Extensible at the Core**: Every significant layer—from LLM providers, to message persistence, to memory and tools—is powered by an internal Extension System.
+- **Agent as an Actor**: Agents communicate via asynchronous message passing (the `Mailbox` protocol). This enables robust multi-agent orchestration without tightly coupled code.
+- **Harness-Managed Lifecycle**: An `AgentHarness` is used to bootstrap, maintain, and gracefully tear down shared resources (like databases or API connections) across all agents in the workspace.
+
+---
+
+## 🔌 Extension Framework
+
+BOS AI utilizes an `ExtensionPoint` pattern for its modular capabilities. You can seamlessly inject your own logic or override defaults decorators.
+
+The framework provides named extension points such as:
+- `@ep_provider`: Connect new LLM backends (OpenAI, Anthropic, Gemini, etc.).
+- `@ep_tool`: Add new conversational tools that the LLM can invoke.
+- `@ep_memory_store`: Connect alternative vector databases or key-value stores.
+- `@ep_message_store`: Custom persistence logic for conversation history.
+- `@ep_mailbox`: Implement distributed message-passing interfaces like Redis or RabbitMQ.
+- `@ep_react_interceptor`: Hooks to orchestrate the internal ReAct loop of an agent.
+
+Example registering a custom tool:
+
+```python
+from bos.core import ep_tool
+
+@ep_tool(
+    name="my_custom_tool",
+    description="Does something awesome.",
+    # ... additional structured metadata ...
+)
+async def my_custom_tool(arg1: str):
+    return f"Processed {arg1}"
+```
+
+To load your extensions, simply add their module paths to the `platform.extensions` array in your workspace's `config.toml`.
+
+---
+
+## ⚙️ Configuration System
+
+BOS AI uses a hierarchical, TOML-based configuration pattern that is specifically designed for isolation per-workspace.
+
+When you run a command, `bos` searches upwards from the current directory to find a `.bos/config.toml` file, eventually falling back to a global `~/.bos/config.toml` if none is found.
+
+### Config Structure (`.bos/config.toml`)
+- **`[platform]`**: Define environment variables, `.env` file locations, and structural extensions loading rules.
+- **`[[platform.agents]]`**: Array of dictionaries defining your agents. You can configure `system_prompt`, limits (`max_tokens`), required `tools`, `skills`, memory definitions, and even `subagents`.
+- **`[harness]`**: Define the overarching services all agents in the environment share. For example, configure the active `memory_store` directory or hook up an interceptor chain.
+- **`[cli]`**: Directives and default options for the command-line application (like specifying a default agent target).

bos_ai-0.1.0/pyproject.toml

@@ -0,0 +1,46 @@
+[project]
+name = "bos-ai"
+version = "0.1.0"
+description = "Lightweight single-file agent framework"
+readme = "README.md"
+requires-python = ">=3.13"
+dependencies = [
+    "litellm==1.83.0",
+    "python-dotenv>=1.2.2",
+    "filelock>=3.19.1",
+    "click>=8.1.7",
+    "rich>=13.0.0",
+    "textual>=1.0.0",
+    "tomli>=2.0.0; python_version<'3.11'",
+    "beautifulsoup4>=4.14.3",
+]
+
+[project.scripts]
+bos = "bos.cli.entry:main"
+
+[project.optional-dependencies]
+providers = [
+    "oauth-cli-kit>=0.1.3",
+]
+all = [
+    "oauth-cli-kit>=0.1.3",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/bos"]
+
+[dependency-groups]
+dev = [
+    "pytest>=8.4.1",
+    "pytest-asyncio>=1.1.0",
+    "pytest-cov>=6.2.1",
+    "ruff>=0.12.4",
+]
+
+[tool.ruff]
+line-length = 120
+lint.select = ["E", "F", "I"]

bos_ai-0.1.0/src/bos/cli/commands/auth.py

@@ -0,0 +1,130 @@
+from pathlib import Path
+
+import click
+
+from bos.core import _flock
+
+
+@click.group(name="auth")
+def auth():
+    """Manage authentication for BOS AI."""
+    pass
+
+
+@auth.command(name="codex")
+@click.option("--name", default="default", help="Name of the credential.")
+def codex(name: str):
+    """Authenticate with OpenAI Codex."""
+    try:
+        from oauth_cli_kit import get_token, login_oauth_interactive
+    except ImportError:
+        raise click.ClickException("oauth_cli_kit not installed. Run: uv pip install oauth-cli-kit")
+
+    creds_dir = Path.home() / ".config" / "bos"
+    creds_dir.mkdir(parents=True, exist_ok=True)
+
+    token_path = creds_dir / f"codex_auth.{name}.json"
+    from oauth_cli_kit.storage import FileTokenStorage
+
+    storage = FileTokenStorage(str(token_path))
+
+    token = None
+    try:
+        token = get_token(storage=storage)
+    except Exception:
+        pass
+
+    if not (token and token.access):
+        click.echo("Starting interactive OAuth login...\n")
+        token = login_oauth_interactive(
+            print_fn=lambda s: click.echo(s),
+            prompt_fn=lambda s: click.prompt(s),
+            originator="tradingdesk",
+            storage=storage,
+        )
+
+    if not (token and token.access):
+        raise click.ClickException("Authentication failed")
+
+    click.echo(f"\n✓ Authenticated with OpenAI Codex ({getattr(token, 'account_id', '') or name})")
+    click.echo(f"  Credentials saved to: {token_path}")
+
+
+@auth.command(name="antigravity")
+@click.option("--name", default="default", help="Name of the credential.")
+def antigravity(name: str):
+    """Authenticate with Google Antigravity."""
+    import asyncio
+    import json
+    import webbrowser
+
+    from bos.providers.antigravity_provider import login_antigravity
+
+    creds_dir = Path.home() / ".config" / "bos"
+    creds_dir.mkdir(parents=True, exist_ok=True)
+    token_path = creds_dir / f"antigravity_auth.{name}.json"
+
+    def on_url(url: str, msg: str):
+        click.echo(f"{msg}\n\n  {url}\n")
+        try:
+            webbrowser.open(url)
+        except Exception:
+            pass
+
+    try:
+        creds = asyncio.run(
+            login_antigravity(
+                on_auth=on_url,
+                on_progress=lambda msg: click.echo(f"  {msg}"),
+            )
+        )
+        with _flock(token_path):
+            token_path.write_text(json.dumps(dict(creds.__dict__)))
+        click.echo(f"\n✓ Authenticated with Google Antigravity ({creds.email or ''})")
+        click.echo(f"  Credentials saved to: {token_path}")
+    except Exception as e:
+        raise click.ClickException(f"Authentication failed: {e}")
+
+
+@auth.command(name="gemini-cli")
+@click.option("--name", default="default", help="Name of the credential.")
+def gemini_cli(name: str) -> None:
+    """Authenticate with Gemini CLI.
+
+    This command initiates an interactive OAuth 2.0 PKCE flow to authenticate with
+    Google Cloud Code Assist using your personal Google account.
+    """
+    import asyncio
+    import json
+    import webbrowser
+
+    from bos.providers.gemini_cli_provider import login_gemini_cli
+
+    click.echo("\n" + "=" * 50)
+    click.echo(" Gemini CLI Authentication")
+    click.echo("=" * 50)
+
+    creds_dir = Path.home() / ".config" / "bos"
+    creds_dir.mkdir(parents=True, exist_ok=True)
+    token_path = creds_dir / f"gemini_cli_auth.{name}.json"
+
+    def on_url(url: str, msg: str):
+        click.echo(f"{msg}\n\n  {url}\n")
+        try:
+            webbrowser.open(url)
+        except Exception:
+            pass
+
+    try:
+        creds = asyncio.run(
+            login_gemini_cli(
+                on_auth=on_url,
+                on_progress=lambda msg: click.echo(f"  {msg}"),
+            )
+        )
+        with _flock(token_path):
+            token_path.write_text(json.dumps(dict(creds.__dict__)))
+        click.echo(f"\n✓ Authenticated with Gemini CLI ({creds.email or ''})")
+        click.echo(f"  Credentials saved to: {token_path}")
+    except Exception as e:
+        raise click.ClickException(f"Authentication failed: {e}")