mirror of
https://github.com/usestrix/strix.git
synced 2026-08-22 11:02:08 +02:00
yoni
218470f14d
The Run details panel hardcoded "ChatGPT subscription" for any subscription run. Emit subscription_provider (ChatGPT/Grok) in the run record and render it in the viewer, so Grok runs read "Grok subscription". Rebuilds the committed viewer bundle.
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
"""Shared helpers across model-subscription providers (ChatGPT/Codex and Grok).
|
|
|
|
Each provider module (:mod:`strix.config.codex`, :mod:`strix.config.grok`)
|
|
exposes the same small surface — ``subscription_model``, ``auth_mode``,
|
|
``is_authenticated`` — so callers that only care "is this run on a subscription,
|
|
and which provider?" can stay provider-agnostic.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from strix.config import codex, grok
|
|
|
|
|
|
if TYPE_CHECKING:
|
|
from types import ModuleType
|
|
|
|
|
|
_PROVIDERS: tuple[ModuleType, ...] = (codex, grok)
|
|
|
|
# Human-facing provider names keyed by each module's ``PROVIDER`` constant.
|
|
_DISPLAY_NAMES: dict[str, str] = {codex.PROVIDER: "ChatGPT", grok.PROVIDER: "Grok"}
|
|
|
|
|
|
def provider_for_model(model_name: str | None) -> ModuleType | None:
|
|
"""Return the subscription provider module that owns ``model_name``'s prefix,
|
|
or None when the model isn't a subscription model."""
|
|
for provider in _PROVIDERS:
|
|
if provider.subscription_model(model_name):
|
|
return provider
|
|
return None
|
|
|
|
|
|
def auth_mode(model_name: str | None) -> str:
|
|
return "subscription" if provider_for_model(model_name) is not None else "api_key"
|
|
|
|
|
|
def provider_label(model_name: str | None) -> str | None:
|
|
"""Human-facing name of the subscription provider for ``model_name`` (e.g.
|
|
"ChatGPT" or "Grok"), or None when the model isn't a subscription model."""
|
|
provider = provider_for_model(model_name)
|
|
if provider is None:
|
|
return None
|
|
return _DISPLAY_NAMES.get(provider.PROVIDER)
|