zeno-cli 0.3.4__py3-none-any.whl

zeno_sdk/_migrations/alembic/env.py

@@ -0,0 +1,33 @@
+from __future__ import annotations
+
+from alembic import context
+from sqlalchemy import engine_from_config, pool
+
+config = context.config
+
+
+def run_migrations_offline() -> None:
+    url = config.get_main_option("sqlalchemy.url")
+    context.configure(url=url, literal_binds=True, compare_type=True)
+
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+def run_migrations_online() -> None:
+    connectable = engine_from_config(
+        config.get_section(config.config_ini_section) or {},
+        prefix="sqlalchemy.",
+        poolclass=pool.NullPool,
+    )
+
+    with connectable.connect() as connection:
+        context.configure(connection=connection, compare_type=True)
+        with context.begin_transaction():
+            context.run_migrations()
+
+
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()

zeno_sdk/_migrations/alembic/script.py.mako

@@ -0,0 +1,18 @@
+"""${message}
+
+Revision ID: ${up_revision}
+Revises: ${down_revision | comma,n}
+Create Date: ${create_date}
+"""
+from alembic import op
+import sqlalchemy as sa
+
+${imports if imports else ""}
+
+
+def upgrade() -> None:
+    ${upgrades if upgrades else "pass"}
+
+
+def downgrade() -> None:
+    ${downgrades if downgrades else "pass"}

zeno_sdk/_migrations/alembic/versions/0001_initial.py

@@ -0,0 +1,79 @@
+"""initial schema
+
+Revision ID: 0001_initial
+Revises:
+Create Date: 2026-04-30
+"""
+
+from __future__ import annotations
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "0001_initial"
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "sessions",
+        sa.Column("id", sa.String(), primary_key=True),
+        sa.Column("start_at", sa.String(), nullable=False),
+        sa.Column("end_at", sa.String(), nullable=True),
+        sa.Column("agent_count_max", sa.Integer(), nullable=False),
+        sa.Column("harness", sa.String(), nullable=False),
+        sa.Column("project_id", sa.String(), nullable=False),
+        sa.Column("synced_at", sa.String(), nullable=True),
+    )
+    op.create_table(
+        "agent_runs",
+        sa.Column("id", sa.String(), primary_key=True),
+        sa.Column("session_id", sa.String(), nullable=False, index=True),
+        sa.Column("harness", sa.String(), nullable=False),
+        sa.Column("model", sa.String(), nullable=False),
+        sa.Column("started_at", sa.String(), nullable=False),
+        sa.Column("ended_at", sa.String(), nullable=True),
+        sa.Column("outcome", sa.String(), nullable=False),
+        sa.Column("synced_at", sa.String(), nullable=True),
+    )
+    op.create_table(
+        "supervision_events",
+        sa.Column("id", sa.String(), primary_key=True),
+        sa.Column("session_id", sa.String(), nullable=False, index=True),
+        sa.Column("agent_run_id", sa.String(), nullable=True, index=True),
+        sa.Column("type", sa.String(), nullable=False),
+        sa.Column("timestamp", sa.String(), nullable=False),
+        sa.Column("latency_ms_to_decide", sa.Integer(), nullable=True),
+        sa.Column("metadata_json", sa.Text(), nullable=False, server_default="{}"),
+        sa.Column("synced_at", sa.String(), nullable=True),
+    )
+    op.create_table(
+        "load_probes",
+        sa.Column("id", sa.String(), primary_key=True),
+        sa.Column("session_id", sa.String(), nullable=False, index=True),
+        sa.Column("prompted_at", sa.String(), nullable=False),
+        sa.Column("responded_at", sa.String(), nullable=True),
+        sa.Column("skipped", sa.Integer(), nullable=False),
+        sa.Column("subscales_json", sa.Text(), nullable=True),
+        sa.Column("synced_at", sa.String(), nullable=True),
+    )
+    op.create_table(
+        "babysitting_tax_points",
+        sa.Column("id", sa.String(), primary_key=True),
+        sa.Column("session_id", sa.String(), nullable=False, index=True),
+        sa.Column("n_agents_active", sa.Integer(), nullable=False),
+        sa.Column("composite_load", sa.Float(), nullable=False),
+        sa.Column("output_quality", sa.Float(), nullable=True),
+        sa.Column("synced_at", sa.String(), nullable=True),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("babysitting_tax_points")
+    op.drop_table("load_probes")
+    op.drop_table("supervision_events")
+    op.drop_table("agent_runs")
+    op.drop_table("sessions")

zeno_sdk/_migrations/alembic/versions/0002_cognition_samples.py

@@ -0,0 +1,53 @@
+"""cognition_samples: per-turn cognition stream (tokens + context + attention)
+
+Revision ID: 0002_cognition_samples
+Revises: 0001_initial
+Create Date: 2026-06-16
+
+One row per assistant turn, written by the HUD statusline (the read+write
+surface) so the in-terminal HUD and the tailnet dashboard share the exact
+same attention + token signal. See docs/COGNITION_UNIFIED_PLAN.md.
+
+Kept deliberately schema-compatible with the self-create block in
+integrations/zeno-hud/zeno_hud.py and dotfiles/claude/hooks/zeno-cc-bridge.py:
+if you add a column here, mirror it in those stdlib writers (they never
+import this package, to keep hook/statusline startup sub-100ms).
+"""
+
+from __future__ import annotations
+
+import sqlalchemy as sa
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision = "0002_cognition_samples"
+down_revision = "0001_initial"
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    op.create_table(
+        "cognition_samples",
+        sa.Column("id", sa.String(), primary_key=True),
+        sa.Column("session_id", sa.String(), nullable=False, index=True),
+        sa.Column("ts", sa.String(), nullable=False),
+        # tokens + context captured from the statusline stdin
+        sa.Column("context_pct", sa.Float(), nullable=True),
+        sa.Column("input_tokens", sa.Integer(), nullable=True),
+        sa.Column("output_tokens", sa.Integer(), nullable=True),
+        sa.Column("cache_read_tokens", sa.Integer(), nullable=True),
+        sa.Column("cache_creation_tokens", sa.Integer(), nullable=True),
+        sa.Column("total_tokens", sa.Integer(), nullable=True),
+        # attention signal (score 0..100 + its components) from the shared model
+        sa.Column("attention_score", sa.Integer(), nullable=True),
+        sa.Column("attention_effort", sa.Float(), nullable=True),
+        sa.Column("attention_deliberation", sa.Float(), nullable=True),
+        sa.Column("attention_trend", sa.Float(), nullable=True),
+        sa.Column("model", sa.String(), nullable=True),
+        sa.Column("harness", sa.String(), nullable=True),
+    )
+
+
+def downgrade() -> None:
+    op.drop_table("cognition_samples")

zeno_sdk/_migrations/alembic/versions/0003_cognition_drivers.py

@@ -0,0 +1,41 @@
+"""cognition_samples v2: multi-dimensional driver columns
+
+Revision ID: 0003_cognition_drivers
+Revises: 0002_cognition_samples
+Create Date: 2026-06-16
+
+HUD v2 turns the single attention score into a composite of five named drivers.
+This adds the four new driver columns (attention_effort already exists from 0002).
+
+Kept in lockstep with the canonical schema in
+integrations/zeno-hud/zeno_cognition.py (SAMPLE_COLUMNS) and the stdlib writers
+(zeno_hud.py + dotfiles zeno-cc-bridge.py call zeno_cognition.ensure_schema, which
+ADDs these columns on existing DBs). The schema-drift test guards the two lists.
+"""
+
+from __future__ import annotations
+
+import sqlalchemy as sa
+from alembic import op
+
+revision = "0003_cognition_drivers"
+down_revision = "0002_cognition_samples"
+branch_labels = None
+depends_on = None
+
+_DRIVER_COLUMNS = (
+    "attention_autonomy",
+    "attention_verification",
+    "attention_fatigue",
+    "attention_flow",
+)
+
+
+def upgrade() -> None:
+    for col in _DRIVER_COLUMNS:
+        op.add_column("cognition_samples", sa.Column(col, sa.Float(), nullable=True))
+
+
+def downgrade() -> None:
+    for col in reversed(_DRIVER_COLUMNS):
+        op.drop_column("cognition_samples", col)

zeno_sdk/_migrations/alembic/versions/0004_transcript_intelligence.py

@@ -0,0 +1,248 @@
+"""transcript intelligence: multi-tool session capture + cache-aware token/cost ledger
+
+Revision ID: 0004_transcript_intelligence
+Revises: 0003_cognition_drivers
+Create Date: 2026-06-17
+
+Additive session-intelligence tables, adopted from agentsview (MIT, (c) 2026 Kenn
+Software LLC; see THIRD_PARTY_LICENSES.md). These are a derived cache of coding-agent
+transcripts (Claude Code, Codex, Cursor) scoring *agent* behavior - token/cost, tool
+health, outcomes - ALONGSIDE zeno's primary, untouched human-cognition signal in
+``cognition_samples``. agentsview's ``sessions``/``messages`` collide with zeno's
+existing tables, so they are namespaced ``transcript_*``.
+
+Kept in lockstep with the canonical stdlib schema in
+``packages/session-intel/src/zeno_session_intel/schema.py`` (the ingester writes via
+stdlib sqlite3 and calls ``ensure_schema``; this migration is the SDK-managed-DB path).
+The schema-drift test (``tests/test_session_intel_schema.py``) guards the two lists.
+
+FTS5 is created via raw SQL (alembic does not model virtual tables) and is guarded:
+on a SQLite build without the fts5 module it is skipped, search degrades, ingest still
+works. ``transcript_messages.id`` is a synthetic INTEGER rowid - it is the FTS5
+``content_rowid``; do not make it a composite PK.
+"""
+
+from __future__ import annotations
+
+import sqlalchemy as sa
+from alembic import op
+
+revision = "0004_transcript_intelligence"
+down_revision = "0003_cognition_drivers"
+branch_labels = None
+depends_on = None
+
+_NOW = sa.text("(strftime('%Y-%m-%dT%H:%M:%fZ','now'))")
+
+_FTS_SQL = """
+CREATE VIRTUAL TABLE IF NOT EXISTS transcript_messages_fts USING fts5(
+    content,
+    content='transcript_messages',
+    content_rowid='id',
+    tokenize='porter unicode61'
+);
+CREATE TRIGGER IF NOT EXISTS transcript_messages_ai
+AFTER INSERT ON transcript_messages BEGIN
+    INSERT INTO transcript_messages_fts(rowid, content) VALUES (new.id, new.content);
+END;
+CREATE TRIGGER IF NOT EXISTS transcript_messages_ad
+AFTER DELETE ON transcript_messages BEGIN
+    INSERT INTO transcript_messages_fts(transcript_messages_fts, rowid, content)
+        VALUES ('delete', old.id, old.content);
+END;
+CREATE TRIGGER IF NOT EXISTS transcript_messages_au
+AFTER UPDATE ON transcript_messages BEGIN
+    INSERT INTO transcript_messages_fts(transcript_messages_fts, rowid, content)
+        VALUES ('delete', old.id, old.content);
+    INSERT INTO transcript_messages_fts(rowid, content) VALUES (new.id, new.content);
+END;
+"""
+
+
+def upgrade() -> None:
+    op.create_table(
+        "transcript_sessions",
+        sa.Column("id", sa.String(), primary_key=True),
+        sa.Column("zeno_session_id", sa.String(), nullable=True),
+        sa.Column("agent", sa.String(), nullable=False, server_default="claude"),
+        sa.Column("project", sa.String(), nullable=False, server_default=""),
+        sa.Column("machine", sa.String(), nullable=False, server_default="local"),
+        sa.Column("first_message", sa.Text(), nullable=True),
+        sa.Column("display_name", sa.String(), nullable=True),
+        sa.Column("started_at", sa.String(), nullable=True),
+        sa.Column("ended_at", sa.String(), nullable=True),
+        sa.Column("message_count", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("user_message_count", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("total_output_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("peak_context_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("is_automated", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("tool_failure_signal_count", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("tool_retry_count", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("edit_churn_count", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("consecutive_failure_max", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("outcome", sa.String(), nullable=False, server_default="unknown"),
+        sa.Column("outcome_confidence", sa.String(), nullable=False, server_default="low"),
+        sa.Column("ended_with_role", sa.String(), nullable=False, server_default=""),
+        sa.Column("final_failure_streak", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("compaction_count", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("mid_task_compaction_count", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("context_pressure_max", sa.Float(), nullable=True),
+        sa.Column("health_score", sa.Integer(), nullable=True),
+        sa.Column("health_grade", sa.String(), nullable=True),
+        sa.Column("has_tool_calls", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("has_context_data", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("cwd", sa.String(), nullable=False, server_default=""),
+        sa.Column("git_branch", sa.String(), nullable=False, server_default=""),
+        sa.Column("file_path", sa.String(), nullable=True),
+        sa.Column("file_mtime", sa.Integer(), nullable=True),
+        sa.Column("content_hash", sa.String(), nullable=True),
+        sa.Column("created_at", sa.String(), nullable=False, server_default=_NOW),
+    )
+    op.create_index(
+        "idx_transcript_sessions_ended",
+        "transcript_sessions",
+        [sa.text("ended_at DESC"), "id"],
+    )
+    op.create_index("idx_transcript_sessions_agent", "transcript_sessions", ["agent"])
+    op.create_index("idx_transcript_sessions_project", "transcript_sessions", ["project"])
+    op.create_index("idx_transcript_sessions_zeno", "transcript_sessions", ["zeno_session_id"])
+
+    op.create_table(
+        "transcript_messages",
+        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
+        sa.Column(
+            "session_id",
+            sa.String(),
+            sa.ForeignKey("transcript_sessions.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column("ordinal", sa.Integer(), nullable=False),
+        sa.Column("role", sa.String(), nullable=False),
+        sa.Column("content", sa.Text(), nullable=False),
+        sa.Column("thinking_text", sa.Text(), nullable=False, server_default=""),
+        sa.Column("timestamp", sa.String(), nullable=True),
+        sa.Column("has_thinking", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("has_tool_use", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("content_length", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("model", sa.String(), nullable=False, server_default=""),
+        sa.Column("token_usage", sa.Text(), nullable=False, server_default=""),
+        sa.Column("context_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("output_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("source_uuid", sa.String(), nullable=False, server_default=""),
+        sa.Column("source_parent_uuid", sa.String(), nullable=False, server_default=""),
+        sa.Column("is_sidechain", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("is_compact_boundary", sa.Integer(), nullable=False, server_default="0"),
+    )
+    op.create_index(
+        "idx_transcript_messages_session_ordinal",
+        "transcript_messages",
+        ["session_id", "ordinal"],
+        unique=True,
+    )
+    op.create_index(
+        "idx_transcript_messages_session_role",
+        "transcript_messages",
+        ["session_id", "role"],
+    )
+
+    op.create_table(
+        "transcript_tool_calls",
+        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
+        sa.Column(
+            "session_id",
+            sa.String(),
+            sa.ForeignKey("transcript_sessions.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column("message_ordinal", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("ordinal_in_message", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("tool_name", sa.String(), nullable=False, server_default=""),
+        sa.Column("category", sa.String(), nullable=False, server_default="Other"),
+        sa.Column("is_error", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("source", sa.String(), nullable=False, server_default=""),
+    )
+    op.create_index("idx_transcript_tool_calls_session", "transcript_tool_calls", ["session_id"])
+    op.create_index("idx_transcript_tool_calls_category", "transcript_tool_calls", ["category"])
+
+    op.create_table(
+        "token_usage_events",
+        sa.Column("id", sa.Integer(), primary_key=True, autoincrement=True),
+        sa.Column(
+            "session_id",
+            sa.String(),
+            sa.ForeignKey("transcript_sessions.id", ondelete="CASCADE"),
+            nullable=False,
+        ),
+        sa.Column("message_ordinal", sa.Integer(), nullable=True),
+        sa.Column("source", sa.String(), nullable=False),
+        sa.Column("model", sa.String(), nullable=False),
+        sa.Column("input_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("output_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("cache_creation_input_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("cache_read_input_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("reasoning_tokens", sa.Integer(), nullable=False, server_default="0"),
+        sa.Column("cost_usd", sa.Float(), nullable=True),
+        sa.Column("cost_status", sa.String(), nullable=False, server_default=""),
+        sa.Column("cost_source", sa.String(), nullable=False, server_default=""),
+        sa.Column("occurred_at", sa.String(), nullable=True),
+        sa.Column("dedup_key", sa.String(), nullable=False, server_default=""),
+    )
+    op.create_index(
+        "idx_token_usage_events_dedup",
+        "token_usage_events",
+        ["session_id", "source", "dedup_key"],
+        unique=True,
+        # the partial predicate must survive on BOTH dialects, else Postgres builds a
+        # FULL unique index and id-less (empty dedup_key) session events collide there.
+        sqlite_where=sa.text("dedup_key != ''"),
+        postgresql_where=sa.text("dedup_key != ''"),
+    )
+    op.create_index("idx_token_usage_events_session", "token_usage_events", ["session_id"])
+    op.create_index("idx_token_usage_events_occurred", "token_usage_events", ["occurred_at"])
+
+    op.create_table(
+        "model_pricing",
+        sa.Column("model_pattern", sa.String(), primary_key=True),
+        sa.Column("input_per_mtok", sa.Float(), nullable=False, server_default="0"),
+        sa.Column("output_per_mtok", sa.Float(), nullable=False, server_default="0"),
+        sa.Column("cache_creation_per_mtok", sa.Float(), nullable=False, server_default="0"),
+        sa.Column("cache_read_per_mtok", sa.Float(), nullable=False, server_default="0"),
+        sa.Column("context_window", sa.Integer(), nullable=True),
+        sa.Column("updated_at", sa.String(), nullable=False, server_default=_NOW),
+    )
+
+    # FTS5 virtual table + sync triggers. SQLite-only (Postgres has no fts5; the
+    # local-first ingester never targets prod Postgres) and guarded for SQLite builds
+    # compiled without the fts5 module. Uses the raw driver connection's executescript
+    # because the trigger bodies contain semicolons that a naive split would break.
+    bind = op.get_bind()
+    if bind.dialect.name == "sqlite":
+        raw = bind.connection.driver_connection  # the underlying sqlite3.Connection
+        try:
+            raw.execute("CREATE VIRTUAL TABLE temp._zeno_fts5_probe USING fts5(x)")
+            raw.execute("DROP TABLE temp._zeno_fts5_probe")
+            raw.executescript(_FTS_SQL)
+        except Exception:
+            # fts5 module unavailable in this SQLite build; search degrades, ingest works.
+            pass
+
+
+def downgrade() -> None:
+    bind = op.get_bind()
+    if bind.dialect.name == "sqlite":
+        raw = bind.connection.driver_connection
+        for stmt in (
+            "DROP TRIGGER IF EXISTS transcript_messages_au",
+            "DROP TRIGGER IF EXISTS transcript_messages_ad",
+            "DROP TRIGGER IF EXISTS transcript_messages_ai",
+            "DROP TABLE IF EXISTS transcript_messages_fts",
+        ):
+            try:
+                raw.execute(stmt)
+            except Exception:
+                pass
+    op.drop_table("model_pricing")
+    op.drop_table("token_usage_events")
+    op.drop_table("transcript_tool_calls")
+    op.drop_table("transcript_messages")
+    op.drop_table("transcript_sessions")

zeno_sdk/_migrations/alembic.ini

@@ -0,0 +1,35 @@
+[alembic]
+script_location = alembic
+sqlalchemy.url = sqlite:///placeholder.db
+
+[loggers]
+keys = root,sqlalchemy,alembic
+
+[handlers]
+keys = console
+
+[formatters]
+keys = generic
+
+[logger_root]
+level = WARN
+handlers = console
+
+[logger_sqlalchemy]
+level = WARN
+handlers =
+qualname = sqlalchemy.engine
+
+[logger_alembic]
+level = INFO
+handlers =
+qualname = alembic
+
+[handler_console]
+class = StreamHandler
+args = (sys.stderr,)
+level = NOTSET
+formatter = generic
+
+[formatter_generic]
+format = %(levelname)-5.5s [%(name)s] %(message)s

zeno_sdk/_runtime.py

@@ -0,0 +1,12 @@
+from __future__ import annotations
+
+import asyncio
+from collections.abc import Coroutine
+
+
+def run_coro[T](coro: Coroutine[object, object, T]) -> T:
+    try:
+        asyncio.get_running_loop()
+    except RuntimeError:
+        return asyncio.run(coro)
+    raise RuntimeError("Zeno sync API called inside running event loop; use async entrypoints.")

zeno_sdk/adapters/__init__.py

@@ -0,0 +1,15 @@
+from .anthropic import instrument as instrument_anthropic
+from .claude_code import instrument as instrument_claude_code
+from .crewai import instrument_agent as instrument_crewai_agent
+from .crewai import instrument_crew as instrument_crewai_crew
+from .langgraph import instrument as instrument_langgraph
+from .openai import instrument as instrument_openai
+
+__all__ = [
+    "instrument_anthropic",
+    "instrument_claude_code",
+    "instrument_crewai_agent",
+    "instrument_crewai_crew",
+    "instrument_langgraph",
+    "instrument_openai",
+]

zeno_sdk/adapters/anthropic.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from zeno_adapters.anthropic import instrument
+
+__all__ = ["instrument"]

zeno_sdk/adapters/claude_code.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from zeno_adapters.claude_code import instrument
+
+__all__ = ["instrument"]

zeno_sdk/adapters/crewai.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from zeno_adapters.crewai import instrument_agent, instrument_crew
+
+__all__ = ["instrument_agent", "instrument_crew"]

zeno_sdk/adapters/langgraph.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from zeno_adapters.langgraph import instrument
+
+__all__ = ["instrument"]

zeno_sdk/adapters/openai.py

@@ -0,0 +1,5 @@
+from __future__ import annotations
+
+from zeno_adapters.openai import instrument
+
+__all__ = ["instrument"]

zeno_sdk/auth.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+import keyring
+from keyring.errors import KeyringError
+
+SERVICE = "zeno-sdk"
+TOKEN_KEY = "clerk_jwt"
+
+
+def get_token() -> str | None:
+    try:
+        return keyring.get_password(SERVICE, TOKEN_KEY)
+    except KeyringError:
+        return None
+
+
+def set_token(token: str) -> None:
+    keyring.set_password(SERVICE, TOKEN_KEY, token)
+
+
+def clear_token() -> None:
+    try:
+        keyring.delete_password(SERVICE, TOKEN_KEY)
+    except KeyringError:
+        return