livekit-plugins-typesafe 1.8.8__py3-none-any.whl

livekit/plugins/typesafe/__init__.py

@@ -0,0 +1,45 @@
+# Copyright 2026 LiveKit, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""TypeSafe plugin for LiveKit Agents
+
+Jev decision models.
+"""
+
+from .decision_model import DecisionModel
+from .models import DecisionModels, OpenRouterDecisionModels
+from .version import __version__
+
+__all__ = ["DecisionModel", "DecisionModels", "OpenRouterDecisionModels", "__version__"]
+
+from livekit.agents import Plugin
+
+from .log import logger
+
+
+class TypeSafePlugin(Plugin):
+    def __init__(self) -> None:
+        super().__init__(__name__, __version__, __package__, logger)
+
+
+Plugin.register_plugin(TypeSafePlugin())
+
+# Cleanup docs of unexported modules
+_module = dir()
+NOT_IN_ALL = [m for m in _module if m not in __all__]
+
+__pdoc__ = {}
+
+for n in NOT_IN_ALL:
+    __pdoc__[n] = False

livekit/plugins/typesafe/decision_model.py

@@ -0,0 +1,210 @@
+from __future__ import annotations
+
+import asyncio
+import os
+from collections.abc import Mapping
+from typing import Annotated, Any, Literal
+
+import aiohttp
+from pydantic import BaseModel, Field, TypeAdapter, ValidationError
+
+from livekit.agents import (
+    APIConnectionError,
+    APIError,
+    APIStatusError,
+    APITimeoutError,
+    llm,
+    utils,
+)
+from livekit.agents.llm.decisions import (
+    Choice,
+    ChoiceResult,
+    Decision,
+    DecisionCapabilities,
+    DecisionResponse,
+    DecisionResult,
+    Probability,
+    ProbabilityResult,
+    ScoreResult,
+)
+from livekit.agents.types import APIConnectOptions
+
+from .models import DecisionModels, OpenRouterDecisionModels
+
+TYPESAFE_ENDPOINT = "https://api.typesafe.ai/v1/systemone"
+OPENROUTER_ENDPOINT = "https://openrouter.ai/api/alpha/decisions"
+
+
+class _NoulAnswer(BaseModel):
+    type: Literal["noul"]
+    noul: float
+
+
+class _ChoiceAnswer(BaseModel):
+    type: Literal["choice"]
+    choice: str
+    probabilities: dict[str, float]
+    confidence: float | None = None
+
+
+class _ScoreAnswer(BaseModel):
+    type: Literal["score"]
+    score: float
+    probabilities: dict[int, float]
+    legend: dict[int, str]
+    confidence: float | None = None
+
+
+class _Usage(BaseModel):
+    input_tokens: int | None = None
+    output_tokens: int | None = None
+
+
+_Answer = Annotated[_NoulAnswer | _ChoiceAnswer | _ScoreAnswer, Field(discriminator="type")]
+_answer_adapter = TypeAdapter[_Answer](_Answer)
+
+
+class _Response(BaseModel):
+    answers: dict[str, Any]
+    model: str = ""
+    usage: _Usage = Field(default_factory=_Usage)
+    id: str = ""
+
+
+class DecisionModel(llm.DecisionModel):
+    """TypeSafe Jev, via the TypeSafe API or OpenRouter (:meth:`with_openrouter`).
+
+    Args:
+        model: TypeSafe model identifier.
+        api_key: Defaults to TYPESAFE_API_KEY.
+        endpoint: Complete HTTP endpoint for batched decisions.
+        http_session: Optional caller-owned session. Otherwise uses the job's HTTP
+            session. Outside a job, use ``async with utils.http_context.open()``.
+    """
+
+    def __init__(
+        self,
+        *,
+        model: DecisionModels | str = "jev-1.13.0",
+        api_key: str | None = None,
+        endpoint: str = TYPESAFE_ENDPOINT,
+        http_session: aiohttp.ClientSession | None = None,
+    ) -> None:
+        super().__init__(
+            capabilities=DecisionCapabilities(decision_kinds=["probability", "choice", "score"])
+        )
+        key = api_key or os.environ.get("TYPESAFE_API_KEY")
+        if not key:
+            raise ValueError("TypeSafe requires api_key or TYPESAFE_API_KEY")
+        self._api_key = key
+        self._model = model
+        self._endpoint = endpoint
+        self._session = http_session
+        self._provider = "typesafe"
+
+    @classmethod
+    def with_openrouter(
+        cls,
+        *,
+        model: OpenRouterDecisionModels | str = "typesafe/jev-1.13",
+        api_key: str | None = None,
+        http_session: aiohttp.ClientSession | None = None,
+    ) -> DecisionModel:
+        """Use OpenRouter's Decisions API. ``api_key`` defaults to OPENROUTER_API_KEY."""
+        key = api_key or os.environ.get("OPENROUTER_API_KEY")
+        if not key:
+            raise ValueError("TypeSafe via OpenRouter requires api_key or OPENROUTER_API_KEY")
+        instance = cls(
+            model=model, api_key=key, endpoint=OPENROUTER_ENDPOINT, http_session=http_session
+        )
+        instance._provider = "openrouter"
+        return instance
+
+    @property
+    def model(self) -> str:
+        return self._model
+
+    @property
+    def provider(self) -> str:
+        return self._provider
+
+    async def _evaluate_impl(
+        self,
+        *,
+        input: str,
+        decisions: Mapping[str, Decision],
+        conn_options: APIConnectOptions,
+    ) -> DecisionResponse:
+        questions = {name: _encode_question(definition) for name, definition in decisions.items()}
+        session = self._session or utils.http_context.http_session()
+        try:
+            async with session.post(
+                self._endpoint,
+                headers={"Authorization": f"Bearer {self._api_key}"},
+                json={"model": self._model, "state": input, "questions": questions},
+                timeout=aiohttp.ClientTimeout(total=conn_options.timeout),
+            ) as response:
+                if response.status >= 400:
+                    raise APIStatusError(
+                        "TypeSafe decision request failed",
+                        status_code=response.status,
+                        request_id=response.headers.get("x-request-id"),
+                    )
+                parsed = _Response.model_validate(await response.json())
+                request_id = parsed.id or response.headers.get("x-request-id", "")
+            results: dict[str, DecisionResult] = {}
+            errors: dict[str, str] = {}
+            for name, raw_answer in parsed.answers.items():
+                try:
+                    results[name] = _decode_answer(_answer_adapter.validate_python(raw_answer))
+                except ValidationError:
+                    errors[name] = "invalid Jev decision answer"
+                except APIError as error:
+                    errors[name] = error.message
+            return DecisionResponse(
+                results=results,
+                errors=errors,
+                model=parsed.model,
+                provider=self.provider,
+                request_id=request_id,
+                input_tokens=parsed.usage.input_tokens,
+                output_tokens=parsed.usage.output_tokens,
+            )
+        except asyncio.TimeoutError as exc:
+            raise APITimeoutError("TypeSafe request timed out") from exc
+        except aiohttp.ContentTypeError as exc:
+            raise APIError("invalid TypeSafe decision response", retryable=False) from exc
+        except aiohttp.ClientError as exc:
+            raise APIConnectionError("TypeSafe connection failed") from exc
+        except ValueError as exc:
+            raise APIError("invalid TypeSafe decision response", retryable=False) from exc
+
+
+def _encode_question(definition: Decision) -> dict[str, Any]:
+    question: dict[str, Any] = {"instructions": definition.prompt}
+    if isinstance(definition, Probability):
+        question["type"] = "noul"
+    elif isinstance(definition, Choice):
+        question.update(type="choice", criteria=definition.options)
+    else:
+        question.update(type="score", criteria=definition.levels)
+    return question
+
+
+def _decode_answer(answer: _Answer) -> DecisionResult:
+    if isinstance(answer, _NoulAnswer):
+        return ProbabilityResult(value=answer.noul)
+    if isinstance(answer, _ChoiceAnswer):
+        return ChoiceResult(
+            value=answer.choice,
+            probabilities=answer.probabilities,
+            provider_data={"confidence": answer.confidence},
+        )
+    if answer.legend.keys() != answer.probabilities.keys():
+        raise APIError("score legend does not match distribution", retryable=False)
+    return ScoreResult(
+        value=answer.score,
+        probabilities=answer.probabilities,
+        levels=[answer.legend[index] for index in sorted(answer.legend)],
+        provider_data={"confidence": answer.confidence},
+    )

livekit/plugins/typesafe/log.py

@@ -0,0 +1,3 @@
+import logging
+
+logger = logging.getLogger("livekit.plugins.typesafe")

livekit/plugins/typesafe/models.py

@@ -0,0 +1,5 @@
+from typing import Literal
+
+# jev-latest and jev-preview are TypeSafe aliases that move to newer releases.
+DecisionModels = Literal["jev-latest", "jev-preview", "jev-1.14.0", "jev-1.13.0"]
+OpenRouterDecisionModels = Literal["typesafe/jev-1.13"]

livekit/plugins/typesafe/version.py

@@ -0,0 +1,15 @@
+# Copyright 2026 LiveKit, Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+__version__ = "1.8.8"

livekit_plugins_typesafe-1.8.8.dist-info/METADATA

@@ -0,0 +1,83 @@
+Metadata-Version: 2.5
+Name: livekit-plugins-typesafe
+Version: 1.8.8
+Summary: Agent Framework plugin for TypeSafe Jev decision models.
+Project-URL: Documentation, https://docs.livekit.io
+Project-URL: Website, https://livekit.io/
+Project-URL: Source, https://github.com/livekit/agents
+Author-email: LiveKit <hello@livekit.io>
+License-Expression: Apache-2.0
+Keywords: audio,decisions,jev,livekit,realtime,typesafe,video,webrtc
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.10.0
+Requires-Dist: livekit-agents>=1.8.8
+Description-Content-Type: text/markdown
+
+# TypeSafe plugin for LiveKit Agents
+
+Support for [TypeSafe](https://typesafe.ai) Jev decision models. Jev evaluates probability
+questions, named choices, and ordered scores over a conversation in one request.
+
+## Installation
+
+```bash
+pip install livekit-plugins-typesafe
+```
+
+## Pre-requisites
+
+Set `TYPESAFE_API_KEY` for the TypeSafe API, or `OPENROUTER_API_KEY` to use Jev through
+OpenRouter's Decisions API.
+
+## Usage
+
+```python
+from livekit.agents import llm, utils
+from livekit.plugins import typesafe
+
+async with utils.http_context.open():
+    model = typesafe.DecisionModel(model="jev-latest")
+    # or: typesafe.DecisionModel.with_openrouter()
+    context = llm.ChatContext.empty()
+    context.add_message(role="user", content="Please let me speak to a person.")
+    response = await model.evaluate(
+        input=context,
+        decisions={
+            "handoff": llm.Probability("The caller requests a human."),
+            "intent": llm.Choice(
+                "What does the caller want?",
+                options={"booking": "Make a reservation", "other": "Anything else"},
+            ),
+            "frustration": llm.Score(
+                "How frustrated is the caller?", levels=["Calm", "Annoyed", "Angry"]
+            ),
+        },
+    )
+    for name, result in response.results.items():
+        print(name, result.value)
+```
+
+`input` also accepts a string, which is sent as-is. Each requested decision appears in
+either `response.results` or `response.errors`. Request failures and malformed responses
+raise.
+
+## Models
+
+`jev-latest` is the most recent stable release and `jev-preview` the most recent release
+overall; TypeSafe moves both aliases as new versions ship. Pin `jev-1.14.0` or
+`jev-1.13.0` for a fixed version.
+
+## Notes
+
+- Jev rounds probabilities and scores to two decimal places. Results keep the provider's
+  values.
+- Score answers must return a legend that matches the requested levels.
+- `provider_data["confidence"]` describes how concentrated the distribution is, not the
+  probability that the answer is correct.
+- A conversation is sent as a `role: text` transcript of user and assistant messages.
+  System and developer messages, agent instructions, and non-text content are never sent.

livekit_plugins_typesafe-1.8.8.dist-info/RECORD

@@ -0,0 +1,9 @@
+livekit/plugins/typesafe/__init__.py,sha256=Lq6jsP2P2t6HgPrA0MK78TLFnmjH1Shlp-BLhZut8vc,1274
+livekit/plugins/typesafe/decision_model.py,sha256=gpYbGYUovTxCrMmSnbTbkMgMIdqPqTOthffQwMBrCY4,7238
+livekit/plugins/typesafe/log.py,sha256=wDP2QbBz-5K1z86lbNu1uuY-dBbuK5MoLtyXk4sPzq0,71
+livekit/plugins/typesafe/models.py,sha256=f-4RjbHHVnTvM13v0SXxgg4p1Aa_Wv_KhNKLTrEm6KU,245
+livekit/plugins/typesafe/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+livekit/plugins/typesafe/version.py,sha256=DVN3ewC-8eWaCrgOnx9gWXZNQ9s80t7Q2iys4FwK_Mk,600
+livekit_plugins_typesafe-1.8.8.dist-info/METADATA,sha256=4sjJwe-5KiLQO5pfVgorqEfs46NzvTsSP9eFhKEj4uA,3095
+livekit_plugins_typesafe-1.8.8.dist-info/WHEEL,sha256=W3fkpkm7-wf9vBI5Z-7s0eWkeM-spu78I8Neb98DeEg,87
+livekit_plugins_typesafe-1.8.8.dist-info/RECORD,,

livekit_plugins_typesafe-1.8.8.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.32.4
+Root-Is-Purelib: true
+Tag: py3-none-any