mito-ai-core 0.1.1__tar.gz

mito_ai_core-0.1.1/.gitignore

@@ -0,0 +1,139 @@
+# Ignore the uv.lock file that Cursor keeps making
+uv.lock
+
+# Ignore all the ipynb files in mitosheet and mito-ai
+mitosheet/**/*.ipynb
+mito-ai/**/*.ipynb
+
+# Ignore all the ipynb_checkpoints in mitosheet and mito-ai
+mitosheet/.ipynb_checkpoints/*
+mito-ai/.ipynb_checkpoints/*
+tests/.ipynb_checkpoints/*
+jupyterhub/.ipynb_checkpoints/*
+.ipynb_checkpoints/*
+
+# Ignore all the iframe_figures in mitosheet and mito-ai
+# These get generated when creating plotly graphs
+mitosheet/iframe_figures/*
+mito-ai/iframe_figures/*
+mito-sql-cell
+
+# Evals
+evals/reports/
+
+# Other 
+mito-ai-mcp/*.mcpb
+
+## General
+.DS_Store
+.AppleDouble
+.LSOverride
+.env
+
+## Icon must end with two \r
+Icon
+
+## Thumbnails
+._*
+
+
+## Files that might appear in the root of a volume
+.DocumentRevisions-V100
+.fseventsd
+.Spotlight-V100
+.TemporaryItems
+.Trashes
+.VolumeIcon.icns
+.com.apple.timemachine.donotpresent
+
+## Directories potentially created on remote AFP share
+.AppleDB
+.AppleDesktop
+Network Trash Folder
+Temporary Items
+.apdisk
+
+# Windows
+
+## Windows thumbnail cache files
+Thumbs.db
+Thumbs.db:encryptable
+ehthumbs.db
+ehthumbs_vista.db
+
+## Dump file
+*.stackdump
+
+## Folder config file
+[Dd]esktop.ini
+
+## Recycle Bin used on file shares
+$RECYCLE.BIN/
+
+## Windows Installer files
+*.cab
+*.msi
+*.msix
+*.msm
+*.msp
+
+## Windows shortcuts
+*.lnk
+
+## Python
+
+# Virtual enviornemnts
+venv/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+
+# C extensions
+*.so
+
+# Distribution / packaging
+bin/
+build/
+develop-eggs/
+dist/
+eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+node_modules/
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+.tox/
+.coverage
+.cache
+nosetests.xml
+coverage.xml
+
+# Translations
+*.mo
+
+# Mr Developer
+.mr.developer.cfg
+.project
+.pydevproject
+
+# Rope
+.ropeproject
+
+# Django stuff:
+*.log
+*.pot
+
+# Sphinx documentation
+docs/_build/
+mito-sql-cell/junit.xml

mito_ai_core-0.1.1/PKG-INFO

@@ -0,0 +1,117 @@
+Metadata-Version: 2.4
+Name: mito-ai-core
+Version: 0.1.1
+Summary: Pure-Python AI backbone for Mito – providers, models, prompts, and utilities with zero Jupyter Server dependencies.
+Author: Saga Inc.
+License-Expression: AGPL-3.0-only
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Python: >=3.9
+Requires-Dist: aiohttp>=3.8.0
+Requires-Dist: analytics-python
+Requires-Dist: anthropic>=0.6.81
+Requires-Dist: google-genai
+Requires-Dist: jinja2>=3.0.3
+Requires-Dist: openai>=1.0.0
+Requires-Dist: pydantic
+Requires-Dist: requests>=2.25.0
+Requires-Dist: sseclient-py>=1.8.0
+Requires-Dist: tornado>=6.2.0
+Provides-Extra: litellm
+Requires-Dist: litellm; extra == 'litellm'
+Provides-Extra: test
+Requires-Dist: mypy; extra == 'test'
+Requires-Dist: pytest; extra == 'test'
+Requires-Dist: pytest-asyncio; extra == 'test'
+Requires-Dist: pytest-timeout; extra == 'test'
+Requires-Dist: types-requests; extra == 'test'
+Description-Content-Type: text/markdown
+
+# mito-ai-core
+
+The shared Python AI layer for Mito -- LLM providers, models, prompts, and utils. No Jupyter Server or Tornado dependency.
+
+`mito-ai` (the JupyterLab extension) depends on this package for all its AI logic. If you're building something else on top of Mito's AI stack, this is the package you want.
+
+## Install
+
+```bash
+pip install mito-ai-core
+```
+
+## Development
+
+### Working on mito-ai-core only
+
+```bash
+cd mito-ai-core
+pip install -e ".[test]"
+```
+
+### Working on both mito-ai-core and mito-ai at the same time
+
+Install both packages in editable mode so changes in either package are picked up immediately without reinstalling:
+
+```bash
+# From the repo root
+pip install -e ./mito-ai-core
+pip install -e "./mito-ai[test]"
+```
+
+**Order matters!** Install `mito-ai-core` first since `mito-ai` depends on it. After this, any change you make to a `.py` file in either package takes effect on the next import (no rebuild needed).
+
+### How it relates to mito-ai
+
+`mito-ai-core` contains all the pure-Python AI logic -- providers, models, prompts, message history, utils.
+
+`mito-ai` is the JupyterLab extension that adds WebSocket/REST handlers, UI integration, and Streamlit conversion on top of it.
+
+If you're changing provider logic, prompt templates, or models, you're working here. If you're changing how the extension talks to the frontend or handles HTTP requests, you're working in `mito-ai`.
+
+## Usage
+
+Everything goes through `ProviderManager`:
+
+```python
+from mito_ai_core.provider_manager import ProviderManager
+from mito_ai_core.completions.models import MessageType
+
+pm = ProviderManager()
+pm.set_selected_model("gpt-4.1")
+
+# One-shot
+response = await pm.request_completions(
+    message_type=MessageType.CHAT,
+    messages=[{"role": "user", "content": "Hello!"}],
+)
+
+# Streaming -- pass a callback
+response = await pm.stream_completions(
+    message_type=MessageType.CHAT,
+    messages=[{"role": "user", "content": "Hello!"}],
+    message_id="msg-1",
+    thread_id="thread-1",
+    reply_fn=lambda chunk: print(chunk),
+)
+```
+
+`ProviderManager` picks the right client (OpenAI, Anthropic, Gemini, Copilot, LiteLLM, Abacus) based on the selected model. It handles retries, telemetry, and token logging.
+
+This is a library, not a server. The caller owns the transport (WebSocket, HTTP, CLI, whatever).
+
+## Key modules
+
+- **`provider_manager`** -- routes requests to the right LLM client
+- **`clients/`** -- OpenAI, Anthropic, Gemini, Copilot wrappers
+- **`completions/models`** -- all the dataclasses and Pydantic models
+- **`completions/message_history`** -- thread-safe chat persistence
+- **`completions/prompt_builders/`** -- prompt construction for chat, agent, debug, inline completion, charts
+- **`utils/`** -- model resolution, token estimation, telemetry, rate limits
+

mito_ai_core-0.1.1/README.md

@@ -0,0 +1,81 @@
+# mito-ai-core
+
+The shared Python AI layer for Mito -- LLM providers, models, prompts, and utils. No Jupyter Server or Tornado dependency.
+
+`mito-ai` (the JupyterLab extension) depends on this package for all its AI logic. If you're building something else on top of Mito's AI stack, this is the package you want.
+
+## Install
+
+```bash
+pip install mito-ai-core
+```
+
+## Development
+
+### Working on mito-ai-core only
+
+```bash
+cd mito-ai-core
+pip install -e ".[test]"
+```
+
+### Working on both mito-ai-core and mito-ai at the same time
+
+Install both packages in editable mode so changes in either package are picked up immediately without reinstalling:
+
+```bash
+# From the repo root
+pip install -e ./mito-ai-core
+pip install -e "./mito-ai[test]"
+```
+
+**Order matters!** Install `mito-ai-core` first since `mito-ai` depends on it. After this, any change you make to a `.py` file in either package takes effect on the next import (no rebuild needed).
+
+### How it relates to mito-ai
+
+`mito-ai-core` contains all the pure-Python AI logic -- providers, models, prompts, message history, utils.
+
+`mito-ai` is the JupyterLab extension that adds WebSocket/REST handlers, UI integration, and Streamlit conversion on top of it.
+
+If you're changing provider logic, prompt templates, or models, you're working here. If you're changing how the extension talks to the frontend or handles HTTP requests, you're working in `mito-ai`.
+
+## Usage
+
+Everything goes through `ProviderManager`:
+
+```python
+from mito_ai_core.provider_manager import ProviderManager
+from mito_ai_core.completions.models import MessageType
+
+pm = ProviderManager()
+pm.set_selected_model("gpt-4.1")
+
+# One-shot
+response = await pm.request_completions(
+    message_type=MessageType.CHAT,
+    messages=[{"role": "user", "content": "Hello!"}],
+)
+
+# Streaming -- pass a callback
+response = await pm.stream_completions(
+    message_type=MessageType.CHAT,
+    messages=[{"role": "user", "content": "Hello!"}],
+    message_id="msg-1",
+    thread_id="thread-1",
+    reply_fn=lambda chunk: print(chunk),
+)
+```
+
+`ProviderManager` picks the right client (OpenAI, Anthropic, Gemini, Copilot, LiteLLM, Abacus) based on the selected model. It handles retries, telemetry, and token logging.
+
+This is a library, not a server. The caller owns the transport (WebSocket, HTTP, CLI, whatever).
+
+## Key modules
+
+- **`provider_manager`** -- routes requests to the right LLM client
+- **`clients/`** -- OpenAI, Anthropic, Gemini, Copilot wrappers
+- **`completions/models`** -- all the dataclasses and Pydantic models
+- **`completions/message_history`** -- thread-safe chat persistence
+- **`completions/prompt_builders/`** -- prompt construction for chat, agent, debug, inline completion, charts
+- **`utils/`** -- model resolution, token estimation, telemetry, rate limits
+

mito_ai_core-0.1.1/mito_ai_core/__init__.py

@@ -0,0 +1,11 @@
+# Copyright (c) Saga Inc.
+# Distributed under the terms of the GNU Affero General Public License v3.0 License.
+
+"""
+mito-ai-core: Pure-Python AI backbone for Mito.
+
+Providers, Pydantic models, prompt builders, message history, and utilities
+with **zero** Jupyter Server / Tornado dependencies.
+"""
+
+from mito_ai_core._version import __version__  # noqa: F401

mito_ai_core-0.1.1/mito_ai_core/_version.py

@@ -0,0 +1,7 @@
+# Copyright (c) Saga Inc.
+# Distributed under the terms of the GNU Affero General Public License v3.0 License.
+
+# This file is auto-generated by Hatchling. As such, do not:
+#   - modify
+#   - track in version control e.g. be sure to add to .gitignore
+__version__ = VERSION = '0.1.228'

mito_ai_core-0.1.1/mito_ai_core/agent/__init__.py

@@ -0,0 +1,25 @@
+# Copyright (c) Saga Inc.
+# Distributed under the terms of the GNU Affero General Public License v3.0 License.
+
+"""Agent layer public types and protocols.
+
+Import :class:`~mito_ai_core.agent.agent_runner.AgentRunner` from
+``mito_ai_core.agent.agent_runner`` so this package's ``__init__`` does not
+eagerly load the runner (which pulls in prompt builders and would create import
+cycles with ``mito_ai_core.agent.types``).
+"""
+
+from __future__ import annotations
+
+from mito_ai_core.agent.types import AgentContext, AgentRunResult, CompletionProvider, ToolResult
+from mito_ai_core.agent.tool_executor import ToolExecutor
+from mito_ai_core.completions.models import KernelVariable
+
+__all__ = [
+    "AgentContext",
+    "KernelVariable",
+    "ToolResult",
+    "ToolExecutor",
+    "AgentRunResult",
+    "CompletionProvider",
+]