polyllmkit 0.1.0__tar.gz → 0.1.2__tar.gz

{polyllmkit-0.1.0 → polyllmkit-0.1.2}/PKG-INFO

@@ -1,7 +1,11 @@
 Metadata-Version: 2.3
 Name: polyllmkit
-Version: 0.1.0
+Version: 0.1.2
 Summary: Streaming, multi-provider LLM bridge + markdown rendering toolkit for terminal apps.
+Requires-Dist: openai>=1.0 ; extra == 'all'
+Requires-Dist: claude-agent-sdk>=0.2 ; extra == 'all'
+Requires-Dist: anthropic>=0.40 ; extra == 'all'
+Requires-Dist: google-genai>=1.0 ; extra == 'all'
 Requires-Dist: anthropic>=0.40 ; extra == 'anthropic'
 Requires-Dist: openai>=1.0 ; extra == 'bridge'
 Requires-Dist: claude-agent-sdk>=0.2 ; extra == 'claude'
@@ -10,6 +14,7 @@ Requires-Dist: rich>=13.0 ; extra == 'md'
 Requires-Dist: textual>=0.50 ; extra == 'md'
 Requires-Dist: markdown-it-py>=3.0 ; extra == 'md'
 Requires-Python: >=3.11
+Provides-Extra: all
 Provides-Extra: anthropic
 Provides-Extra: bridge
 Provides-Extra: claude

polyllmkit-0.1.2/pyproject.toml

@@ -0,0 +1,37 @@
+[project]
+name = "polyllmkit"
+version = "0.1.2"
+description = "Streaming, multi-provider LLM bridge + markdown rendering toolkit for terminal apps."
+requires-python = ">=3.11"
+dependencies = []
+
+[project.optional-dependencies]
+bridge = ["openai>=1.0"]
+claude = ["claude-agent-sdk>=0.2"]
+anthropic = ["anthropic>=0.40"]
+google = ["google-genai>=1.0"]
+all = [
+    "openai>=1.0",
+    "claude-agent-sdk>=0.2",
+    "anthropic>=0.40",
+    "google-genai>=1.0",
+]
+md = [
+    "rich>=13.0",
+    "textual>=0.50",
+    "markdown-it-py>=3.0",
+]
+
+[build-system]
+requires = ["uv_build>=0.5,<0.12"]
+build-backend = "uv_build"
+
+[tool.uv.build-backend]
+module-root = "src"
+module-name = "llmkit"
+
+[dependency-groups]
+dev = [
+    "ruff>=0.6",
+    "mypy>=1.10",
+]

polyllmkit-0.1.0/pyproject.toml → polyllmkit-0.1.2/pyproject.toml.orig

@@ -4,7 +4,7 @@
 # module-name below): consumers keep `import llmkit`, `pip install polyllmkit`.
 [project]
 name = "polyllmkit"
-version = "0.1.0"
+version = "0.1.2"
 description = "Streaming, multi-provider LLM bridge + markdown rendering toolkit for terminal apps."
 requires-python = ">=3.11"  # tomllib (bridge.config) is stdlib only from 3.11
 # Base install pulls nothing — pick the subpackage(s) you need via extras.
@@ -21,6 +21,14 @@ bridge = ["openai>=1.0"]
 claude = ["claude-agent-sdk>=0.2"]  # claude_code adapter (Claude Agent SDK)
 anthropic = ["anthropic>=0.40"]  # anthropic adapter (native Messages API)
 google = ["google-genai>=1.0"]  # google adapter (native Gemini API)
+# Every backend at once — for hosts that resolve the adapter at runtime and
+# must have all of them present (e.g. cribsheet's daemon).
+all = [
+    "openai>=1.0",
+    "claude-agent-sdk>=0.2",
+    "anthropic>=0.40",
+    "google-genai>=1.0",
+]
 # llmkit.md — markdown rendering (one-shot + live stream + scrollable modal).
 # markdown-it-py is a direct import in md.render.stream (textual also pulls
 # it, but we depend on it explicitly since we use it ourselves).

{polyllmkit-0.1.0 → polyllmkit-0.1.2}/src/llmkit/bridge/adapters/anthropic.py

@@ -29,7 +29,6 @@ openai-compatible path never requires it.
 from __future__ import annotations
 
 import os
-import sys
 from typing import Any, Optional
 from urllib.parse import urlsplit, urlunsplit
 
@@ -57,17 +56,18 @@ def _base_url(endpoint: str) -> str:
 
 
 def _require_sdk() -> Any:
-    """Import the SDK or exit(2) with an actionable install hint."""
+    """Import the SDK, or raise ImportError with an actionable install hint.
+
+    Catchable exception, never ``SystemExit`` — see the note in
+    ``claude_code._require_sdk``."""
     try:
         import anthropic  # type: ignore[import-not-found]
-    except ImportError:
-        print(
+    except ImportError as e:
+        raise ImportError(
             "llmkit: the anthropic adapter needs the Anthropic SDK. "
             "Install it with `pip install llmkit[anthropic]` (or `pip install "
-            "anthropic`).",
-            file=sys.stderr,
-        )
-        raise SystemExit(2)
+            "anthropic`)."
+        ) from e
     return anthropic
 
 

{polyllmkit-0.1.0 → polyllmkit-0.1.2}/src/llmkit/bridge/adapters/claude_code.py

@@ -18,7 +18,6 @@ CLI governs those itself. Only ``model``, the request's ``system``, and
 from __future__ import annotations
 
 import os
-import sys
 from typing import Any
 
 from ..bridge import ChatRequest, _Emitter
@@ -26,17 +25,21 @@ from ..config import Provider
 
 
 def _require_sdk() -> Any:
-    """Import the SDK or exit(2) with an actionable install hint."""
+    """Import the SDK, or raise ImportError with an actionable install hint.
+
+    A missing optional dependency must surface as a *catchable* exception, never
+    ``SystemExit``: llmkit is a library, and hosts embed it inside long-lived
+    processes (e.g. an MCP daemon's lifespan startup) where a stray ``SystemExit``
+    — a ``BaseException`` — escapes ``except Exception`` and tears the whole
+    process down. The CLI translates this back into a clean exit(2)."""
     try:
         import claude_agent_sdk as sdk  # type: ignore[import-not-found]
-    except ImportError:
-        print(
+    except ImportError as e:
+        raise ImportError(
             "llmkit: the claude_code adapter needs the Claude Agent SDK. "
             "Install it with `pip install llmkit[claude]` (or `pip install "
-            "claude-agent-sdk`), and make sure the `claude` CLI is on PATH.",
-            file=sys.stderr,
-        )
-        raise SystemExit(2)
+            "claude-agent-sdk`), and make sure the `claude` CLI is on PATH."
+        ) from e
     return sdk
 
 
@@ -55,15 +58,15 @@ def _thinking_budget() -> int:
 
 def _build_options(sdk: Any, provider: Provider, request: ChatRequest) -> Any:
     """Assemble ClaudeAgentOptions for a tool-free single-turn chat."""
-    opts: dict[str, Any] = dict(
-        tools=[],  # no built-in tools — plain chat, no file/shell access
-        allowed_tools=[],
-        disallowed_tools=[],
-        max_turns=1,
-        permission_mode="bypassPermissions",
-        include_partial_messages=True,  # token-level streaming deltas
-        setting_sources=[],  # ignore project/user CLAUDE.md, settings, etc.
-    )
+    opts: dict[str, Any] = {
+        "tools": [],  # no built-in tools — plain chat, no file/shell access
+        "allowed_tools": [],
+        "disallowed_tools": [],
+        "max_turns": 1,
+        "permission_mode": "bypassPermissions",
+        "include_partial_messages": True,  # token-level streaming deltas
+        "setting_sources": [],  # ignore project/user CLAUDE.md, settings, etc.
+    }
     if provider.model:
         opts["model"] = provider.model
     if request.system:
@@ -84,13 +87,22 @@ def _build_options(sdk: Any, provider: Provider, request: ChatRequest) -> Any:
         opts["thinking"] = {"type": "disabled"}
         return sdk.ClaudeAgentOptions(**opts)
 
+    # ``display: "summarized"`` is required on current models: they default it
+    # to ``omitted``, which silences thinking entirely — no deltas stream and
+    # the final ThinkingBlock is empty (verified: enabled+budget without display
+    # streams 0 chars of thinking; with display, ~tokens arrive). Mirrors the
+    # anthropic adapter's same-named field.
     flag = provider.enable_thinking or "auto"
     if flag == "false":
         opts["thinking"] = {"type": "disabled"}
     elif flag == "true":
-        opts["thinking"] = {"type": "enabled", "budget_tokens": _thinking_budget()}
+        opts["thinking"] = {
+            "type": "enabled",
+            "budget_tokens": _thinking_budget(),
+            "display": "summarized",
+        }
     else:
-        opts["thinking"] = {"type": "adaptive"}
+        opts["thinking"] = {"type": "adaptive", "display": "summarized"}
 
     return sdk.ClaudeAgentOptions(**opts)
 
@@ -143,7 +155,7 @@ async def _run(
 
 def structured_claude_code(
     provider: Provider, request: ChatRequest
-) -> "tuple[int, Any]":
+) -> tuple[int, Any]:
     """Structured output via the SDK's native ``output_format`` (set in
     ``_build_options`` when the request carries a schema). Runs the single-turn
     query and returns ``(exit_code, structured_output)`` from the ResultMessage —

{polyllmkit-0.1.0 → polyllmkit-0.1.2}/src/llmkit/bridge/adapters/google.py

@@ -20,7 +20,6 @@ openai-compatible path never requires it.
 from __future__ import annotations
 
 import os
-import sys
 from typing import Any, Optional
 
 from ..bridge import DEFAULT_ENDPOINT, ChatRequest, _Emitter
@@ -33,14 +32,12 @@ def _require_sdk() -> Any:
     try:
         from google import genai  # type: ignore[import-untyped,import-not-found]
         from google.genai import types  # type: ignore[import-untyped,import-not-found]
-    except ImportError:
-        print(
+    except ImportError as e:
+        raise ImportError(
             "llmkit: the google adapter needs the Google GenAI SDK. "
             "Install it with `pip install llmkit[google]` (or `pip install "
-            "google-genai`).",
-            file=sys.stderr,
-        )
-        raise SystemExit(2)
+            "google-genai`)."
+        ) from e
     return genai, types
 
 

{polyllmkit-0.1.0 → polyllmkit-0.1.2}/src/llmkit/bridge/bridge.py

@@ -336,7 +336,7 @@ def complete(
 ) -> int:
     """Run a text-completion (FIM) request, streaming text to ``out``
     (default stdout). ``openai-compatible`` only."""
-    out = out if out is not None else sys.stdout
+    sink = out if out is not None else sys.stdout
     provider = with_defaults(provider)
     if not provider_supports_complete(provider):
         print(
@@ -349,12 +349,12 @@ def complete(
         return 2
 
     client = build_client(provider)
-    kwargs: dict[str, Any] = dict(
-        model=provider.model,
-        prompt=request.prompt,
-        max_tokens=provider.max_tokens,
-        temperature=provider.temperature,
-    )
+    kwargs: dict[str, Any] = {
+        "model": provider.model,
+        "prompt": request.prompt,
+        "max_tokens": provider.max_tokens,
+        "temperature": provider.temperature,
+    }
     if request.suffix:
         kwargs["suffix"] = request.suffix
     if request.stop:
@@ -364,8 +364,8 @@ def complete(
         for chunk in client.completions.create(**kwargs):
             t = chunk.choices[0].text if chunk.choices else ""
             if t:
-                out.write(t)
-                out.flush()
+                sink.write(t)
+                sink.flush()
     except KeyboardInterrupt:
         return 130
     except Exception as e:

{polyllmkit-0.1.0 → polyllmkit-0.1.2}/src/llmkit/bridge/cli.py

@@ -172,3 +172,9 @@ if __name__ == "__main__":
         sys.exit(main())
     except BrokenPipeError:
         sys.exit(0)
+    except ImportError as e:
+        # A missing optional-adapter dependency. The adapters raise a catchable
+        # ImportError (never SystemExit, so library hosts survive); the CLI turns
+        # it back into the actionable message + exit(2).
+        print(e, file=sys.stderr)
+        sys.exit(2)

{polyllmkit-0.1.0 → polyllmkit-0.1.2}/src/llmkit/bridge/client.py

@@ -10,7 +10,6 @@ from __future__ import annotations
 
 import os
 import re
-import sys
 import time
 import uuid
 from typing import Any, Callable, Dict, Optional
@@ -55,14 +54,12 @@ def build_client(provider: Provider) -> Any:
     hard import of ``openai.OpenAI`` at module load."""
     try:
         from openai import OpenAI
-    except ImportError:
-        print(
+    except ImportError as e:
+        raise ImportError(
             "llmkit: the openai package is not installed. "
             "Install it with `pip install llmkit[bridge]` (or `pip install "
-            "openai`).",
-            file=sys.stderr,
-        )
-        raise SystemExit(2)
+            "openai`)."
+        ) from e
     return OpenAI(
         base_url=provider.endpoint,
         api_key=_resolve_api_key(provider),