openbb-agent-server 0.1.0__tar.gz

openbb_agent_server-0.1.0/.gitignore

@@ -0,0 +1,43 @@
+# --- Python ---
+__pycache__/
+*.py[cod]
+.venv/
+*.egg-info/
+build/
+dist/
+.coverage
+.coverage.*
+coverage.xml
+htmlcov/
+.pytest_cache/
+.ruff_cache/
+.ty_cache/
+.mypy_cache/
+
+# --- Docs site (Docusaurus, in ./website) ---
+node_modules/
+website/build/
+website/.docusaurus/
+website/.cache-loader/
+npm-debug.log*
+yarn-error.log*
+
+# Auto-generated API reference: rebuilt from the package docstrings by
+# ``npm run gen-api`` (wiped + regenerated on every run, never
+# hand-edited). The deploy workflow regenerates it in CI, so it is not
+# tracked here.
+/docs/reference/
+
+# --- Secrets / local environment ---
+.env
+.env.*
+!.env.example
+
+# --- Editors / OS ---
+.vscode/
+.idea/
+.DS_Store
+Thumbs.db
+
+# --- Local agent tooling ---
+.claude/

openbb_agent_server-0.1.0/PKG-INFO

@@ -0,0 +1,137 @@
+Metadata-Version: 2.4
+Name: openbb-agent-server
+Version: 0.1.0
+Summary: OpenBB Agent Server: pluggable, multi-tenant agent backend wired to the OpenBB Workspace UI.
+Project-URL: Homepage, https://github.com/deeleeramone/openbb-agent-server
+Project-URL: Repository, https://github.com/deeleeramone/openbb-agent-server
+Author-email: Danglewood <85772166+deeleeramone@users.noreply.github.com>
+License: AGPL-3.0-only
+Requires-Python: <4,>=3.11
+Requires-Dist: aiosqlite>=0.20
+Requires-Dist: alembic>=1.13
+Requires-Dist: argon2-cffi>=23.1
+Requires-Dist: ddgs>=9.0
+Requires-Dist: deepagents>=0.5.0
+Requires-Dist: fastapi>=0.115.0
+Requires-Dist: fastmcp>=3.2.0
+Requires-Dist: httpx2>=0.1
+Requires-Dist: langchain-core>=1.0.0
+Requires-Dist: langchain-mcp-adapters>=0.2.0
+Requires-Dist: langchain-nvidia-ai-endpoints>=1.0.0
+Requires-Dist: langchain-text-splitters>=0.3
+Requires-Dist: langchain>=1.0.0
+Requires-Dist: langgraph-checkpoint-sqlite>=2.0
+Requires-Dist: langgraph>=1.0.0
+Requires-Dist: pdfplumber>=0.11
+Requires-Dist: pydantic-settings>=2.5
+Requires-Dist: pydantic<3,>=2.7
+Requires-Dist: pyjwt[crypto]>=2.9
+Requires-Dist: python-dotenv>=1.0
+Requires-Dist: sqlalchemy[asyncio]<3,>=2.0
+Requires-Dist: sqlite-vec>=0.1.6
+Requires-Dist: sse-starlette>=2.1.0
+Requires-Dist: uuid7>=0.1
+Requires-Dist: uvicorn>=0.32.0
+Provides-Extra: all
+Requires-Dist: google-genai>=1.0.0; extra == 'all'
+Requires-Dist: langchain-anthropic>=1.0.0; extra == 'all'
+Requires-Dist: langchain-aws>=0.2.0; extra == 'all'
+Requires-Dist: langchain-google-genai>=2.0.0; extra == 'all'
+Requires-Dist: langchain-groq>=0.2.0; extra == 'all'
+Requires-Dist: langchain-openai>=0.2.0; extra == 'all'
+Requires-Dist: langgraph-checkpoint-postgres>=2.0; extra == 'all'
+Requires-Dist: pgvector>=0.3; extra == 'all'
+Requires-Dist: psycopg[binary]>=3.2; extra == 'all'
+Requires-Dist: tavily-python>=0.5; extra == 'all'
+Provides-Extra: anthropic
+Requires-Dist: langchain-anthropic>=1.0.0; extra == 'anthropic'
+Provides-Extra: bedrock
+Requires-Dist: langchain-aws>=0.2.0; extra == 'bedrock'
+Provides-Extra: google-genai
+Requires-Dist: google-genai>=1.0.0; extra == 'google-genai'
+Requires-Dist: langchain-google-genai>=2.0.0; extra == 'google-genai'
+Provides-Extra: groq
+Requires-Dist: langchain-groq>=0.2.0; extra == 'groq'
+Provides-Extra: openai
+Requires-Dist: langchain-openai>=0.2.0; extra == 'openai'
+Provides-Extra: postgres
+Requires-Dist: langgraph-checkpoint-postgres>=2.0; extra == 'postgres'
+Requires-Dist: pgvector>=0.3; extra == 'postgres'
+Requires-Dist: psycopg[binary]>=3.2; extra == 'postgres'
+Provides-Extra: pywry
+Requires-Dist: pywry>=2.0.4; extra == 'pywry'
+Provides-Extra: tavily
+Requires-Dist: tavily-python>=0.5; extra == 'tavily'
+Provides-Extra: vertex
+Requires-Dist: google-genai>=1.0.0; extra == 'vertex'
+Requires-Dist: langchain-google-genai>=2.0.0; extra == 'vertex'
+Description-Content-Type: text/markdown
+
+# openbb-agent-server
+
+Pluggable, multi-tenant agent backend that speaks the [OpenBB Workspace
+custom-agent SSE protocol](https://docs.openbb.co/workspace/developers/agents-integration) and runs the agent loop
+on top of the [LangChain DeepAgents harness](https://docs.langchain.com/oss/python/deepagents/overview). One process
+hosts many agent profiles; auth, model provider, tools, sub-agents,
+middleware, checkpointer, and persistence are independent plugin axes
+— anything can be swapped without forking the package.
+
+<img width="1580" height="834" alt="openbb-agent-server-screenshot" src="https://github.com/user-attachments/assets/f6ed834a-4232-4cbb-a039-275f30dafad9" />
+
+The full OpenBB Platform — every command across every installed
+provider — is reachable via the optional `mcp_local` tool source,
+which spawns the
+[`openbb-mcp-server`](https://github.com/OpenBB-finance/OpenBB/tree/main/openbb_platform/extensions/mcp_server)
+extension over stdio.
+
+The default setup uses 100% free tokens and embedding models available from [NVIDIA](https://build.nvidia.com/) by registering for an API key [here](https://developer.nvidia.com/login)
+
+## Install & run
+
+```bash
+# from PyPI
+pip install openbb-agent-server
+
+# from a checkout of this repository
+pip install -e '.[workspace-mcp]'
+# add or combine more extras: [anthropic] [openai] [bedrock]
+#                             [vertex] [google_genai] [groq]
+#                             [tavily] [postgres]
+#                             [pywry]  (desktop chat embedding)
+
+export NVIDIA_API_KEY=...
+
+openbb-agent-server
+```
+
+In OpenBB Workspace, add a custom agent pointing at
+`http://localhost:8010`. Workspace fetches this once and
+reads every agent profile the server registers in a single payload.
+
+The `[workspace-mcp]` extra installs
+[openbb-workspace-mcp](https://github.com/OpenBB-finance/workspace-mcp)
+from its GitHub zip (Python ≥3.13 only). To run it in-process and skip
+the separate `workspace-mcp` sidecar, set `mount_workspace_mcp = true`
+in `openbb.toml` after installing the extra, then point the Workspace
+UI's MCP-servers setting at `http://localhost:8010/mcp/workspace/mcp`.
+The mount is **opt-in** (default `false`) so installing the extra alone
+does not change the server's behavior.
+
+For production, generate the config template and edit it:
+
+```bash
+openbb-agent-server --generate-config /etc/openbb/openbb.toml
+openbb-agent-server --config-file /etc/openbb/openbb.toml --host 0.0.0.0
+```
+
+## Documentation
+
+Documentation currently lives in [`docs/`](docs/README.md), and may move in the future:
+
+| Audience                     | Start here                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                  |
+| ---------------------------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| First-time user              | [Getting started](docs/guides/getting-started.md) → [Architecture](docs/guides/architecture.md) → [Workspace integration](docs/guides/workspace-integration.md)                                                                                                                                                                                                                                                                                                                                            |
+| Desktop embedding            | [PyWry chat (ACP)](docs/guides/pywry-chat.md) — `openbb-agent-canvas` opens a window whose main page is the agent's live canvas (charts, tables, HTML) with the chat attached; or attach the chat to any PyWry widget via the `[pywry]` extra. Same `openbb.toml`, same loop, no HTTP server                                                                                                                                                                                                                |
+| Operator / SRE               | [Configuration](docs/operating/configuration.md) → [Auth](docs/operating/auth.md) → [Persistence](docs/operating/persistence.md) → [Observability](docs/operating/observability.md)                                                                                                                                                                                                                                                                                                                        |
+| Plugin author                | [Plugin system](docs/developing/plugin-system.md) → writing a [tool source](docs/developing/writing-a-tool-source.md) / [model provider](docs/developing/writing-a-model-provider.md) / [middleware](docs/developing/writing-a-middleware.md) / [sub-agent](docs/developing/writing-a-subagent.md) / [auth backend](docs/developing/writing-an-auth-backend.md) → [Conventions](docs/developing/conventions.md) → [Testing](docs/developing/testing.md)                                                     |
+| API lookup                   | [Reference](docs/reference/) — module-by-module, mirrors the package tree                                                                                                                                                                                                                                                                                                                                                                                                                                  |

openbb_agent_server-0.1.0/README.md

@@ -0,0 +1,68 @@
+# openbb-agent-server
+
+Pluggable, multi-tenant agent backend that speaks the [OpenBB Workspace
+custom-agent SSE protocol](https://docs.openbb.co/workspace/developers/agents-integration) and runs the agent loop
+on top of the [LangChain DeepAgents harness](https://docs.langchain.com/oss/python/deepagents/overview). One process
+hosts many agent profiles; auth, model provider, tools, sub-agents,
+middleware, checkpointer, and persistence are independent plugin axes
+— anything can be swapped without forking the package.
+
+<img width="1580" height="834" alt="openbb-agent-server-screenshot" src="https://github.com/user-attachments/assets/f6ed834a-4232-4cbb-a039-275f30dafad9" />
+
+The full OpenBB Platform — every command across every installed
+provider — is reachable via the optional `mcp_local` tool source,
+which spawns the
+[`openbb-mcp-server`](https://github.com/OpenBB-finance/OpenBB/tree/main/openbb_platform/extensions/mcp_server)
+extension over stdio.
+
+The default setup uses 100% free tokens and embedding models available from [NVIDIA](https://build.nvidia.com/) by registering for an API key [here](https://developer.nvidia.com/login)
+
+## Install & run
+
+```bash
+# from PyPI
+pip install openbb-agent-server
+
+# from a checkout of this repository
+pip install -e '.[workspace-mcp]'
+# add or combine more extras: [anthropic] [openai] [bedrock]
+#                             [vertex] [google_genai] [groq]
+#                             [tavily] [postgres]
+#                             [pywry]  (desktop chat embedding)
+
+export NVIDIA_API_KEY=...
+
+openbb-agent-server
+```
+
+In OpenBB Workspace, add a custom agent pointing at
+`http://localhost:8010`. Workspace fetches this once and
+reads every agent profile the server registers in a single payload.
+
+The `[workspace-mcp]` extra installs
+[openbb-workspace-mcp](https://github.com/OpenBB-finance/workspace-mcp)
+from its GitHub zip (Python ≥3.13 only). To run it in-process and skip
+the separate `workspace-mcp` sidecar, set `mount_workspace_mcp = true`
+in `openbb.toml` after installing the extra, then point the Workspace
+UI's MCP-servers setting at `http://localhost:8010/mcp/workspace/mcp`.
+The mount is **opt-in** (default `false`) so installing the extra alone
+does not change the server's behavior.
+
+For production, generate the config template and edit it:
+
+```bash
+openbb-agent-server --generate-config /etc/openbb/openbb.toml
+openbb-agent-server --config-file /etc/openbb/openbb.toml --host 0.0.0.0
+```
+
+## Documentation
+
+Documentation currently lives in [`docs/`](docs/README.md), and may move in the future:
+
+| Audience                     | Start here                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                  |
+| ---------------------------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
+| First-time user              | [Getting started](docs/guides/getting-started.md) → [Architecture](docs/guides/architecture.md) → [Workspace integration](docs/guides/workspace-integration.md)                                                                                                                                                                                                                                                                                                                                            |
+| Desktop embedding            | [PyWry chat (ACP)](docs/guides/pywry-chat.md) — `openbb-agent-canvas` opens a window whose main page is the agent's live canvas (charts, tables, HTML) with the chat attached; or attach the chat to any PyWry widget via the `[pywry]` extra. Same `openbb.toml`, same loop, no HTTP server                                                                                                                                                                                                                |
+| Operator / SRE               | [Configuration](docs/operating/configuration.md) → [Auth](docs/operating/auth.md) → [Persistence](docs/operating/persistence.md) → [Observability](docs/operating/observability.md)                                                                                                                                                                                                                                                                                                                        |
+| Plugin author                | [Plugin system](docs/developing/plugin-system.md) → writing a [tool source](docs/developing/writing-a-tool-source.md) / [model provider](docs/developing/writing-a-model-provider.md) / [middleware](docs/developing/writing-a-middleware.md) / [sub-agent](docs/developing/writing-a-subagent.md) / [auth backend](docs/developing/writing-an-auth-backend.md) → [Conventions](docs/developing/conventions.md) → [Testing](docs/developing/testing.md)                                                     |
+| API lookup                   | [Reference](docs/reference/) — module-by-module, mirrors the package tree                                                                                                                                                                                                                                                                                                                                                                                                                                  |

openbb_agent_server-0.1.0/openbb_agent_server/__init__.py

@@ -0,0 +1,17 @@
+"""OpenBB Platform Agent Server — pluggable, multi-tenant agent backend."""
+
+from __future__ import annotations
+
+import warnings
+
+__version__ = "0.1.0"
+
+from langchain_core._api.deprecation import (  # noqa: E402
+    LangChainPendingDeprecationWarning,
+)
+
+warnings.filterwarnings(
+    "ignore",
+    message=r"The default value of `allowed_objects` will change.*",
+    category=LangChainPendingDeprecationWarning,
+)

openbb_agent_server-0.1.0/openbb_agent_server/_vendor/sqlitevec.py

@@ -0,0 +1,135 @@
+"""Vendored SQLiteVec vector store.
+
+Adapted from ``langchain_community.vectorstores.sqlitevec`` (MIT licence)
+to remove the ``langchain-community`` dependency, which is now archived.
+
+Only the surface used by this project is kept:
+
+* ``create_table_if_not_exists``
+* ``add_texts``
+* ``similarity_search_with_score``
+
+The class intentionally does **not** manage connection lifecycle — the
+caller is responsible for opening and closing the ``sqlite3.Connection``.
+"""
+
+from __future__ import annotations
+
+import json
+import sqlite3
+import struct
+from typing import Any
+
+from langchain_core.documents import Document
+from langchain_core.embeddings import Embeddings
+
+
+def _serialize_f32(vector: list[float]) -> bytes:
+    """Pack a float vector into the raw-bytes format sqlite-vec expects."""
+    return struct.pack(f"{len(vector)}f", *vector)
+
+
+class SQLiteVec:
+    """Thin wrapper around a ``sqlite-vec`` virtual table.
+
+    Requires a pre-configured ``sqlite3.Connection`` with the ``sqlite-vec``
+    extension already loaded.
+    """
+
+    def __init__(
+        self,
+        table: str,
+        connection: sqlite3.Connection,
+        embedding: Embeddings,
+        db_file: str = "vec.db",
+    ) -> None:
+        self._connection = connection
+        self._table = table
+        self._embedding = embedding
+
+    # ------------------------------------------------------------------
+    # Schema helpers
+    # ------------------------------------------------------------------
+
+    def create_table_if_not_exists(self) -> None:
+        """Create the data table, vec0 virtual table, and insert trigger."""
+        dim = len(self._embedding.embed_query("dim probe"))
+        self._connection.execute(
+            f"CREATE TABLE IF NOT EXISTS {self._table}"
+            " (rowid INTEGER PRIMARY KEY AUTOINCREMENT,"
+            "  text TEXT, metadata BLOB, text_embedding BLOB)"
+        )
+        self._connection.execute(
+            f"CREATE VIRTUAL TABLE IF NOT EXISTS {self._table}_vec"
+            f" USING vec0(rowid INTEGER PRIMARY KEY,"
+            f" text_embedding float[{dim}])"
+        )
+        self._connection.execute(
+            f"CREATE TRIGGER IF NOT EXISTS {self._table}_embed_text "
+            f"AFTER INSERT ON {self._table} BEGIN"
+            f"  INSERT INTO {self._table}_vec(rowid, text_embedding)"
+            f"  VALUES (new.rowid, new.text_embedding); END;"
+        )
+        self._connection.commit()
+
+    # ------------------------------------------------------------------
+    # Write
+    # ------------------------------------------------------------------
+
+    def add_texts(
+        self,
+        texts: list[str],
+        metadatas: list[dict[str, Any]],
+        **kwargs: Any,
+    ) -> list[int]:
+        """Embed *texts* and insert them into the store.
+
+        Returns the ``rowid`` values of the newly inserted rows.
+        """
+        row = self._connection.execute(
+            f"SELECT max(rowid) AS rowid FROM {self._table}"
+        ).fetchone()
+        max_id: int = row["rowid"] if row["rowid"] is not None else 0
+
+        embeds = self._embedding.embed_documents(list(texts))
+
+        self._connection.executemany(
+            f"INSERT INTO {self._table}(text, metadata, text_embedding)"
+            " VALUES (?, ?, ?)",
+            [
+                (text, json.dumps(meta), _serialize_f32(emb))
+                for text, meta, emb in zip(texts, metadatas, embeds)
+            ],
+        )
+        self._connection.commit()
+
+        rows = self._connection.execute(
+            f"SELECT rowid FROM {self._table} WHERE rowid > {max_id}"
+        )
+        return [r["rowid"] for r in rows]
+
+    # ------------------------------------------------------------------
+    # Read
+    # ------------------------------------------------------------------
+
+    def similarity_search_with_score(
+        self, query: str, k: int = 4, **kwargs: Any
+    ) -> list[tuple[Document, float]]:
+        """Return the *k* closest documents with their distance scores."""
+        embedding = self._embedding.embed_query(query)
+        cursor = self._connection.cursor()
+        cursor.execute(
+            f"SELECT text, metadata, distance"
+            f" FROM {self._table} AS e"
+            f" INNER JOIN {self._table}_vec AS v ON v.rowid = e.rowid"
+            f" WHERE v.text_embedding MATCH ? AND k = ?"
+            f" ORDER BY distance",
+            [_serialize_f32(embedding), k],
+        )
+        results: list[tuple[Document, float]] = []
+        for row in cursor.fetchall():
+            meta = json.loads(row["metadata"]) if row["metadata"] else {}
+            results.append(
+                (Document(page_content=row["text"], metadata=meta), row["distance"])
+            )
+        return results

openbb_agent_server-0.1.0/openbb_agent_server/acp/__init__.py

@@ -0,0 +1,32 @@
+"""ACP shim — expose the agent loop to PyWry chat components.
+
+Lazy re-exports so ``import openbb_agent_server.acp`` works without
+pywry installed; the ImportError with install instructions surfaces on
+first attribute access instead. ``PyWryCanvas`` / ``build_canvas_html``
+are pywry-free and always importable.
+"""
+
+from __future__ import annotations
+
+from importlib import import_module
+from typing import Any
+
+_EXPORTS: dict[str, str] = {
+    "OpenBBAgentProvider": "openbb_agent_server.acp.provider",
+    "create_chat_manager": "openbb_agent_server.acp.provider",
+    "translate_sse": "openbb_agent_server.acp.provider",
+    "launch": "openbb_agent_server.acp.canvas_app",
+    "CanvasApp": "openbb_agent_server.acp.canvas_app",
+    "PyWryCanvas": "openbb_agent_server.acp.canvas",
+    "build_canvas_html": "openbb_agent_server.acp.canvas",
+}
+
+__all__ = list(_EXPORTS)
+
+
+def __getattr__(name: str) -> Any:
+    """Defer submodule imports until a symbol is actually used."""
+    module_path = _EXPORTS.get(name)
+    if module_path is None:
+        raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+    return getattr(import_module(module_path), name)