openshell 0.0.113__py3-none-any.whl

openshell/__init__.py

@@ -0,0 +1,46 @@
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+"""OpenShell - Agent execution and management SDK."""
+
+from __future__ import annotations
+
+from .sandbox import (
+    ExecChunk,
+    ExecResult,
+    InferenceRouteClient,
+    InferenceRouteConfig,
+    Sandbox,
+    SandboxClient,
+    SandboxError,
+    SandboxRef,
+    SandboxSession,
+    SandboxStatusRef,
+    TlsConfig,
+    WorkspaceClient,
+    WorkspaceRef,
+)
+
+try:
+    from importlib.metadata import version
+
+    __version__ = version("openshell")
+except Exception:
+    __version__ = "0.0.0"
+
+__all__ = [
+    "ExecChunk",
+    "ExecResult",
+    "InferenceRouteClient",
+    "InferenceRouteConfig",
+    "Sandbox",
+    "SandboxClient",
+    "SandboxError",
+    "SandboxRef",
+    "SandboxSession",
+    "SandboxStatusRef",
+    "TlsConfig",
+    "WorkspaceClient",
+    "WorkspaceRef",
+    "__version__",
+]

openshell/_proto/__init__.py

@@ -0,0 +1,15 @@
+from . import datamodel_pb2, openshell_pb2
+
+# Sandbox messages and phase enums moved into openshell.proto. Keep aliases on
+# datamodel_pb2 so existing Python callers and E2E tests continue to work.
+for _name in (
+    "Sandbox",
+    "SandboxSpec",
+    "SandboxTemplate",
+):
+    if not hasattr(datamodel_pb2, _name):
+        setattr(datamodel_pb2, _name, getattr(openshell_pb2, _name))
+
+for _name in dir(openshell_pb2):
+    if _name.startswith("SANDBOX_PHASE_") and not hasattr(datamodel_pb2, _name):
+        setattr(datamodel_pb2, _name, getattr(openshell_pb2, _name))

openshell/_proto/datamodel_pb2.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: datamodel.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'datamodel.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from . import options_pb2 as options__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0f\x64\x61tamodel.proto\x12\x16openshell.datamodel.v1\x1a\roptions.proto\"\xf6\x02\n\nObjectMeta\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x15\n\rcreated_at_ms\x18\x03 \x01(\x03\x12>\n\x06labels\x18\x04 \x03(\x0b\x32..openshell.datamodel.v1.ObjectMeta.LabelsEntry\x12\x18\n\x10resource_version\x18\x05 \x01(\x04\x12H\n\x0b\x61nnotations\x18\x06 \x03(\x0b\x32\x33.openshell.datamodel.v1.ObjectMeta.AnnotationsEntry\x12\x11\n\tworkspace\x18\x07 \x01(\t\x12\x1d\n\x15\x64\x65letion_timestamp_ms\x18\x08 \x01(\x03\x1a-\n\x0bLabelsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a\x32\n\x10\x41nnotationsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"H\n\x0fWorkspaceStatus\x12\x35\n\x05phase\x18\x01 \x01(\x0e\x32&.openshell.datamodel.v1.WorkspacePhase\"z\n\tWorkspace\x12\x34\n\x08metadata\x18\x01 \x01(\x0b\x32\".openshell.datamodel.v1.ObjectMeta\x12\x37\n\x06status\x18\x02 \x01(\x0b\x32\'.openshell.datamodel.v1.WorkspaceStatus\"\xad\x01\n\x10\x43redentialHandle\x12\x0e\n\x06\x64river\x18\x01 \x01(\t\x12\x0e\n\x06handle\x18\x02 \x01(\t\x12H\n\x08metadata\x18\x03 \x03(\x0b\x32\x36.openshell.datamodel.v1.CredentialHandle.MetadataEntry\x1a/\n\rMetadataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xae\x05\n\x08Provider\x12\x34\n\x08metadata\x18\x01 \x01(\x0b\x32\".openshell.datamodel.v1.ObjectMeta\x12\x0c\n\x04type\x18\x02 \x01(\t\x12L\n\x0b\x63redentials\x18\x03 \x03(\x0b\x32\x31.openshell.datamodel.v1.Provider.CredentialsEntryB\x04\x88\xb5\x18\x01\x12<\n\x06\x63onfig\x18\x04 \x03(\x0b\x32,.openshell.datamodel.v1.Provider.ConfigEntry\x12]\n\x18\x63redential_expires_at_ms\x18\x05 \x03(\x0b\x32;.openshell.datamodel.v1.Provider.CredentialExpiresAtMsEntry\x12\x19\n\x11profile_workspace\x18\x06 \x01(\t\x12S\n\x12\x63redential_handles\x18\x07 \x03(\x0b\x32\x37.openshell.datamodel.v1.Provider.CredentialHandlesEntry\x1a\x32\n\x10\x43redentialsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a-\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a<\n\x1a\x43redentialExpiresAtMsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x03:\x02\x38\x01\x1a\x62\n\x16\x43redentialHandlesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\x37\n\x05value\x18\x02 \x01(\x0b\x32(.openshell.datamodel.v1.CredentialHandle:\x02\x38\x01*n\n\x0eWorkspacePhase\x12\x1f\n\x1bWORKSPACE_PHASE_UNSPECIFIED\x10\x00\x12\x1a\n\x16WORKSPACE_PHASE_ACTIVE\x10\x01\x12\x1f\n\x1bWORKSPACE_PHASE_TERMINATING\x10\x02\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'datamodel_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_OBJECTMETA_LABELSENTRY']._loaded_options = None
+  _globals['_OBJECTMETA_LABELSENTRY']._serialized_options = b'8\001'
+  _globals['_OBJECTMETA_ANNOTATIONSENTRY']._loaded_options = None
+  _globals['_OBJECTMETA_ANNOTATIONSENTRY']._serialized_options = b'8\001'
+  _globals['_CREDENTIALHANDLE_METADATAENTRY']._loaded_options = None
+  _globals['_CREDENTIALHANDLE_METADATAENTRY']._serialized_options = b'8\001'
+  _globals['_PROVIDER_CREDENTIALSENTRY']._loaded_options = None
+  _globals['_PROVIDER_CREDENTIALSENTRY']._serialized_options = b'8\001'
+  _globals['_PROVIDER_CONFIGENTRY']._loaded_options = None
+  _globals['_PROVIDER_CONFIGENTRY']._serialized_options = b'8\001'
+  _globals['_PROVIDER_CREDENTIALEXPIRESATMSENTRY']._loaded_options = None
+  _globals['_PROVIDER_CREDENTIALEXPIRESATMSENTRY']._serialized_options = b'8\001'
+  _globals['_PROVIDER_CREDENTIALHANDLESENTRY']._loaded_options = None
+  _globals['_PROVIDER_CREDENTIALHANDLESENTRY']._serialized_options = b'8\001'
+  _globals['_PROVIDER'].fields_by_name['credentials']._loaded_options = None
+  _globals['_PROVIDER'].fields_by_name['credentials']._serialized_options = b'\210\265\030\001'
+  _globals['_WORKSPACEPHASE']._serialized_start=1498
+  _globals['_WORKSPACEPHASE']._serialized_end=1608
+  _globals['_OBJECTMETA']._serialized_start=59
+  _globals['_OBJECTMETA']._serialized_end=433
+  _globals['_OBJECTMETA_LABELSENTRY']._serialized_start=336
+  _globals['_OBJECTMETA_LABELSENTRY']._serialized_end=381
+  _globals['_OBJECTMETA_ANNOTATIONSENTRY']._serialized_start=383
+  _globals['_OBJECTMETA_ANNOTATIONSENTRY']._serialized_end=433
+  _globals['_WORKSPACESTATUS']._serialized_start=435
+  _globals['_WORKSPACESTATUS']._serialized_end=507
+  _globals['_WORKSPACE']._serialized_start=509
+  _globals['_WORKSPACE']._serialized_end=631
+  _globals['_CREDENTIALHANDLE']._serialized_start=634
+  _globals['_CREDENTIALHANDLE']._serialized_end=807
+  _globals['_CREDENTIALHANDLE_METADATAENTRY']._serialized_start=760
+  _globals['_CREDENTIALHANDLE_METADATAENTRY']._serialized_end=807
+  _globals['_PROVIDER']._serialized_start=810
+  _globals['_PROVIDER']._serialized_end=1496
+  _globals['_PROVIDER_CREDENTIALSENTRY']._serialized_start=1237
+  _globals['_PROVIDER_CREDENTIALSENTRY']._serialized_end=1287
+  _globals['_PROVIDER_CONFIGENTRY']._serialized_start=1289
+  _globals['_PROVIDER_CONFIGENTRY']._serialized_end=1334
+  _globals['_PROVIDER_CREDENTIALEXPIRESATMSENTRY']._serialized_start=1336
+  _globals['_PROVIDER_CREDENTIALEXPIRESATMSENTRY']._serialized_end=1396
+  _globals['_PROVIDER_CREDENTIALHANDLESENTRY']._serialized_start=1398
+  _globals['_PROVIDER_CREDENTIALHANDLESENTRY']._serialized_end=1496
+# @@protoc_insertion_point(module_scope)

openshell/_proto/datamodel_pb2.pyi

@@ -0,0 +1,129 @@
+import options_pb2 as _options_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class WorkspacePhase(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
+    __slots__ = ()
+    WORKSPACE_PHASE_UNSPECIFIED: _ClassVar[WorkspacePhase]
+    WORKSPACE_PHASE_ACTIVE: _ClassVar[WorkspacePhase]
+    WORKSPACE_PHASE_TERMINATING: _ClassVar[WorkspacePhase]
+WORKSPACE_PHASE_UNSPECIFIED: WorkspacePhase
+WORKSPACE_PHASE_ACTIVE: WorkspacePhase
+WORKSPACE_PHASE_TERMINATING: WorkspacePhase
+
+class ObjectMeta(_message.Message):
+    __slots__ = ("id", "name", "created_at_ms", "labels", "resource_version", "annotations", "workspace", "deletion_timestamp_ms")
+    class LabelsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    class AnnotationsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    ID_FIELD_NUMBER: _ClassVar[int]
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    CREATED_AT_MS_FIELD_NUMBER: _ClassVar[int]
+    LABELS_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_VERSION_FIELD_NUMBER: _ClassVar[int]
+    ANNOTATIONS_FIELD_NUMBER: _ClassVar[int]
+    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    DELETION_TIMESTAMP_MS_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    name: str
+    created_at_ms: int
+    labels: _containers.ScalarMap[str, str]
+    resource_version: int
+    annotations: _containers.ScalarMap[str, str]
+    workspace: str
+    deletion_timestamp_ms: int
+    def __init__(self, id: _Optional[str] = ..., name: _Optional[str] = ..., created_at_ms: _Optional[int] = ..., labels: _Optional[_Mapping[str, str]] = ..., resource_version: _Optional[int] = ..., annotations: _Optional[_Mapping[str, str]] = ..., workspace: _Optional[str] = ..., deletion_timestamp_ms: _Optional[int] = ...) -> None: ...
+
+class WorkspaceStatus(_message.Message):
+    __slots__ = ("phase",)
+    PHASE_FIELD_NUMBER: _ClassVar[int]
+    phase: WorkspacePhase
+    def __init__(self, phase: _Optional[_Union[WorkspacePhase, str]] = ...) -> None: ...
+
+class Workspace(_message.Message):
+    __slots__ = ("metadata", "status")
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    STATUS_FIELD_NUMBER: _ClassVar[int]
+    metadata: ObjectMeta
+    status: WorkspaceStatus
+    def __init__(self, metadata: _Optional[_Union[ObjectMeta, _Mapping]] = ..., status: _Optional[_Union[WorkspaceStatus, _Mapping]] = ...) -> None: ...
+
+class CredentialHandle(_message.Message):
+    __slots__ = ("driver", "handle", "metadata")
+    class MetadataEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    DRIVER_FIELD_NUMBER: _ClassVar[int]
+    HANDLE_FIELD_NUMBER: _ClassVar[int]
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    driver: str
+    handle: str
+    metadata: _containers.ScalarMap[str, str]
+    def __init__(self, driver: _Optional[str] = ..., handle: _Optional[str] = ..., metadata: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class Provider(_message.Message):
+    __slots__ = ("metadata", "type", "credentials", "config", "credential_expires_at_ms", "profile_workspace", "credential_handles")
+    class CredentialsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    class ConfigEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    class CredentialExpiresAtMsEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: int
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[int] = ...) -> None: ...
+    class CredentialHandlesEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: CredentialHandle
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[CredentialHandle, _Mapping]] = ...) -> None: ...
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    TYPE_FIELD_NUMBER: _ClassVar[int]
+    CREDENTIALS_FIELD_NUMBER: _ClassVar[int]
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    CREDENTIAL_EXPIRES_AT_MS_FIELD_NUMBER: _ClassVar[int]
+    PROFILE_WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    CREDENTIAL_HANDLES_FIELD_NUMBER: _ClassVar[int]
+    metadata: ObjectMeta
+    type: str
+    credentials: _containers.ScalarMap[str, str]
+    config: _containers.ScalarMap[str, str]
+    credential_expires_at_ms: _containers.ScalarMap[str, int]
+    profile_workspace: str
+    credential_handles: _containers.MessageMap[str, CredentialHandle]
+    def __init__(self, metadata: _Optional[_Union[ObjectMeta, _Mapping]] = ..., type: _Optional[str] = ..., credentials: _Optional[_Mapping[str, str]] = ..., config: _Optional[_Mapping[str, str]] = ..., credential_expires_at_ms: _Optional[_Mapping[str, int]] = ..., profile_workspace: _Optional[str] = ..., credential_handles: _Optional[_Mapping[str, CredentialHandle]] = ...) -> None: ...

openshell/_proto/datamodel_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.78.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + ' but the generated code in datamodel_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

openshell/_proto/inference_pb2.py

@@ -0,0 +1,72 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: inference.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'inference.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from . import datamodel_pb2 as datamodel__pb2
+from . import options_pb2 as options__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0finference.proto\x12\x16openshell.inference.v1\x1a\x0f\x64\x61tamodel.proto\x1a\roptions.proto\"U\n\x14InferenceRouteConfig\x12\x15\n\rprovider_name\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x14\n\x0ctimeout_secs\x18\x03 \x01(\x04\"\x95\x01\n\x0eInferenceRoute\x12\x34\n\x08metadata\x18\x01 \x01(\x0b\x32\".openshell.datamodel.v1.ObjectMeta\x12<\n\x06\x63onfig\x18\x02 \x01(\x0b\x32,.openshell.inference.v1.InferenceRouteConfig\x12\x0f\n\x07version\x18\x03 \x01(\x04\"\xa3\x01\n\x18SetInferenceRouteRequest\x12\x15\n\rprovider_name\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x12\n\nroute_name\x18\x03 \x01(\t\x12\x0e\n\x06verify\x18\x04 \x01(\x08\x12\x11\n\tno_verify\x18\x05 \x01(\x08\x12\x14\n\x0ctimeout_secs\x18\x06 \x01(\x04\x12\x11\n\tworkspace\x18\x07 \x01(\t\"2\n\x11ValidatedEndpoint\x12\x0b\n\x03url\x18\x01 \x01(\t\x12\x10\n\x08protocol\x18\x02 \x01(\t\"\xf8\x01\n\x19SetInferenceRouteResponse\x12\x15\n\rprovider_name\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\x04\x12\x12\n\nroute_name\x18\x04 \x01(\t\x12\x1c\n\x14validation_performed\x18\x05 \x01(\x08\x12\x46\n\x13validated_endpoints\x18\x06 \x03(\x0b\x32).openshell.inference.v1.ValidatedEndpoint\x12\x14\n\x0ctimeout_secs\x18\x07 \x01(\x04\x12\x11\n\tworkspace\x18\x08 \x01(\t\"A\n\x18GetInferenceRouteRequest\x12\x12\n\nroute_name\x18\x01 \x01(\t\x12\x11\n\tworkspace\x18\x02 \x01(\t\"\x92\x01\n\x19GetInferenceRouteResponse\x12\x15\n\rprovider_name\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x0f\n\x07version\x18\x03 \x01(\x04\x12\x12\n\nroute_name\x18\x04 \x01(\t\x12\x14\n\x0ctimeout_secs\x18\x05 \x01(\x04\x12\x11\n\tworkspace\x18\x06 \x01(\t\"D\n\x1b\x44\x65leteInferenceRouteRequest\x12\x12\n\nroute_name\x18\x01 \x01(\t\x12\x11\n\tworkspace\x18\x02 \x01(\t\"/\n\x1c\x44\x65leteInferenceRouteResponse\x12\x0f\n\x07\x64\x65leted\x18\x01 \x01(\x08\"\x1b\n\x19GetInferenceBundleRequest\"\xed\x01\n\rResolvedRoute\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x10\n\x08\x62\x61se_url\x18\x02 \x01(\t\x12\x11\n\tprotocols\x18\x03 \x03(\t\x12\x15\n\x07\x61pi_key\x18\x04 \x01(\tB\x04\x88\xb5\x18\x01\x12\x10\n\x08model_id\x18\x05 \x01(\t\x12\x15\n\rprovider_type\x18\x06 \x01(\t\x12\x14\n\x0ctimeout_secs\x18\x07 \x01(\x04\x12\x15\n\rmodel_in_path\x18\x08 \x01(\x08\x12\"\n\x15request_path_override\x18\t \x01(\tH\x00\x88\x01\x01\x42\x18\n\x16_request_path_override\"~\n\x1aGetInferenceBundleResponse\x12\x35\n\x06routes\x18\x01 \x03(\x0b\x32%.openshell.inference.v1.ResolvedRoute\x12\x10\n\x08revision\x18\x02 \x01(\t\x12\x17\n\x0fgenerated_at_ms\x18\x03 \x01(\x03\x32\x82\x05\n\tInference\x12\x8a\x01\n\x12GetInferenceBundle\x12\x31.openshell.inference.v1.GetInferenceBundleRequest\x1a\x32.openshell.inference.v1.GetInferenceBundleResponse\"\r\x82\xb5\x18\t\n\x07sandbox\x12\x9e\x01\n\x11SetInferenceRoute\x12\x30.openshell.inference.v1.SetInferenceRouteRequest\x1a\x31.openshell.inference.v1.SetInferenceRouteResponse\"$\x82\xb5\x18 \n\x06\x62\x65\x61rer\x12\x05\x61\x64min\"\x0finference:write\x12\x9c\x01\n\x11GetInferenceRoute\x12\x30.openshell.inference.v1.GetInferenceRouteRequest\x1a\x31.openshell.inference.v1.GetInferenceRouteResponse\"\"\x82\xb5\x18\x1e\n\x06\x62\x65\x61rer\x12\x04user\"\x0einference:read\x12\xa7\x01\n\x14\x44\x65leteInferenceRoute\x12\x33.openshell.inference.v1.DeleteInferenceRouteRequest\x1a\x34.openshell.inference.v1.DeleteInferenceRouteResponse\"$\x82\xb5\x18 \n\x06\x62\x65\x61rer\x12\x05\x61\x64min\"\x0finference:writeb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'inference_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_RESOLVEDROUTE'].fields_by_name['api_key']._loaded_options = None
+  _globals['_RESOLVEDROUTE'].fields_by_name['api_key']._serialized_options = b'\210\265\030\001'
+  _globals['_INFERENCE'].methods_by_name['GetInferenceBundle']._loaded_options = None
+  _globals['_INFERENCE'].methods_by_name['GetInferenceBundle']._serialized_options = b'\202\265\030\t\n\007sandbox'
+  _globals['_INFERENCE'].methods_by_name['SetInferenceRoute']._loaded_options = None
+  _globals['_INFERENCE'].methods_by_name['SetInferenceRoute']._serialized_options = b'\202\265\030 \n\006bearer\022\005admin\"\017inference:write'
+  _globals['_INFERENCE'].methods_by_name['GetInferenceRoute']._loaded_options = None
+  _globals['_INFERENCE'].methods_by_name['GetInferenceRoute']._serialized_options = b'\202\265\030\036\n\006bearer\022\004user\"\016inference:read'
+  _globals['_INFERENCE'].methods_by_name['DeleteInferenceRoute']._loaded_options = None
+  _globals['_INFERENCE'].methods_by_name['DeleteInferenceRoute']._serialized_options = b'\202\265\030 \n\006bearer\022\005admin\"\017inference:write'
+  _globals['_INFERENCEROUTECONFIG']._serialized_start=75
+  _globals['_INFERENCEROUTECONFIG']._serialized_end=160
+  _globals['_INFERENCEROUTE']._serialized_start=163
+  _globals['_INFERENCEROUTE']._serialized_end=312
+  _globals['_SETINFERENCEROUTEREQUEST']._serialized_start=315
+  _globals['_SETINFERENCEROUTEREQUEST']._serialized_end=478
+  _globals['_VALIDATEDENDPOINT']._serialized_start=480
+  _globals['_VALIDATEDENDPOINT']._serialized_end=530
+  _globals['_SETINFERENCEROUTERESPONSE']._serialized_start=533
+  _globals['_SETINFERENCEROUTERESPONSE']._serialized_end=781
+  _globals['_GETINFERENCEROUTEREQUEST']._serialized_start=783
+  _globals['_GETINFERENCEROUTEREQUEST']._serialized_end=848
+  _globals['_GETINFERENCEROUTERESPONSE']._serialized_start=851
+  _globals['_GETINFERENCEROUTERESPONSE']._serialized_end=997
+  _globals['_DELETEINFERENCEROUTEREQUEST']._serialized_start=999
+  _globals['_DELETEINFERENCEROUTEREQUEST']._serialized_end=1067
+  _globals['_DELETEINFERENCEROUTERESPONSE']._serialized_start=1069
+  _globals['_DELETEINFERENCEROUTERESPONSE']._serialized_end=1116
+  _globals['_GETINFERENCEBUNDLEREQUEST']._serialized_start=1118
+  _globals['_GETINFERENCEBUNDLEREQUEST']._serialized_end=1145
+  _globals['_RESOLVEDROUTE']._serialized_start=1148
+  _globals['_RESOLVEDROUTE']._serialized_end=1385
+  _globals['_GETINFERENCEBUNDLERESPONSE']._serialized_start=1387
+  _globals['_GETINFERENCEBUNDLERESPONSE']._serialized_end=1513
+  _globals['_INFERENCE']._serialized_start=1516
+  _globals['_INFERENCE']._serialized_end=2158
+# @@protoc_insertion_point(module_scope)

openshell/_proto/inference_pb2.pyi

@@ -0,0 +1,149 @@
+import datamodel_pb2 as _datamodel_pb2
+import options_pb2 as _options_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class InferenceRouteConfig(_message.Message):
+    __slots__ = ("provider_name", "model_id", "timeout_secs")
+    PROVIDER_NAME_FIELD_NUMBER: _ClassVar[int]
+    MODEL_ID_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_SECS_FIELD_NUMBER: _ClassVar[int]
+    provider_name: str
+    model_id: str
+    timeout_secs: int
+    def __init__(self, provider_name: _Optional[str] = ..., model_id: _Optional[str] = ..., timeout_secs: _Optional[int] = ...) -> None: ...
+
+class InferenceRoute(_message.Message):
+    __slots__ = ("metadata", "config", "version")
+    METADATA_FIELD_NUMBER: _ClassVar[int]
+    CONFIG_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    metadata: _datamodel_pb2.ObjectMeta
+    config: InferenceRouteConfig
+    version: int
+    def __init__(self, metadata: _Optional[_Union[_datamodel_pb2.ObjectMeta, _Mapping]] = ..., config: _Optional[_Union[InferenceRouteConfig, _Mapping]] = ..., version: _Optional[int] = ...) -> None: ...
+
+class SetInferenceRouteRequest(_message.Message):
+    __slots__ = ("provider_name", "model_id", "route_name", "verify", "no_verify", "timeout_secs", "workspace")
+    PROVIDER_NAME_FIELD_NUMBER: _ClassVar[int]
+    MODEL_ID_FIELD_NUMBER: _ClassVar[int]
+    ROUTE_NAME_FIELD_NUMBER: _ClassVar[int]
+    VERIFY_FIELD_NUMBER: _ClassVar[int]
+    NO_VERIFY_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_SECS_FIELD_NUMBER: _ClassVar[int]
+    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    provider_name: str
+    model_id: str
+    route_name: str
+    verify: bool
+    no_verify: bool
+    timeout_secs: int
+    workspace: str
+    def __init__(self, provider_name: _Optional[str] = ..., model_id: _Optional[str] = ..., route_name: _Optional[str] = ..., verify: bool = ..., no_verify: bool = ..., timeout_secs: _Optional[int] = ..., workspace: _Optional[str] = ...) -> None: ...
+
+class ValidatedEndpoint(_message.Message):
+    __slots__ = ("url", "protocol")
+    URL_FIELD_NUMBER: _ClassVar[int]
+    PROTOCOL_FIELD_NUMBER: _ClassVar[int]
+    url: str
+    protocol: str
+    def __init__(self, url: _Optional[str] = ..., protocol: _Optional[str] = ...) -> None: ...
+
+class SetInferenceRouteResponse(_message.Message):
+    __slots__ = ("provider_name", "model_id", "version", "route_name", "validation_performed", "validated_endpoints", "timeout_secs", "workspace")
+    PROVIDER_NAME_FIELD_NUMBER: _ClassVar[int]
+    MODEL_ID_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    ROUTE_NAME_FIELD_NUMBER: _ClassVar[int]
+    VALIDATION_PERFORMED_FIELD_NUMBER: _ClassVar[int]
+    VALIDATED_ENDPOINTS_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_SECS_FIELD_NUMBER: _ClassVar[int]
+    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    provider_name: str
+    model_id: str
+    version: int
+    route_name: str
+    validation_performed: bool
+    validated_endpoints: _containers.RepeatedCompositeFieldContainer[ValidatedEndpoint]
+    timeout_secs: int
+    workspace: str
+    def __init__(self, provider_name: _Optional[str] = ..., model_id: _Optional[str] = ..., version: _Optional[int] = ..., route_name: _Optional[str] = ..., validation_performed: bool = ..., validated_endpoints: _Optional[_Iterable[_Union[ValidatedEndpoint, _Mapping]]] = ..., timeout_secs: _Optional[int] = ..., workspace: _Optional[str] = ...) -> None: ...
+
+class GetInferenceRouteRequest(_message.Message):
+    __slots__ = ("route_name", "workspace")
+    ROUTE_NAME_FIELD_NUMBER: _ClassVar[int]
+    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    route_name: str
+    workspace: str
+    def __init__(self, route_name: _Optional[str] = ..., workspace: _Optional[str] = ...) -> None: ...
+
+class GetInferenceRouteResponse(_message.Message):
+    __slots__ = ("provider_name", "model_id", "version", "route_name", "timeout_secs", "workspace")
+    PROVIDER_NAME_FIELD_NUMBER: _ClassVar[int]
+    MODEL_ID_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    ROUTE_NAME_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_SECS_FIELD_NUMBER: _ClassVar[int]
+    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    provider_name: str
+    model_id: str
+    version: int
+    route_name: str
+    timeout_secs: int
+    workspace: str
+    def __init__(self, provider_name: _Optional[str] = ..., model_id: _Optional[str] = ..., version: _Optional[int] = ..., route_name: _Optional[str] = ..., timeout_secs: _Optional[int] = ..., workspace: _Optional[str] = ...) -> None: ...
+
+class DeleteInferenceRouteRequest(_message.Message):
+    __slots__ = ("route_name", "workspace")
+    ROUTE_NAME_FIELD_NUMBER: _ClassVar[int]
+    WORKSPACE_FIELD_NUMBER: _ClassVar[int]
+    route_name: str
+    workspace: str
+    def __init__(self, route_name: _Optional[str] = ..., workspace: _Optional[str] = ...) -> None: ...
+
+class DeleteInferenceRouteResponse(_message.Message):
+    __slots__ = ("deleted",)
+    DELETED_FIELD_NUMBER: _ClassVar[int]
+    deleted: bool
+    def __init__(self, deleted: bool = ...) -> None: ...
+
+class GetInferenceBundleRequest(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class ResolvedRoute(_message.Message):
+    __slots__ = ("name", "base_url", "protocols", "api_key", "model_id", "provider_type", "timeout_secs", "model_in_path", "request_path_override")
+    NAME_FIELD_NUMBER: _ClassVar[int]
+    BASE_URL_FIELD_NUMBER: _ClassVar[int]
+    PROTOCOLS_FIELD_NUMBER: _ClassVar[int]
+    API_KEY_FIELD_NUMBER: _ClassVar[int]
+    MODEL_ID_FIELD_NUMBER: _ClassVar[int]
+    PROVIDER_TYPE_FIELD_NUMBER: _ClassVar[int]
+    TIMEOUT_SECS_FIELD_NUMBER: _ClassVar[int]
+    MODEL_IN_PATH_FIELD_NUMBER: _ClassVar[int]
+    REQUEST_PATH_OVERRIDE_FIELD_NUMBER: _ClassVar[int]
+    name: str
+    base_url: str
+    protocols: _containers.RepeatedScalarFieldContainer[str]
+    api_key: str
+    model_id: str
+    provider_type: str
+    timeout_secs: int
+    model_in_path: bool
+    request_path_override: str
+    def __init__(self, name: _Optional[str] = ..., base_url: _Optional[str] = ..., protocols: _Optional[_Iterable[str]] = ..., api_key: _Optional[str] = ..., model_id: _Optional[str] = ..., provider_type: _Optional[str] = ..., timeout_secs: _Optional[int] = ..., model_in_path: bool = ..., request_path_override: _Optional[str] = ...) -> None: ...
+
+class GetInferenceBundleResponse(_message.Message):
+    __slots__ = ("routes", "revision", "generated_at_ms")
+    ROUTES_FIELD_NUMBER: _ClassVar[int]
+    REVISION_FIELD_NUMBER: _ClassVar[int]
+    GENERATED_AT_MS_FIELD_NUMBER: _ClassVar[int]
+    routes: _containers.RepeatedCompositeFieldContainer[ResolvedRoute]
+    revision: str
+    generated_at_ms: int
+    def __init__(self, routes: _Optional[_Iterable[_Union[ResolvedRoute, _Mapping]]] = ..., revision: _Optional[str] = ..., generated_at_ms: _Optional[int] = ...) -> None: ...