cuvis-ai-schemas 0.1.0__py3-none-any.whl

cuvis_ai_schemas/__init__.py

@@ -0,0 +1,5 @@
+"""cuvis-ai-schemas: Lightweight schema definitions for the cuvis-ai ecosystem."""
+
+__version__ = "0.1.0"
+
+__all__ = ["__version__"]

cuvis_ai_schemas/discovery/__init__.py

@@ -0,0 +1,6 @@
+"""Discovery and metadata schemas."""
+
+# Discovery schemas are primarily defined in proto
+# Python implementations will be added as needed
+
+__all__: list[str] = []

cuvis_ai_schemas/enums/__init__.py

@@ -0,0 +1,5 @@
+"""Shared enums across cuvis-ai ecosystem."""
+
+from cuvis_ai_schemas.enums.types import ArtifactType, ExecutionStage
+
+__all__ = ["ExecutionStage", "ArtifactType"]

cuvis_ai_schemas/enums/types.py

@@ -0,0 +1,30 @@
+"""Shared type enums for cuvis-ai ecosystem."""
+
+from enum import StrEnum
+
+
+class ExecutionStage(StrEnum):
+    """Execution stages for node filtering.
+
+    Nodes can specify which stages they should execute in to enable
+    stage-aware graph execution (e.g., loss nodes only in training).
+    """
+
+    ALWAYS = "always"
+    TRAIN = "train"
+    VAL = "val"
+    VALIDATE = "val"
+    TEST = "test"
+    INFERENCE = "inference"
+
+
+class ArtifactType(StrEnum):
+    """Types of artifacts with different validation/logging policies.
+
+    Attributes
+    ----------
+    IMAGE : str
+        Image artifact - expects shape (H, W, 1) monocular or (H, W, 3) RGB
+    """
+
+    IMAGE = "image"

cuvis_ai_schemas/execution/__init__.py

@@ -0,0 +1,12 @@
+"""Execution context schemas."""
+
+from collections.abc import Iterator
+from typing import Any
+
+from cuvis_ai_schemas.execution.context import Context
+from cuvis_ai_schemas.execution.monitoring import Artifact, Metric
+
+# Type alias for data streaming
+InputStream = Iterator[dict[str, Any]]
+
+__all__ = ["Context", "Artifact", "Metric", "InputStream"]

cuvis_ai_schemas/execution/context.py

@@ -0,0 +1,41 @@
+"""Execution context for pipeline execution."""
+
+from dataclasses import dataclass
+
+from cuvis_ai_schemas.enums.types import ExecutionStage
+
+
+@dataclass
+class Context:
+    """Execution context passed to executor and nodes.
+
+    Contains runtime information that doesn't flow through data edges.
+    This replaces mutable global state with explicit context parameters.
+
+    Attributes
+    ----------
+    stage : ExecutionStage
+        Execution stage: "train", "val", "test", "inference"
+    epoch : int
+        Current training epoch
+    batch_idx : int
+        Current batch index within epoch
+    global_step : int
+        Global training step across all epochs
+
+    Examples
+    --------
+    >>> context = Context(stage=ExecutionStage.TRAIN, epoch=5, batch_idx=42, global_step=1337)
+    >>> executor.forward(context=context, batch=batch)
+
+    Notes
+    -----
+    Future extensions for distributed training:
+    - rank: int (process rank in distributed training)
+    - world_size: int (total number of processes)
+    """
+
+    stage: ExecutionStage = ExecutionStage.INFERENCE
+    epoch: int = 0
+    batch_idx: int = 0
+    global_step: int = 0

cuvis_ai_schemas/execution/monitoring.py

@@ -0,0 +1,83 @@
+"""Monitoring schemas for artifacts and metrics."""
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING
+
+from cuvis_ai_schemas.enums.types import ArtifactType, ExecutionStage
+
+if TYPE_CHECKING:
+    import numpy as np
+
+
+@dataclass
+class Artifact:
+    """Artifact for logging visualizations and data to monitoring systems.
+
+    Attributes
+    ----------
+    name : str
+        Name/identifier for the artifact
+    value : np.ndarray
+        Numpy array containing the artifact data (shape validated by type)
+    el_id : int
+        Element ID (e.g., batch item index, image index)
+    desc : str
+        Human-readable description of the artifact
+    type : ArtifactType
+        Type of artifact, determines validation and logging policy
+    stage : ExecutionStage
+        Execution stage when artifact was generated
+    epoch : int
+        Training epoch when artifact was generated
+    batch_idx : int
+        Batch index when artifact was generated
+
+    Examples
+    --------
+    >>> import numpy as np
+    >>> artifact = Artifact(
+    ...     name="heatmap_img_0",
+    ...     value=np.random.rand(256, 256, 3),
+    ...     el_id=0,
+    ...     desc="Anomaly heatmap for first image",
+    ...     type=ArtifactType.IMAGE
+    ... )
+    """
+
+    name: str
+    value: "np.ndarray"
+    el_id: int
+    desc: str
+    type: ArtifactType
+    stage: ExecutionStage = ExecutionStage.INFERENCE
+    epoch: int = 0
+    batch_idx: int = 0
+
+
+@dataclass
+class Metric:
+    """Metric for logging scalar values to monitoring systems.
+
+    Attributes
+    ----------
+    name : str
+        Name/identifier for the metric
+    value : float
+        Scalar metric value
+    stage : ExecutionStage
+        Execution stage when metric was recorded
+    epoch : int
+        Training epoch when metric was recorded
+    batch_idx : int
+        Batch index when metric was recorded
+
+    Examples
+    --------
+    >>> metric = Metric(name="loss/train", value=0.123, stage=ExecutionStage.TRAIN)
+    """
+
+    name: str
+    value: float
+    stage: ExecutionStage = ExecutionStage.INFERENCE
+    epoch: int = 0
+    batch_idx: int = 0

cuvis_ai_schemas/extensions/__init__.py

@@ -0,0 +1,3 @@
+"""Optional extensions for specific use cases."""
+
+__all__: list[str] = []

cuvis_ai_schemas/extensions/ui/__init__.py

@@ -0,0 +1,8 @@
+"""UI-specific extensions for port display."""
+
+from cuvis_ai_schemas.extensions.ui.port_display import (
+    DTYPE_COLORS,
+    PortDisplaySpec,
+)
+
+__all__ = ["PortDisplaySpec", "DTYPE_COLORS"]

cuvis_ai_schemas/extensions/ui/port_display.py

@@ -0,0 +1,159 @@
+"""UI display extensions for PortSpec."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from cuvis_ai_schemas.pipeline.ports import PortSpec
+
+# Default RGB colors for different data types
+# Extended with hyperspectral/cuvis-ai specific types for comprehensive UI support
+DTYPE_COLORS: dict[str, tuple[int, int, int]] = {
+    # === PyTorch Tensor Types ===
+    "torch.Tensor": (100, 150, 255),  # Blue for generic tensors
+    "tensor": (255, 200, 150),  # Peach for generic tensor type
+    # Floating point types - warm orange/yellow tones
+    "torch.float16": (255, 180, 80),  # Light orange
+    "torch.float32": (255, 150, 50),  # Orange
+    "torch.float64": (255, 120, 30),  # Dark orange
+    "torch.bfloat16": (255, 200, 100),  # Pale orange
+    "float16": (255, 180, 80),
+    "float32": (255, 150, 50),
+    "float64": (255, 120, 30),
+    "float": (150, 255, 150),  # Green for Python floats
+    # Integer types - cool blue tones
+    "torch.int8": (150, 200, 255),  # Light blue
+    "torch.int16": (120, 180, 255),  # Medium blue
+    "torch.int32": (100, 160, 255),  # Blue
+    "torch.int64": (80, 140, 255),  # Dark blue
+    "torch.uint8": (180, 220, 255),  # Pale blue
+    "torch.uint16": (160, 210, 255),  # Light pale blue
+    "int8": (150, 200, 255),
+    "int16": (120, 180, 255),
+    "int32": (100, 160, 255),
+    "int64": (80, 140, 255),
+    "uint8": (180, 220, 255),
+    "uint16": (160, 210, 255),
+    "int": (255, 200, 100),  # Orange for Python ints
+    # Boolean types - green
+    "torch.bool": (100, 220, 100),  # Green
+    "bool": (255, 150, 150),  # Pink for Python bools (keep original for backward compat)
+    # Complex types - purple tones
+    "torch.complex64": (200, 150, 255),  # Light purple
+    "torch.complex128": (180, 120, 255),  # Purple
+    # === Python Built-in Types ===
+    "str": (200, 200, 200),  # Gray for strings
+    "list": (180, 180, 255),  # Light blue for lists
+    "dict": (255, 180, 255),  # Light purple for dicts
+    "object": (180, 180, 180),  # Gray for generic objects
+    # === Hyperspectral/Cuvis-AI Specific Types ===
+    "cube": (255, 180, 100),  # Orange-ish (hyperspectral data cube)
+    "mask": (150, 255, 150),  # Light green (binary/segmentation mask)
+    "wavelengths": (100, 200, 255),  # Light blue (spectral wavelength array)
+    "bbox": (255, 150, 150),  # Light red (bounding box coordinates)
+    "points": (200, 255, 200),  # Pale green (point cloud/coordinates)
+    "labels": (255, 200, 200),  # Light pink (classification labels)
+    "indices": (200, 180, 255),  # Light purple (index array)
+    "scalar": (255, 255, 180),  # Light yellow (single value)
+    "image": (200, 255, 255),  # Light cyan (2D image output)
+    # === Special/Generic Types ===
+    "any": (255, 255, 255),  # White (accepts anything)
+}
+
+DEFAULT_COLOR = (200, 200, 200)  # Light gray for unknown types
+
+
+class PortDisplaySpec:
+    """UI display wrapper for PortSpec providing color and formatting.
+
+    This class wraps a PortSpec and adds UI-specific display properties
+    like colors and formatted tooltips.
+
+    Attributes
+    ----------
+    port_spec : PortSpec
+        The underlying port specification
+
+    Examples
+    --------
+    >>> from cuvis_ai_schemas.pipeline.ports import PortSpec
+    >>> port = PortSpec(dtype="torch.Tensor", shape=(1, 3, 224, 224))
+    >>> display = PortDisplaySpec(port)
+    >>> display.color
+    (100, 150, 255)
+    >>> display.format_tooltip()
+    'torch.Tensor\\nShape: (1, 3, 224, 224)'
+    """
+
+    def __init__(self, port_spec: PortSpec) -> None:
+        """Initialize display spec.
+
+        Parameters
+        ----------
+        port_spec : PortSpec
+            Port specification to wrap
+        """
+        self.port_spec = port_spec
+
+    @property
+    def color(self) -> tuple[int, int, int]:
+        """Get RGB color for this port's dtype.
+
+        Returns
+        -------
+        tuple[int, int, int]
+            RGB color tuple (0-255 range)
+        """
+        dtype_str = str(self.port_spec.dtype)
+
+        # Try exact match first
+        if dtype_str in DTYPE_COLORS:
+            return DTYPE_COLORS[dtype_str]
+
+        # Try to extract typename for partial matches
+        if hasattr(self.port_spec.dtype, "__name__"):
+            typename = self.port_spec.dtype.__name__
+            if typename in DTYPE_COLORS:
+                return DTYPE_COLORS[typename]
+
+        # Check if it contains "Tensor"
+        if "Tensor" in dtype_str or "tensor" in dtype_str:
+            return DTYPE_COLORS["torch.Tensor"]
+
+        return DEFAULT_COLOR
+
+    @property
+    def display_name(self) -> str:
+        """Get formatted display name for the dtype.
+
+        Returns
+        -------
+        str
+            Human-readable dtype name
+        """
+        if hasattr(self.port_spec.dtype, "__name__"):
+            return self.port_spec.dtype.__name__
+        return str(self.port_spec.dtype)
+
+    def format_tooltip(self) -> str:
+        """Format a tooltip string for UI display.
+
+        Returns
+        -------
+        str
+            Formatted tooltip with dtype, shape, and description
+        """
+        lines = [self.display_name]
+
+        if self.port_spec.shape:
+            shape_str = ", ".join(str(d) for d in self.port_spec.shape)
+            lines.append(f"Shape: ({shape_str})")
+
+        if self.port_spec.description:
+            lines.append(f"\\n{self.port_spec.description}")
+
+        if self.port_spec.optional:
+            lines.append("[Optional]")
+
+        return "\\n".join(lines)

cuvis_ai_schemas/grpc/__init__.py

@@ -0,0 +1,3 @@
+"""gRPC proto messages and helpers."""
+
+__all__: list[str] = []

cuvis_ai_schemas/grpc/v1/__init__.py

@@ -0,0 +1,11 @@
+"""Generated proto files for cuvis.ai gRPC API v1."""
+
+__all__: list[str] = []
+
+try:
+    from cuvis_ai_schemas.grpc.v1 import cuvis_ai_pb2, cuvis_ai_pb2_grpc
+
+    __all__ = ["cuvis_ai_pb2", "cuvis_ai_pb2_grpc"]
+except ImportError:
+    # Proto files not generated yet or proto extra not installed
+    pass

cuvis_ai_schemas/grpc/v1/cuvis_ai_pb2.py

@@ -0,0 +1,240 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: cuvis_ai/grpc/v1/cuvis_ai.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, "", "cuvis_ai/grpc/v1/cuvis_ai.proto"
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n\x1f\x63uvis_ai/grpc/v1/cuvis_ai.proto\x12\x0b\x63uvis_ai.v1"L\n\x06Tensor\x12\r\n\x05shape\x18\x01 \x03(\x03\x12!\n\x05\x64type\x18\x02 \x01(\x0e\x32\x12.cuvis_ai.v1.DType\x12\x10\n\x08raw_data\x18\x03 \x01(\x0c"l\n\x07\x43ontext\x12*\n\x05stage\x18\x01 \x01(\x0e\x32\x1b.cuvis_ai.v1.ExecutionStage\x12\r\n\x05\x65poch\x18\x02 \x01(\x05\x12\x11\n\tbatch_idx\x18\x03 \x01(\x05\x12\x13\n\x0bglobal_step\x18\x04 \x01(\x05"&\n\x0ePipelineConfig\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c""\n\nDataConfig\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"\'\n\x0fOptimizerConfig\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"\'\n\x0fSchedulerConfig\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"\'\n\x0f\x43\x61llbacksConfig\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"~\n\x10PipelineMetadata\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12\x0f\n\x07\x63reated\x18\x03 \x01(\t\x12\x18\n\x10\x63uvis_ai_version\x18\x04 \x01(\t\x12\x0c\n\x04tags\x18\x05 \x03(\t\x12\x0e\n\x06\x61uthor\x18\x06 \x01(\t"\xaa\x01\n\x0cPipelineInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04path\x18\x02 \x01(\t\x12/\n\x08metadata\x18\x03 \x01(\x0b\x32\x1d.cuvis_ai.v1.PipelineMetadata\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x13\n\x0bhas_weights\x18\x05 \x01(\x08\x12\x14\n\x0cweights_path\x18\x06 \x01(\t\x12\x14\n\x0cyaml_content\x18\x07 \x01(\t"&\n\x0eTrainingConfig\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"&\n\x0eTrainRunConfig\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"]\n\x0b\x42oundingBox\x12\x12\n\nelement_id\x18\x01 \x01(\x05\x12\r\n\x05x_min\x18\x02 \x01(\x02\x12\r\n\x05y_min\x18\x03 \x01(\x02\x12\r\n\x05x_max\x18\x04 \x01(\x02\x12\r\n\x05y_max\x18\x05 \x01(\x02"8\n\rBoundingBoxes\x12\'\n\x05\x62oxes\x18\x01 \x03(\x0b\x32\x18.cuvis_ai.v1.BoundingBox"W\n\x05Point\x12\x12\n\nelement_id\x18\x01 \x01(\x05\x12\t\n\x01x\x18\x02 \x01(\x02\x12\t\n\x01y\x18\x03 \x01(\x02\x12$\n\x04type\x18\x04 \x01(\x0e\x32\x16.cuvis_ai.v1.PointType",\n\x06Points\x12"\n\x06points\x18\x01 \x03(\x0b\x32\x12.cuvis_ai.v1.Point"\xeb\x02\n\nInputBatch\x12(\n\x0bwavelengths\x18\x01 \x01(\x0b\x32\x13.cuvis_ai.v1.Tensor\x12!\n\x04\x63ube\x18\x02 \x01(\x0b\x32\x13.cuvis_ai.v1.Tensor\x12!\n\x04mask\x18\x03 \x01(\x0b\x32\x13.cuvis_ai.v1.Tensor\x12*\n\x06\x62\x62oxes\x18\x04 \x01(\x0b\x32\x1a.cuvis_ai.v1.BoundingBoxes\x12#\n\x06points\x18\x05 \x01(\x0b\x32\x13.cuvis_ai.v1.Points\x12\x13\n\x0btext_prompt\x18\x06 \x01(\t\x12>\n\x0c\x65xtra_inputs\x18\x07 \x03(\x0b\x32(.cuvis_ai.v1.InputBatch.ExtraInputsEntry\x1aG\n\x10\x45xtraInputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12"\n\x05value\x18\x02 \x01(\x0b\x32\x13.cuvis_ai.v1.Tensor:\x02\x38\x01"^\n\nTensorSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\r\n\x05shape\x18\x02 \x03(\x03\x12!\n\x05\x64type\x18\x03 \x01(\x0e\x32\x12.cuvis_ai.v1.DType\x12\x10\n\x08required\x18\x04 \x01(\x08"\xc2\x02\n\rTrainResponse\x12%\n\x07\x63ontext\x18\x01 \x01(\x0b\x32\x14.cuvis_ai.v1.Context\x12\x36\n\x06losses\x18\x02 \x03(\x0b\x32&.cuvis_ai.v1.TrainResponse.LossesEntry\x12\x38\n\x07metrics\x18\x03 \x03(\x0b\x32\'.cuvis_ai.v1.TrainResponse.MetricsEntry\x12(\n\x06status\x18\x04 \x01(\x0e\x32\x18.cuvis_ai.v1.TrainStatus\x12\x0f\n\x07message\x18\x05 \x01(\t\x1a-\n\x0bLossesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x02:\x02\x38\x01\x1a.\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x02:\x02\x38\x01"y\n\tParamSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t\x12\x10\n\x08required\x18\x03 \x01(\x08\x12\x15\n\rdefault_value\x18\x04 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x05 \x01(\t\x12\x12\n\nvalidation\x18\x06 \x01(\t"a\n\x10\x43\x61llbackTypeInfo\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t\x12*\n\nparameters\x18\x03 \x03(\x0b\x32\x16.cuvis_ai.v1.ParamSpec"C\n\x15OptimizerParamsSchema\x12*\n\nparameters\x18\x01 \x03(\x0b\x32\x16.cuvis_ai.v1.ParamSpec"C\n\x15SchedulerParamsSchema\x12*\n\nparameters\x18\x01 \x03(\x0b\x32\x16.cuvis_ai.v1.ParamSpec"H\n\x1eListAvailablePipelineesRequest\x12\x17\n\nfilter_tag\x18\x01 \x01(\tH\x00\x88\x01\x01\x42\r\n\x0b_filter_tag"P\n\x1fListAvailablePipelineesResponse\x12-\n\npipelinees\x18\x01 \x03(\x0b\x32\x19.cuvis_ai.v1.PipelineInfo"/\n\x16GetPipelineInfoRequest\x12\x15\n\rpipeline_name\x18\x01 \x01(\t"K\n\x17GetPipelineInfoResponse\x12\x30\n\rpipeline_info\x18\x01 \x01(\x0b\x32\x19.cuvis_ai.v1.PipelineInfo"\x16\n\x14\x43reateSessionRequest"+\n\x15\x43reateSessionResponse\x12\x12\n\nsession_id\x18\x01 \x01(\t"h\n\x1cSetSessionSearchPathsRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\x14\n\x0csearch_paths\x18\x02 \x03(\t\x12\x13\n\x06\x61ppend\x18\x03 \x01(\x08H\x00\x88\x01\x01\x42\t\n\x07_append"_\n\x1dSetSessionSearchPathsResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x15\n\rcurrent_paths\x18\x02 \x03(\t\x12\x16\n\x0erejected_paths\x18\x03 \x03(\t")\n\x13\x43loseSessionRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t"\'\n\x14\x43loseSessionResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08"`\n\x14ResolveConfigRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\x13\n\x0b\x63onfig_type\x18\x02 \x01(\t\x12\x0c\n\x04path\x18\x03 \x01(\t\x12\x11\n\toverrides\x18\x04 \x03(\t"-\n\x15ResolveConfigResponse\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"0\n\x19GetParameterSchemaRequest\x12\x13\n\x0b\x63onfig_type\x18\x01 \x01(\t"1\n\x1aGetParameterSchemaResponse\x12\x13\n\x0bjson_schema\x18\x01 \x01(\t"B\n\x15ValidateConfigRequest\x12\x13\n\x0b\x63onfig_type\x18\x01 \x01(\t\x12\x14\n\x0c\x63onfig_bytes\x18\x02 \x01(\x0c"I\n\x16ValidateConfigResponse\x12\r\n\x05valid\x18\x01 \x01(\x08\x12\x0e\n\x06\x65rrors\x18\x02 \x03(\t\x12\x10\n\x08warnings\x18\x03 \x03(\t"\x93\x01\n\x1aLoadPipelineWeightsRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\x16\n\x0cweights_path\x18\x02 \x01(\tH\x00\x12\x17\n\rweights_bytes\x18\x03 \x01(\x0cH\x00\x12\x13\n\x06strict\x18\x04 \x01(\x08H\x01\x88\x01\x01\x42\x10\n\x0eweights_sourceB\t\n\x07_strict"E\n\x1bLoadPipelineWeightsResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x15\n\rresolved_path\x18\x02 \x01(\t"[\n\x18SetTrainRunConfigRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12+\n\x06\x63onfig\x18\x02 \x01(\x0b\x32\x1b.cuvis_ai.v1.TrainRunConfig"J\n\x19SetTrainRunConfigResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x1c\n\x14pipeline_from_config\x18\x02 \x01(\x08"\xa8\x01\n\x0cTrainRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12.\n\x0ctrainer_type\x18\x02 \x01(\x0e\x32\x18.cuvis_ai.v1.TrainerType\x12%\n\x04\x64\x61ta\x18\x03 \x01(\x0b\x32\x17.cuvis_ai.v1.DataConfig\x12-\n\x08training\x18\x04 \x01(\x0b\x32\x1b.cuvis_ai.v1.TrainingConfig"+\n\x15GetTrainStatusRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t"M\n\x16GetTrainStatusResponse\x12\x33\n\x0flatest_progress\x18\x01 \x01(\x0b\x32\x1a.cuvis_ai.v1.TrainResponse" \n\x1eGetTrainingCapabilitiesRequest"\x95\x02\n\x1fGetTrainingCapabilitiesResponse\x12\x1c\n\x14supported_optimizers\x18\x01 \x03(\t\x12\x1c\n\x14supported_schedulers\x18\x02 \x03(\t\x12:\n\x13supported_callbacks\x18\x03 \x03(\x0b\x32\x1d.cuvis_ai.v1.CallbackTypeInfo\x12<\n\x10optimizer_params\x18\x04 \x01(\x0b\x32".cuvis_ai.v1.OptimizerParamsSchema\x12<\n\x10scheduler_params\x18\x05 \x01(\x0b\x32".cuvis_ai.v1.SchedulerParamsSchema"q\n\x13SavePipelineRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\x15\n\rpipeline_path\x18\x02 \x01(\t\x12/\n\x08metadata\x18\x03 \x01(\x0b\x32\x1d.cuvis_ai.v1.PipelineMetadata"T\n\x14SavePipelineResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x15\n\rpipeline_path\x18\x02 \x01(\t\x12\x14\n\x0cweights_path\x18\x03 \x01(\t"X\n\x13LoadPipelineRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12-\n\x08pipeline\x18\x02 \x01(\x0b\x32\x1b.cuvis_ai.v1.PipelineConfig"X\n\x14LoadPipelineResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12/\n\x08metadata\x18\x02 \x01(\x0b\x32\x1d.cuvis_ai.v1.PipelineMetadata"V\n\x13SaveTrainRunRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\x15\n\rtrainrun_path\x18\x02 \x01(\t\x12\x14\n\x0csave_weights\x18\x03 \x01(\x08"k\n\x14SaveTrainRunResponse\x12\x0f\n\x07success\x18\x01 \x01(\x08\x12\x15\n\rtrainrun_path\x18\x02 \x01(\t\x12\x15\n\rpipeline_path\x18\x03 \x01(\t\x12\x14\n\x0cweights_path\x18\x04 \x01(\t"{\n\x16RestoreTrainRunRequest\x12\x15\n\rtrainrun_path\x18\x01 \x01(\t\x12\x19\n\x0cweights_path\x18\x02 \x01(\tH\x00\x88\x01\x01\x12\x13\n\x06strict\x18\x03 \x01(\x08H\x01\x88\x01\x01\x42\x0f\n\r_weights_pathB\t\n\x07_strict"\\\n\x17RestoreTrainRunResponse\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12-\n\x08trainrun\x18\x02 \x01(\x0b\x32\x1b.cuvis_ai.v1.TrainRunConfig".\n\x18GetPipelineInputsRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t"\xc9\x01\n\x19GetPipelineInputsResponse\x12\x13\n\x0binput_names\x18\x01 \x03(\t\x12K\n\x0binput_specs\x18\x02 \x03(\x0b\x32\x36.cuvis_ai.v1.GetPipelineInputsResponse.InputSpecsEntry\x1aJ\n\x0fInputSpecsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.cuvis_ai.v1.TensorSpec:\x02\x38\x01"/\n\x19GetPipelineOutputsRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t"\xcf\x01\n\x1aGetPipelineOutputsResponse\x12\x14\n\x0coutput_names\x18\x01 \x03(\t\x12N\n\x0coutput_specs\x18\x02 \x03(\x0b\x32\x38.cuvis_ai.v1.GetPipelineOutputsResponse.OutputSpecsEntry\x1aK\n\x10OutputSpecsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12&\n\x05value\x18\x02 \x01(\x0b\x32\x17.cuvis_ai.v1.TensorSpec:\x02\x38\x01"E\n\x1fGetPipelineVisualizationRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\x0e\n\x06\x66ormat\x18\x02 \x01(\t"F\n GetPipelineVisualizationResponse\x12\x12\n\nimage_data\x18\x01 \x01(\x0c\x12\x0e\n\x06\x66ormat\x18\x02 \x01(\t"e\n\x10InferenceRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\'\n\x06inputs\x18\x02 \x01(\x0b\x32\x17.cuvis_ai.v1.InputBatch\x12\x14\n\x0coutput_specs\x18\x03 \x03(\t"\x84\x02\n\x11InferenceResponse\x12<\n\x07outputs\x18\x01 \x03(\x0b\x32+.cuvis_ai.v1.InferenceResponse.OutputsEntry\x12<\n\x07metrics\x18\x02 \x03(\x0b\x32+.cuvis_ai.v1.InferenceResponse.MetricsEntry\x1a\x43\n\x0cOutputsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12"\n\x05value\x18\x02 \x01(\x0b\x32\x13.cuvis_ai.v1.Tensor:\x02\x38\x01\x1a.\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\x02:\x02\x38\x01"&\n\x0ePluginManifest\x12\x14\n\x0c\x63onfig_bytes\x18\x01 \x01(\x0c"W\n\nPluginInfo\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t\x12\x0e\n\x06source\x18\x03 \x01(\t\x12\x0b\n\x03tag\x18\x04 \x01(\t\x12\x10\n\x08provides\x18\x05 \x03(\t"q\n\x08PortSpec\x12\x0c\n\x04name\x18\x01 \x01(\t\x12!\n\x05\x64type\x18\x02 \x01(\x0e\x32\x12.cuvis_ai.v1.DType\x12\r\n\x05shape\x18\x03 \x03(\x03\x12\x10\n\x08optional\x18\x04 \x01(\x08\x12\x13\n\x0b\x64\x65scription\x18\x05 \x01(\t"4\n\x0cPortSpecList\x12$\n\x05specs\x18\x01 \x03(\x0b\x32\x15.cuvis_ai.v1.PortSpec"\xed\x02\n\x08NodeInfo\x12\x12\n\nclass_name\x18\x01 \x01(\t\x12\x11\n\tfull_path\x18\x02 \x01(\t\x12\x0e\n\x06source\x18\x03 \x01(\t\x12\x13\n\x0bplugin_name\x18\x04 \x01(\t\x12:\n\x0binput_specs\x18\x05 \x03(\x0b\x32%.cuvis_ai.v1.NodeInfo.InputSpecsEntry\x12<\n\x0coutput_specs\x18\x06 \x03(\x0b\x32&.cuvis_ai.v1.NodeInfo.OutputSpecsEntry\x1aL\n\x0fInputSpecsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12(\n\x05value\x18\x02 \x01(\x0b\x32\x19.cuvis_ai.v1.PortSpecList:\x02\x38\x01\x1aM\n\x10OutputSpecsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12(\n\x05value\x18\x02 \x01(\x0b\x32\x19.cuvis_ai.v1.PortSpecList:\x02\x38\x01"W\n\x12LoadPluginsRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12-\n\x08manifest\x18\x02 \x01(\x0b\x32\x1b.cuvis_ai.v1.PluginManifest"\xb0\x01\n\x13LoadPluginsResponse\x12\x16\n\x0eloaded_plugins\x18\x01 \x03(\t\x12K\n\x0e\x66\x61iled_plugins\x18\x02 \x03(\x0b\x32\x33.cuvis_ai.v1.LoadPluginsResponse.FailedPluginsEntry\x1a\x34\n\x12\x46\x61iledPluginsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01".\n\x18ListLoadedPluginsRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t"E\n\x19ListLoadedPluginsResponse\x12(\n\x07plugins\x18\x01 \x03(\x0b\x32\x17.cuvis_ai.v1.PluginInfo"?\n\x14GetPluginInfoRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t\x12\x13\n\x0bplugin_name\x18\x02 \x01(\t"@\n\x15GetPluginInfoResponse\x12\'\n\x06plugin\x18\x01 \x01(\x0b\x32\x17.cuvis_ai.v1.PluginInfo"/\n\x19ListAvailableNodesRequest\x12\x12\n\nsession_id\x18\x01 \x01(\t"B\n\x1aListAvailableNodesResponse\x12$\n\x05nodes\x18\x01 \x03(\x0b\x32\x15.cuvis_ai.v1.NodeInfo".\n\x17\x43learPluginCacheRequest\x12\x13\n\x0bplugin_name\x18\x01 \x01(\t"1\n\x18\x43learPluginCacheResponse\x12\x15\n\rcleared_count\x18\x01 \x01(\x05*\xb4\x01\n\x0eProcessingMode\x12\x1f\n\x1bPROCESSING_MODE_UNSPECIFIED\x10\x00\x12\x17\n\x13PROCESSING_MODE_RAW\x10\x01\x12\x1f\n\x1bPROCESSING_MODE_REFLECTANCE\x10\x02\x12 \n\x1cPROCESSING_MODE_DARKSUBTRACT\x10\x03\x12%\n!PROCESSING_MODE_SPECTRAL_RADIANCE\x10\x04*\x9e\x01\n\x0e\x45xecutionStage\x12\x1f\n\x1b\x45XECUTION_STAGE_UNSPECIFIED\x10\x00\x12\x19\n\x15\x45XECUTION_STAGE_TRAIN\x10\x01\x12\x17\n\x13\x45XECUTION_STAGE_VAL\x10\x02\x12\x18\n\x14\x45XECUTION_STAGE_TEST\x10\x03\x12\x1d\n\x19\x45XECUTION_STAGE_INFERENCE\x10\x04*\xb5\x01\n\x05\x44Type\x12\x16\n\x12\x44_TYPE_UNSPECIFIED\x10\x00\x12\x12\n\x0e\x44_TYPE_FLOAT32\x10\x01\x12\x12\n\x0e\x44_TYPE_FLOAT64\x10\x02\x12\x10\n\x0c\x44_TYPE_INT32\x10\x03\x12\x10\n\x0c\x44_TYPE_INT64\x10\x04\x12\x10\n\x0c\x44_TYPE_UINT8\x10\x05\x12\x0f\n\x0b\x44_TYPE_BOOL\x10\x06\x12\x12\n\x0e\x44_TYPE_FLOAT16\x10\x07\x12\x11\n\rD_TYPE_UINT16\x10\x08*d\n\x0bTrainerType\x12\x1c\n\x18TRAINER_TYPE_UNSPECIFIED\x10\x00\x12\x1c\n\x18TRAINER_TYPE_STATISTICAL\x10\x01\x12\x19\n\x15TRAINER_TYPE_GRADIENT\x10\x02*x\n\x0bTrainStatus\x12\x1c\n\x18TRAIN_STATUS_UNSPECIFIED\x10\x00\x12\x18\n\x14TRAIN_STATUS_RUNNING\x10\x01\x12\x19\n\x15TRAIN_STATUS_COMPLETE\x10\x02\x12\x16\n\x12TRAIN_STATUS_ERROR\x10\x03*q\n\tPointType\x12\x1a\n\x16POINT_TYPE_UNSPECIFIED\x10\x00\x12\x17\n\x13POINT_TYPE_POSITIVE\x10\x01\x12\x17\n\x13POINT_TYPE_NEGATIVE\x10\x02\x12\x16\n\x12POINT_TYPE_NEUTRAL\x10\x03\x32\xbf\x13\n\x0e\x43uvisAIService\x12t\n\x17ListAvailablePipelinees\x12+.cuvis_ai.v1.ListAvailablePipelineesRequest\x1a,.cuvis_ai.v1.ListAvailablePipelineesResponse\x12\\\n\x0fGetPipelineInfo\x12#.cuvis_ai.v1.GetPipelineInfoRequest\x1a$.cuvis_ai.v1.GetPipelineInfoResponse\x12V\n\rCreateSession\x12!.cuvis_ai.v1.CreateSessionRequest\x1a".cuvis_ai.v1.CreateSessionResponse\x12n\n\x15SetSessionSearchPaths\x12).cuvis_ai.v1.SetSessionSearchPathsRequest\x1a*.cuvis_ai.v1.SetSessionSearchPathsResponse\x12S\n\x0c\x43loseSession\x12 .cuvis_ai.v1.CloseSessionRequest\x1a!.cuvis_ai.v1.CloseSessionResponse\x12V\n\rResolveConfig\x12!.cuvis_ai.v1.ResolveConfigRequest\x1a".cuvis_ai.v1.ResolveConfigResponse\x12\x65\n\x12GetParameterSchema\x12&.cuvis_ai.v1.GetParameterSchemaRequest\x1a\'.cuvis_ai.v1.GetParameterSchemaResponse\x12Y\n\x0eValidateConfig\x12".cuvis_ai.v1.ValidateConfigRequest\x1a#.cuvis_ai.v1.ValidateConfigResponse\x12h\n\x13LoadPipelineWeights\x12\'.cuvis_ai.v1.LoadPipelineWeightsRequest\x1a(.cuvis_ai.v1.LoadPipelineWeightsResponse\x12\x62\n\x11SetTrainRunConfig\x12%.cuvis_ai.v1.SetTrainRunConfigRequest\x1a&.cuvis_ai.v1.SetTrainRunConfigResponse\x12@\n\x05Train\x12\x19.cuvis_ai.v1.TrainRequest\x1a\x1a.cuvis_ai.v1.TrainResponse0\x01\x12Y\n\x0eGetTrainStatus\x12".cuvis_ai.v1.GetTrainStatusRequest\x1a#.cuvis_ai.v1.GetTrainStatusResponse\x12t\n\x17GetTrainingCapabilities\x12+.cuvis_ai.v1.GetTrainingCapabilitiesRequest\x1a,.cuvis_ai.v1.GetTrainingCapabilitiesResponse\x12S\n\x0cSavePipeline\x12 .cuvis_ai.v1.SavePipelineRequest\x1a!.cuvis_ai.v1.SavePipelineResponse\x12S\n\x0cLoadPipeline\x12 .cuvis_ai.v1.LoadPipelineRequest\x1a!.cuvis_ai.v1.LoadPipelineResponse\x12S\n\x0cSaveTrainRun\x12 .cuvis_ai.v1.SaveTrainRunRequest\x1a!.cuvis_ai.v1.SaveTrainRunResponse\x12\\\n\x0fRestoreTrainRun\x12#.cuvis_ai.v1.RestoreTrainRunRequest\x1a$.cuvis_ai.v1.RestoreTrainRunResponse\x12\x62\n\x11GetPipelineInputs\x12%.cuvis_ai.v1.GetPipelineInputsRequest\x1a&.cuvis_ai.v1.GetPipelineInputsResponse\x12\x65\n\x12GetPipelineOutputs\x12&.cuvis_ai.v1.GetPipelineOutputsRequest\x1a\'.cuvis_ai.v1.GetPipelineOutputsResponse\x12w\n\x18GetPipelineVisualization\x12,.cuvis_ai.v1.GetPipelineVisualizationRequest\x1a-.cuvis_ai.v1.GetPipelineVisualizationResponse\x12J\n\tInference\x12\x1d.cuvis_ai.v1.InferenceRequest\x1a\x1e.cuvis_ai.v1.InferenceResponse\x12P\n\x0bLoadPlugins\x12\x1f.cuvis_ai.v1.LoadPluginsRequest\x1a .cuvis_ai.v1.LoadPluginsResponse\x12\x62\n\x11ListLoadedPlugins\x12%.cuvis_ai.v1.ListLoadedPluginsRequest\x1a&.cuvis_ai.v1.ListLoadedPluginsResponse\x12V\n\rGetPluginInfo\x12!.cuvis_ai.v1.GetPluginInfoRequest\x1a".cuvis_ai.v1.GetPluginInfoResponse\x12\x65\n\x12ListAvailableNodes\x12&.cuvis_ai.v1.ListAvailableNodesRequest\x1a\'.cuvis_ai.v1.ListAvailableNodesResponse\x12_\n\x10\x43learPluginCache\x12$.cuvis_ai.v1.ClearPluginCacheRequest\x1a%.cuvis_ai.v1.ClearPluginCacheResponseb\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "cuvis_ai.grpc.v1.cuvis_ai_pb2", _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+    DESCRIPTOR._loaded_options = None
+    _globals["_INPUTBATCH_EXTRAINPUTSENTRY"]._loaded_options = None
+    _globals["_INPUTBATCH_EXTRAINPUTSENTRY"]._serialized_options = b"8\001"
+    _globals["_TRAINRESPONSE_LOSSESENTRY"]._loaded_options = None
+    _globals["_TRAINRESPONSE_LOSSESENTRY"]._serialized_options = b"8\001"
+    _globals["_TRAINRESPONSE_METRICSENTRY"]._loaded_options = None
+    _globals["_TRAINRESPONSE_METRICSENTRY"]._serialized_options = b"8\001"
+    _globals["_GETPIPELINEINPUTSRESPONSE_INPUTSPECSENTRY"]._loaded_options = None
+    _globals["_GETPIPELINEINPUTSRESPONSE_INPUTSPECSENTRY"]._serialized_options = b"8\001"
+    _globals["_GETPIPELINEOUTPUTSRESPONSE_OUTPUTSPECSENTRY"]._loaded_options = None
+    _globals["_GETPIPELINEOUTPUTSRESPONSE_OUTPUTSPECSENTRY"]._serialized_options = b"8\001"
+    _globals["_INFERENCERESPONSE_OUTPUTSENTRY"]._loaded_options = None
+    _globals["_INFERENCERESPONSE_OUTPUTSENTRY"]._serialized_options = b"8\001"
+    _globals["_INFERENCERESPONSE_METRICSENTRY"]._loaded_options = None
+    _globals["_INFERENCERESPONSE_METRICSENTRY"]._serialized_options = b"8\001"
+    _globals["_NODEINFO_INPUTSPECSENTRY"]._loaded_options = None
+    _globals["_NODEINFO_INPUTSPECSENTRY"]._serialized_options = b"8\001"
+    _globals["_NODEINFO_OUTPUTSPECSENTRY"]._loaded_options = None
+    _globals["_NODEINFO_OUTPUTSPECSENTRY"]._serialized_options = b"8\001"
+    _globals["_LOADPLUGINSRESPONSE_FAILEDPLUGINSENTRY"]._loaded_options = None
+    _globals["_LOADPLUGINSRESPONSE_FAILEDPLUGINSENTRY"]._serialized_options = b"8\001"
+    _globals["_PROCESSINGMODE"]._serialized_start = 7495
+    _globals["_PROCESSINGMODE"]._serialized_end = 7675
+    _globals["_EXECUTIONSTAGE"]._serialized_start = 7678
+    _globals["_EXECUTIONSTAGE"]._serialized_end = 7836
+    _globals["_DTYPE"]._serialized_start = 7839
+    _globals["_DTYPE"]._serialized_end = 8020
+    _globals["_TRAINERTYPE"]._serialized_start = 8022
+    _globals["_TRAINERTYPE"]._serialized_end = 8122
+    _globals["_TRAINSTATUS"]._serialized_start = 8124
+    _globals["_TRAINSTATUS"]._serialized_end = 8244
+    _globals["_POINTTYPE"]._serialized_start = 8246
+    _globals["_POINTTYPE"]._serialized_end = 8359
+    _globals["_TENSOR"]._serialized_start = 48
+    _globals["_TENSOR"]._serialized_end = 124
+    _globals["_CONTEXT"]._serialized_start = 126
+    _globals["_CONTEXT"]._serialized_end = 234
+    _globals["_PIPELINECONFIG"]._serialized_start = 236
+    _globals["_PIPELINECONFIG"]._serialized_end = 274
+    _globals["_DATACONFIG"]._serialized_start = 276
+    _globals["_DATACONFIG"]._serialized_end = 310
+    _globals["_OPTIMIZERCONFIG"]._serialized_start = 312
+    _globals["_OPTIMIZERCONFIG"]._serialized_end = 351
+    _globals["_SCHEDULERCONFIG"]._serialized_start = 353
+    _globals["_SCHEDULERCONFIG"]._serialized_end = 392
+    _globals["_CALLBACKSCONFIG"]._serialized_start = 394
+    _globals["_CALLBACKSCONFIG"]._serialized_end = 433
+    _globals["_PIPELINEMETADATA"]._serialized_start = 435
+    _globals["_PIPELINEMETADATA"]._serialized_end = 561
+    _globals["_PIPELINEINFO"]._serialized_start = 564
+    _globals["_PIPELINEINFO"]._serialized_end = 734
+    _globals["_TRAININGCONFIG"]._serialized_start = 736
+    _globals["_TRAININGCONFIG"]._serialized_end = 774
+    _globals["_TRAINRUNCONFIG"]._serialized_start = 776
+    _globals["_TRAINRUNCONFIG"]._serialized_end = 814
+    _globals["_BOUNDINGBOX"]._serialized_start = 816
+    _globals["_BOUNDINGBOX"]._serialized_end = 909
+    _globals["_BOUNDINGBOXES"]._serialized_start = 911
+    _globals["_BOUNDINGBOXES"]._serialized_end = 967
+    _globals["_POINT"]._serialized_start = 969
+    _globals["_POINT"]._serialized_end = 1056
+    _globals["_POINTS"]._serialized_start = 1058
+    _globals["_POINTS"]._serialized_end = 1102
+    _globals["_INPUTBATCH"]._serialized_start = 1105
+    _globals["_INPUTBATCH"]._serialized_end = 1468
+    _globals["_INPUTBATCH_EXTRAINPUTSENTRY"]._serialized_start = 1397
+    _globals["_INPUTBATCH_EXTRAINPUTSENTRY"]._serialized_end = 1468
+    _globals["_TENSORSPEC"]._serialized_start = 1470
+    _globals["_TENSORSPEC"]._serialized_end = 1564
+    _globals["_TRAINRESPONSE"]._serialized_start = 1567
+    _globals["_TRAINRESPONSE"]._serialized_end = 1889
+    _globals["_TRAINRESPONSE_LOSSESENTRY"]._serialized_start = 1796
+    _globals["_TRAINRESPONSE_LOSSESENTRY"]._serialized_end = 1841
+    _globals["_TRAINRESPONSE_METRICSENTRY"]._serialized_start = 1843
+    _globals["_TRAINRESPONSE_METRICSENTRY"]._serialized_end = 1889
+    _globals["_PARAMSPEC"]._serialized_start = 1891
+    _globals["_PARAMSPEC"]._serialized_end = 2012
+    _globals["_CALLBACKTYPEINFO"]._serialized_start = 2014
+    _globals["_CALLBACKTYPEINFO"]._serialized_end = 2111
+    _globals["_OPTIMIZERPARAMSSCHEMA"]._serialized_start = 2113
+    _globals["_OPTIMIZERPARAMSSCHEMA"]._serialized_end = 2180
+    _globals["_SCHEDULERPARAMSSCHEMA"]._serialized_start = 2182
+    _globals["_SCHEDULERPARAMSSCHEMA"]._serialized_end = 2249
+    _globals["_LISTAVAILABLEPIPELINEESREQUEST"]._serialized_start = 2251
+    _globals["_LISTAVAILABLEPIPELINEESREQUEST"]._serialized_end = 2323
+    _globals["_LISTAVAILABLEPIPELINEESRESPONSE"]._serialized_start = 2325
+    _globals["_LISTAVAILABLEPIPELINEESRESPONSE"]._serialized_end = 2405
+    _globals["_GETPIPELINEINFOREQUEST"]._serialized_start = 2407
+    _globals["_GETPIPELINEINFOREQUEST"]._serialized_end = 2454
+    _globals["_GETPIPELINEINFORESPONSE"]._serialized_start = 2456
+    _globals["_GETPIPELINEINFORESPONSE"]._serialized_end = 2531
+    _globals["_CREATESESSIONREQUEST"]._serialized_start = 2533
+    _globals["_CREATESESSIONREQUEST"]._serialized_end = 2555
+    _globals["_CREATESESSIONRESPONSE"]._serialized_start = 2557
+    _globals["_CREATESESSIONRESPONSE"]._serialized_end = 2600
+    _globals["_SETSESSIONSEARCHPATHSREQUEST"]._serialized_start = 2602
+    _globals["_SETSESSIONSEARCHPATHSREQUEST"]._serialized_end = 2706
+    _globals["_SETSESSIONSEARCHPATHSRESPONSE"]._serialized_start = 2708
+    _globals["_SETSESSIONSEARCHPATHSRESPONSE"]._serialized_end = 2803
+    _globals["_CLOSESESSIONREQUEST"]._serialized_start = 2805
+    _globals["_CLOSESESSIONREQUEST"]._serialized_end = 2846
+    _globals["_CLOSESESSIONRESPONSE"]._serialized_start = 2848
+    _globals["_CLOSESESSIONRESPONSE"]._serialized_end = 2887
+    _globals["_RESOLVECONFIGREQUEST"]._serialized_start = 2889
+    _globals["_RESOLVECONFIGREQUEST"]._serialized_end = 2985
+    _globals["_RESOLVECONFIGRESPONSE"]._serialized_start = 2987
+    _globals["_RESOLVECONFIGRESPONSE"]._serialized_end = 3032
+    _globals["_GETPARAMETERSCHEMAREQUEST"]._serialized_start = 3034
+    _globals["_GETPARAMETERSCHEMAREQUEST"]._serialized_end = 3082
+    _globals["_GETPARAMETERSCHEMARESPONSE"]._serialized_start = 3084
+    _globals["_GETPARAMETERSCHEMARESPONSE"]._serialized_end = 3133
+    _globals["_VALIDATECONFIGREQUEST"]._serialized_start = 3135
+    _globals["_VALIDATECONFIGREQUEST"]._serialized_end = 3201
+    _globals["_VALIDATECONFIGRESPONSE"]._serialized_start = 3203
+    _globals["_VALIDATECONFIGRESPONSE"]._serialized_end = 3276
+    _globals["_LOADPIPELINEWEIGHTSREQUEST"]._serialized_start = 3279
+    _globals["_LOADPIPELINEWEIGHTSREQUEST"]._serialized_end = 3426
+    _globals["_LOADPIPELINEWEIGHTSRESPONSE"]._serialized_start = 3428
+    _globals["_LOADPIPELINEWEIGHTSRESPONSE"]._serialized_end = 3497
+    _globals["_SETTRAINRUNCONFIGREQUEST"]._serialized_start = 3499
+    _globals["_SETTRAINRUNCONFIGREQUEST"]._serialized_end = 3590
+    _globals["_SETTRAINRUNCONFIGRESPONSE"]._serialized_start = 3592
+    _globals["_SETTRAINRUNCONFIGRESPONSE"]._serialized_end = 3666
+    _globals["_TRAINREQUEST"]._serialized_start = 3669
+    _globals["_TRAINREQUEST"]._serialized_end = 3837
+    _globals["_GETTRAINSTATUSREQUEST"]._serialized_start = 3839
+    _globals["_GETTRAINSTATUSREQUEST"]._serialized_end = 3882
+    _globals["_GETTRAINSTATUSRESPONSE"]._serialized_start = 3884
+    _globals["_GETTRAINSTATUSRESPONSE"]._serialized_end = 3961
+    _globals["_GETTRAININGCAPABILITIESREQUEST"]._serialized_start = 3963
+    _globals["_GETTRAININGCAPABILITIESREQUEST"]._serialized_end = 3995
+    _globals["_GETTRAININGCAPABILITIESRESPONSE"]._serialized_start = 3998
+    _globals["_GETTRAININGCAPABILITIESRESPONSE"]._serialized_end = 4275
+    _globals["_SAVEPIPELINEREQUEST"]._serialized_start = 4277
+    _globals["_SAVEPIPELINEREQUEST"]._serialized_end = 4390
+    _globals["_SAVEPIPELINERESPONSE"]._serialized_start = 4392
+    _globals["_SAVEPIPELINERESPONSE"]._serialized_end = 4476
+    _globals["_LOADPIPELINEREQUEST"]._serialized_start = 4478
+    _globals["_LOADPIPELINEREQUEST"]._serialized_end = 4566
+    _globals["_LOADPIPELINERESPONSE"]._serialized_start = 4568
+    _globals["_LOADPIPELINERESPONSE"]._serialized_end = 4656
+    _globals["_SAVETRAINRUNREQUEST"]._serialized_start = 4658
+    _globals["_SAVETRAINRUNREQUEST"]._serialized_end = 4744
+    _globals["_SAVETRAINRUNRESPONSE"]._serialized_start = 4746
+    _globals["_SAVETRAINRUNRESPONSE"]._serialized_end = 4853
+    _globals["_RESTORETRAINRUNREQUEST"]._serialized_start = 4855
+    _globals["_RESTORETRAINRUNREQUEST"]._serialized_end = 4978
+    _globals["_RESTORETRAINRUNRESPONSE"]._serialized_start = 4980
+    _globals["_RESTORETRAINRUNRESPONSE"]._serialized_end = 5072
+    _globals["_GETPIPELINEINPUTSREQUEST"]._serialized_start = 5074
+    _globals["_GETPIPELINEINPUTSREQUEST"]._serialized_end = 5120
+    _globals["_GETPIPELINEINPUTSRESPONSE"]._serialized_start = 5123
+    _globals["_GETPIPELINEINPUTSRESPONSE"]._serialized_end = 5324
+    _globals["_GETPIPELINEINPUTSRESPONSE_INPUTSPECSENTRY"]._serialized_start = 5250
+    _globals["_GETPIPELINEINPUTSRESPONSE_INPUTSPECSENTRY"]._serialized_end = 5324
+    _globals["_GETPIPELINEOUTPUTSREQUEST"]._serialized_start = 5326
+    _globals["_GETPIPELINEOUTPUTSREQUEST"]._serialized_end = 5373
+    _globals["_GETPIPELINEOUTPUTSRESPONSE"]._serialized_start = 5376
+    _globals["_GETPIPELINEOUTPUTSRESPONSE"]._serialized_end = 5583
+    _globals["_GETPIPELINEOUTPUTSRESPONSE_OUTPUTSPECSENTRY"]._serialized_start = 5508
+    _globals["_GETPIPELINEOUTPUTSRESPONSE_OUTPUTSPECSENTRY"]._serialized_end = 5583
+    _globals["_GETPIPELINEVISUALIZATIONREQUEST"]._serialized_start = 5585
+    _globals["_GETPIPELINEVISUALIZATIONREQUEST"]._serialized_end = 5654
+    _globals["_GETPIPELINEVISUALIZATIONRESPONSE"]._serialized_start = 5656
+    _globals["_GETPIPELINEVISUALIZATIONRESPONSE"]._serialized_end = 5726
+    _globals["_INFERENCEREQUEST"]._serialized_start = 5728
+    _globals["_INFERENCEREQUEST"]._serialized_end = 5829
+    _globals["_INFERENCERESPONSE"]._serialized_start = 5832
+    _globals["_INFERENCERESPONSE"]._serialized_end = 6092
+    _globals["_INFERENCERESPONSE_OUTPUTSENTRY"]._serialized_start = 5977
+    _globals["_INFERENCERESPONSE_OUTPUTSENTRY"]._serialized_end = 6044
+    _globals["_INFERENCERESPONSE_METRICSENTRY"]._serialized_start = 1843
+    _globals["_INFERENCERESPONSE_METRICSENTRY"]._serialized_end = 1889
+    _globals["_PLUGINMANIFEST"]._serialized_start = 6094
+    _globals["_PLUGINMANIFEST"]._serialized_end = 6132
+    _globals["_PLUGININFO"]._serialized_start = 6134
+    _globals["_PLUGININFO"]._serialized_end = 6221
+    _globals["_PORTSPEC"]._serialized_start = 6223
+    _globals["_PORTSPEC"]._serialized_end = 6336
+    _globals["_PORTSPECLIST"]._serialized_start = 6338
+    _globals["_PORTSPECLIST"]._serialized_end = 6390
+    _globals["_NODEINFO"]._serialized_start = 6393
+    _globals["_NODEINFO"]._serialized_end = 6758
+    _globals["_NODEINFO_INPUTSPECSENTRY"]._serialized_start = 6603
+    _globals["_NODEINFO_INPUTSPECSENTRY"]._serialized_end = 6679
+    _globals["_NODEINFO_OUTPUTSPECSENTRY"]._serialized_start = 6681
+    _globals["_NODEINFO_OUTPUTSPECSENTRY"]._serialized_end = 6758
+    _globals["_LOADPLUGINSREQUEST"]._serialized_start = 6760
+    _globals["_LOADPLUGINSREQUEST"]._serialized_end = 6847
+    _globals["_LOADPLUGINSRESPONSE"]._serialized_start = 6850
+    _globals["_LOADPLUGINSRESPONSE"]._serialized_end = 7026
+    _globals["_LOADPLUGINSRESPONSE_FAILEDPLUGINSENTRY"]._serialized_start = 6974
+    _globals["_LOADPLUGINSRESPONSE_FAILEDPLUGINSENTRY"]._serialized_end = 7026
+    _globals["_LISTLOADEDPLUGINSREQUEST"]._serialized_start = 7028
+    _globals["_LISTLOADEDPLUGINSREQUEST"]._serialized_end = 7074
+    _globals["_LISTLOADEDPLUGINSRESPONSE"]._serialized_start = 7076
+    _globals["_LISTLOADEDPLUGINSRESPONSE"]._serialized_end = 7145
+    _globals["_GETPLUGININFOREQUEST"]._serialized_start = 7147
+    _globals["_GETPLUGININFOREQUEST"]._serialized_end = 7210
+    _globals["_GETPLUGININFORESPONSE"]._serialized_start = 7212
+    _globals["_GETPLUGININFORESPONSE"]._serialized_end = 7276
+    _globals["_LISTAVAILABLENODESREQUEST"]._serialized_start = 7278
+    _globals["_LISTAVAILABLENODESREQUEST"]._serialized_end = 7325
+    _globals["_LISTAVAILABLENODESRESPONSE"]._serialized_start = 7327
+    _globals["_LISTAVAILABLENODESRESPONSE"]._serialized_end = 7393
+    _globals["_CLEARPLUGINCACHEREQUEST"]._serialized_start = 7395
+    _globals["_CLEARPLUGINCACHEREQUEST"]._serialized_end = 7441
+    _globals["_CLEARPLUGINCACHERESPONSE"]._serialized_start = 7443
+    _globals["_CLEARPLUGINCACHERESPONSE"]._serialized_end = 7492
+    _globals["_CUVISAISERVICE"]._serialized_start = 8362
+    _globals["_CUVISAISERVICE"]._serialized_end = 10857
+# @@protoc_insertion_point(module_scope)