lumen-app 0.4.2__py3-none-any.whl

lumen_app/core/tests/test_env_checker.py

@@ -0,0 +1,487 @@
+"""
+Tests for EnvironmentChecker utility.
+"""
+
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+import pytest
+
+from lumen_app.core.config import DeviceConfig
+from lumen_app.utils.env_checker import (
+    DependencyInstaller,
+    DriverChecker,
+    DriverStatus,
+    EnvironmentChecker,
+    EnvironmentReport,
+    MicromambaChecker,
+)
+
+# =============================================================================
+# DriverChecker Tests
+# =============================================================================
+
+
+def test_check_nvidia_gpu_available():
+    """Test NVIDIA GPU check when nvidia-smi succeeds."""
+    with patch("subprocess.run") as mock_run:
+        mock_result = MagicMock()
+        mock_result.returncode = 0
+        mock_result.stdout = "CUDA Version: 12.6"
+        mock_run.return_value = mock_result
+
+        result = DriverChecker.check_nvidia_gpu()
+
+        assert result.status == DriverStatus.AVAILABLE
+        assert result.name == "CUDA"
+        assert result.installable_via_mamba is True
+
+
+def test_check_nvidia_gpu_missing():
+    """Test NVIDIA GPU check when nvidia-smi fails."""
+    with patch("subprocess.run") as mock_run:
+        mock_run.side_effect = FileNotFoundError()
+
+        result = DriverChecker.check_nvidia_gpu()
+
+        assert result.status == DriverStatus.MISSING
+        assert result.name == "CUDA"
+
+
+def test_check_amd_ryzen_ai_npu_not_windows():
+    """Test AMD Ryzen AI NPU check on non-Windows."""
+    with patch("platform.system", return_value="Linux"):
+        result = DriverChecker.check_amd_ryzen_ai_npu()
+
+        assert result.status == DriverStatus.MISSING
+        assert "Only available on Windows" in result.details
+
+
+def test_check_amd_ryzen_ai_npu_dll_missing():
+    """Test AMD Ryzen AI NPU check when DLL is missing."""
+    with patch("platform.system", return_value="Windows"):
+        with patch("pathlib.Path.exists", return_value=False):
+            result = DriverChecker.check_amd_ryzen_ai_npu()
+
+            assert result.status == DriverStatus.MISSING
+            assert "amdipu.dll not found" in result.details
+
+
+def test_check_amd_ryzen_ai_npu_service_running():
+    """Test AMD Ryzen AI NPU check when service is running."""
+    with patch("platform.system", return_value="Windows"):
+        with patch("pathlib.Path.exists", return_value=True):
+            with patch("subprocess.run") as mock_run:
+                mock_result = MagicMock()
+                mock_result.returncode = 0
+                mock_result.stdout = "STATE: RUNNING"
+                mock_run.return_value = mock_result
+
+                result = DriverChecker.check_amd_ryzen_ai_npu()
+
+                assert result.status == DriverStatus.AVAILABLE
+                assert "amdipu service is running" in result.details
+
+
+# =============================================================================
+# MicromambaChecker Tests
+# =============================================================================
+
+
+def test_check_micromamba_available():
+    """Test micromamba check when installed."""
+    with patch("subprocess.run") as mock_run:
+        mock_result = MagicMock()
+        mock_result.returncode = 0
+        mock_result.stdout = "micromamba 1.5.3\n"
+        mock_run.return_value = mock_result
+
+        result = MicromambaChecker.check_micromamba()
+
+        assert result.status == DriverStatus.AVAILABLE
+        assert result.name == "micromamba"
+        assert "1.5.3" in result.details
+
+
+def test_check_micromamba_missing():
+    """Test micromamba check when not installed."""
+    with patch("subprocess.run") as mock_run:
+        mock_run.side_effect = FileNotFoundError()
+
+        result = MicromambaChecker.check_micromamba()
+
+        assert result.status == DriverStatus.MISSING
+        assert result.name == "micromamba"
+        assert "not found" in result.details
+
+
+def test_micromamba_install_to_cache_dir():
+    """Test micromamba installation to cache directory."""
+    with patch("subprocess.run") as mock_run:
+        # Mock successful installation
+        mock_result = MagicMock()
+        mock_result.returncode = 0
+        mock_run.return_value = mock_result
+
+        with patch("pathlib.Path.exists", return_value=False):
+            with patch("pathlib.Path.mkdir"):  # Mock mkdir to avoid creating dirs
+                success, message = MicromambaChecker.install_micromamba(
+                    cache_dir="/test/cache", dry_run=True
+                )
+
+            assert success is True
+            assert "Would run" in message
+
+
+def test_micromamba_get_executable_path():
+    """Test getting micromamba executable path."""
+    with patch("platform.system", return_value="Darwin"):
+        path = MicromambaChecker.get_executable_path("/test/cache")
+        assert "bin/micromamba" in path
+
+    with patch("platform.system", return_value="Windows"):
+        path = MicromambaChecker.get_executable_path("/test/cache")
+        assert "bin/micromamba.exe" in path
+
+
+# =============================================================================
+# DriverChecker Tests
+# =============================================================================
+
+
+def test_check_intel_gpu_openvino_genuine_intel():
+    """Test Intel GPU/OpenVINO check."""
+    result = DriverChecker.check_intel_gpu_openvino()
+
+    # Should return OpenVINO check result
+    assert result.name == "OpenVINO"
+    # Status depends on platform (INCOMPATIBLE on Apple Silicon, MISSING on Intel without package, etc.)
+    assert isinstance(result.status, DriverStatus)
+
+
+def test_check_intel_gpu_openvino_not_intel():
+    """Test Intel GPU/OpenVINO with non-Intel CPU."""
+    with patch("platform.system", return_value="Linux"):
+        with patch("builtins.open", create=True) as mock_open:
+            mock_open.return_value.__enter__ = MagicMock()
+            mock_open.return_value.__exit__ = MagicMock()
+            mock_open.return_value.read.return_value = "vendor_id\t: AuthenticAMD\n"
+
+            result = DriverChecker.check_intel_gpu_openvino()
+
+            assert result.status == DriverStatus.INCOMPATIBLE
+            assert "not Intel" in result.details
+
+
+def test_check_apple_silicon_not_macos():
+    """Test Apple Silicon check on non-macOS."""
+    with patch("platform.system", return_value="Linux"):
+        result = DriverChecker.check_apple_silicon()
+
+        assert result.status == DriverStatus.INCOMPATIBLE
+        assert "Only available on macOS" in result.details
+
+
+def test_check_apple_silicon_not_arm64():
+    """Test Apple Silicon check on x86_64 macOS."""
+    with patch("platform.system", return_value="Darwin"):
+        with patch("platform.machine", return_value="x86_64"):
+            result = DriverChecker.check_apple_silicon()
+
+            assert result.status == DriverStatus.INCOMPATIBLE
+            assert "not arm64" in result.details
+
+
+def test_check_apple_silicon_available():
+    """Test Apple Silicon check on M1/M2/M3."""
+    with patch("platform.system", return_value="Darwin"):
+        with patch("platform.machine", return_value="arm64"):
+            with patch("subprocess.run") as mock_run:
+                mock_result = MagicMock()
+                mock_result.returncode = 0
+                mock_result.stdout = "Apple M2"
+                mock_run.return_value = mock_result
+
+                result = DriverChecker.check_apple_silicon()
+
+                assert result.status == DriverStatus.AVAILABLE
+                assert "Apple" in result.details
+
+
+def test_check_rockchip_rknn_not_linux():
+    """Test RKNN check on non-Linux."""
+    with patch("platform.system", return_value="Windows"):
+        result = DriverChecker.check_rockchip_rknn()
+
+        assert result.status == DriverStatus.INCOMPATIBLE
+        assert "Only available on Linux" in result.details
+
+
+def test_check_rockchip_rknn_device_found():
+    """Test RKNN check when device node exists."""
+    with patch("platform.system", return_value="Linux"):
+        # Mock Path.exists to return True for /dev/rknpu
+        original_exists = Path.exists
+
+        def mock_exists(self):
+            return str(self) == "/dev/rknpu" or original_exists(self)
+
+        with patch.object(Path, "exists", mock_exists):
+            result = DriverChecker.check_rockchip_rknn()
+
+            assert result.status == DriverStatus.AVAILABLE
+            assert "Device found" in result.details
+
+
+def test_check_rockchip_rknn_missing():
+    """Test RKNN check when device not found."""
+    with patch("platform.system", return_value="Linux"):
+        with patch("pathlib.Path.exists", return_value=False):
+            with patch("builtins.open", create=True) as mock_open:
+                mock_open.return_value.__enter__ = MagicMock()
+                mock_open.return_value.__exit__ = MagicMock()
+                mock_open.return_value.read.return_value = "vendor_id\t: GenuineIntel\n"
+
+                result = DriverChecker.check_rockchip_rknn()
+
+                assert result.status == DriverStatus.MISSING
+
+
+def test_check_amd_gpu_directml_not_windows():
+    """Test AMD GPU DirectML check on non-Windows."""
+    with patch("platform.system", return_value="Linux"):
+        result = DriverChecker.check_amd_gpu_directml()
+
+        assert result.status == DriverStatus.INCOMPATIBLE
+        assert "Only available on Windows" in result.details
+
+
+def test_check_amd_gpu_directml_radeon_found():
+    """Test AMD GPU DirectML check when Radeon GPU found."""
+    with patch("platform.system", return_value="Windows"):
+        with patch("subprocess.run") as mock_run:
+            mock_result = MagicMock()
+            mock_result.returncode = 0
+            mock_result.stdout = "AMD Radeon RX 7900 XTX"
+            mock_run.return_value = mock_result
+
+            result = DriverChecker.check_amd_gpu_directml()
+
+            assert result.status == DriverStatus.AVAILABLE
+            assert "Radeon" in result.details
+
+
+def test_check_for_preset_nvidia():
+    """Test checking drivers for NVIDIA GPU preset."""
+    with patch("subprocess.run") as mock_run:
+        mock_run.side_effect = FileNotFoundError()
+
+        results = DriverChecker.check_for_preset("nvidia_gpu")
+
+        assert len(results) == 1
+        assert results[0].name == "CUDA"
+
+
+def test_check_for_preset_cpu():
+    """Test checking drivers for CPU preset (no special drivers)."""
+    results = DriverChecker.check_for_preset("cpu")
+
+    assert len(results) == 0
+
+
+def test_check_for_preset_apple_silicon():
+    """Test checking drivers for Apple Silicon preset."""
+    with patch("platform.system", return_value="Darwin"):
+        with patch("platform.machine", return_value="arm64"):
+            with patch("subprocess.run") as mock_run:
+                mock_result = MagicMock()
+                mock_result.returncode = 0
+                mock_result.stdout = "Apple M2"
+                mock_run.return_value = mock_result
+
+                results = DriverChecker.check_for_preset("apple_silicon")
+
+                assert len(results) == 1
+                assert results[0].name == "CoreML"
+
+
+def test_check_for_device_config_cpu():
+    """Test checking drivers for CPU DeviceConfig."""
+    config = DeviceConfig.cpu()
+    results = DriverChecker.check_for_device_config(config)
+
+    assert len(results) == 0
+
+
+def test_check_for_device_config_nvidia():
+    """Test checking drivers for NVIDIA DeviceConfig."""
+    config = DeviceConfig.nvidia_gpu()
+
+    with patch("subprocess.run") as mock_run:
+        mock_run.side_effect = FileNotFoundError()
+
+        results = DriverChecker.check_for_device_config(config)
+
+        assert len(results) >= 1
+        assert any(r.name == "CUDA" for r in results)
+
+
+# =============================================================================
+# DependencyInstaller Tests
+# =============================================================================
+
+
+def test_installer_init_default_path():
+    """Test installer initialization with default path."""
+    installer = DependencyInstaller()
+
+    assert installer.configs_dir is not None
+    assert installer.configs_dir.name == "mamba"
+
+
+def test_installer_get_install_command_cuda():
+    """Test getting install command for CUDA."""
+    with patch("pathlib.Path.exists", return_value=True):
+        installer = DependencyInstaller(mamba_configs_dir="/fake/path")
+
+        cmd = installer.get_install_command("cuda")
+
+        assert "micromamba" in cmd
+        assert "cuda" in cmd.lower()
+
+
+def test_installer_install_dry_run():
+    """Test installer dry run mode."""
+    with patch("pathlib.Path.exists", return_value=True):
+        installer = DependencyInstaller(mamba_configs_dir="/fake/path")
+
+        success, message = installer.install_driver("cuda", dry_run=True)
+
+        assert success is True
+        assert "Would run" in message
+
+
+def test_installer_install_unsupported_driver():
+    """Test installer with unsupported driver."""
+    installer = DependencyInstaller()
+
+    success, message = installer.install_driver("unsupported_driver")
+
+    assert success is False
+    assert "No mamba config available" in message
+
+
+# =============================================================================
+# EnvironmentChecker Tests
+# =============================================================================
+
+
+def test_check_preset_nvidia_missing():
+    """Test environment check for NVIDIA preset with missing driver."""
+    with patch("subprocess.run") as mock_run:
+        mock_run.side_effect = FileNotFoundError()
+
+        report = EnvironmentChecker.check_preset("nvidia_gpu")
+
+        assert isinstance(report, EnvironmentReport)
+        assert report.preset_name == "nvidia_gpu"
+        assert report.ready is False
+        # missing_installable now contains lowercase names
+        assert "cuda" in report.missing_installable
+
+
+def test_check_preset_cpu_ready():
+    """Test environment check for CPU preset (always ready)."""
+    report = EnvironmentChecker.check_preset("cpu")
+
+    assert isinstance(report, EnvironmentReport)
+    assert report.preset_name == "cpu"
+    assert report.ready is True
+    assert len(report.drivers) == 0
+
+
+def test_check_device_config_cpu():
+    """Test environment check for CPU DeviceConfig."""
+    config = DeviceConfig.cpu()
+    report = EnvironmentChecker.check_device_config(config)
+
+    assert isinstance(report, EnvironmentReport)
+    assert report.ready is True
+    assert len(report.drivers) == 0
+
+
+def test_check_device_config_apple_silicon():
+    """Test environment check for Apple Silicon DeviceConfig."""
+    config = DeviceConfig.apple_silicon()
+
+    with patch("platform.system", return_value="Darwin"):
+        with patch("platform.machine", return_value="arm64"):
+            with patch("subprocess.run") as mock_run:
+                mock_result = MagicMock()
+                mock_result.returncode = 0
+                mock_result.stdout = "Apple M2"
+                mock_run.return_value = mock_result
+
+                report = EnvironmentChecker.check_device_config(config)
+
+                assert isinstance(report, EnvironmentReport)
+                assert report.ready is True
+
+
+# =============================================================================
+# Integration Tests
+# =============================================================================
+
+
+def test_full_workflow_check_and_install():
+    """Test full workflow: check preset, get install commands."""
+    with patch("subprocess.run") as mock_run:
+        mock_run.side_effect = FileNotFoundError()
+
+        # 1. Check preset
+        report = EnvironmentChecker.check_preset("nvidia_gpu")
+
+        assert report.ready is False
+        assert len(report.missing_installable) > 0
+
+        # 2. Get install command (driver names are uppercase in result)
+        installer = DependencyInstaller()
+        for driver_name in report.missing_installable:
+            # Convert to lowercase for installer
+            cmd = installer.get_install_command(driver_name.lower())
+            # Should have valid command or error message
+            assert isinstance(cmd, str)
+            assert len(cmd) > 0
+
+
+def test_environment_report_all_available():
+    """Test EnvironmentReport when all drivers available."""
+    report = EnvironmentReport(preset_name="cpu", drivers=[], ready=True)
+
+    assert report.ready is True
+
+
+def test_environment_report_missing_drivers():
+    """Test EnvironmentReport with missing drivers."""
+    from lumen_app.utils.env_checker import DriverCheckResult
+
+    report = EnvironmentReport(
+        preset_name="nvidia_gpu",
+        drivers=[
+            DriverCheckResult(
+                name="CUDA",
+                status=DriverStatus.MISSING,
+                installable_via_mamba=True,
+                mamba_config_path="cuda.yaml",
+            )
+        ],
+        ready=False,
+        missing_installable=["CUDA"],
+    )
+
+    assert report.ready is False
+    assert "CUDA" in report.missing_installable
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])

lumen_app/proto/README.md

@@ -0,0 +1,12 @@
+## Proto Code Gen
+
+Before run the command, please make sure you're right in `Lumen/lumen-*/` directory.
+
+```bash
+python -m grpc_tools.protoc \
+    -I src/lumen_clip \
+    --python_out=src/lumen_clip \
+    --grpc_python_out=src/lumen_clip \
+    --pyi_out=src/lumen_clip \
+    src/lumen_clip/proto/ml_service.proto
+```

lumen_app/proto/ml_service.proto

@@ -0,0 +1,88 @@
+syntax = "proto3";
+
+package home_native.v1;
+
+option go_package = "server/proto";
+
+import "google/protobuf/empty.proto";
+
+// ---- Unified error codes (concise) ----
+enum ErrorCode {
+  ERROR_CODE_UNSPECIFIED = 0;
+  ERROR_CODE_INVALID_ARGUMENT = 1;
+  ERROR_CODE_UNAVAILABLE = 2;      // Downstream/hardware/resource unavailable
+  ERROR_CODE_DEADLINE_EXCEEDED = 3;
+  ERROR_CODE_INTERNAL = 4;
+}
+
+// ---- Standard error payload (used with gRPC status; fatal errors may terminate the stream) ----
+message Error {
+  ErrorCode code = 1;
+  string    message = 2;
+  string    detail = 3;             // For logging/troubleshooting (stack, node ID, etc.)
+}
+
+// ---- Structured task I/O description (for central routing and client negotiation) ----
+message IOTask {
+  string name = 1;                           // "embed","detect","ocr","asr","generate","tts",...
+  repeated string input_mimes = 2;           // Allow multiple input types: "image/jpeg","audio/pcm;rate=16000","application/json"
+  repeated string output_mimes = 3;          // Typical outputs: "application/json;schema=bbox_v1","audio/wav"
+  map<string, string> limits = 4;            // e.g., max_hw=1024, max_batch=8, max_length=4096
+}
+
+// ---- Capability declaration (retrieve at startup or on demand) ----
+message Capability {
+  string service_name = 1;                   // "clip-embedder","ocr","llm","tts",...
+  repeated string model_ids = 2;             // Supported model IDs/versions
+  string runtime = 3;                        // "onnxrt-cuda","tensorrt","coreml","rknn","qnn","cpu"
+  uint32 max_concurrency = 4;                // Suggested max concurrency
+  repeated string precisions = 5;            // ["fp32","fp16","int8"]
+  map<string, string> extra = 6;             // Resolution limits, ANE/NPU features, etc.
+  repeated IOTask tasks = 7;                 // Structured task capabilities (recommended)
+  string protocol_version = 8;               // Version of current porotocal, should be semantic versioning. e.g., 1.0.0
+}
+
+// ---- Inference request (inbound message of bidi stream) ----
+message InferRequest {
+  string correlation_id = 1;                 // Trace/correlation
+  string task = 2;                           // "embed","classify","detect","ocr","asr","generate","tts",...
+  bytes  payload = 3;                        // Raw payload (binary or UTF-8 text)
+  map<string, string> meta = 4;              // Task-specific parameters: model_id, conf_thres, stop, etc.
+
+  // --- Added: input content type and chunking control ---
+  string payload_mime = 5;                   // e.g. "image/jpeg","audio/pcm;rate=16000","application/json"
+  uint64 seq = 6;                            // Chunk index (starting at 0)
+  uint64 total = 7;                          // Optional: total number of chunks (omit if unknown)
+  uint64 offset = 8;                         // Optional: byte offset of payload in the overall stream
+}
+
+// ---- Inference response (outbound message of bidi stream) ----
+message InferResponse {
+  string correlation_id = 1;
+  bool   is_final = 2;                       // Streaming partials or final result
+  bytes  result = 3;                         // Result bytes; if JSON, specify via result_mime
+  map<string, string> meta = 4;              // e.g., lat_ms, tokens, bboxes_count
+  Error  error = 5;                          // Populated only on failure (fatal errors may also terminate the stream)
+
+  // --- Added: output content type and chunking control ---
+  uint64 seq = 6;                            // Chunk index (starting at 0), used for incremental/audio/video chunks
+  uint64 total = 7;                          // Optional: total number of chunks
+  uint64 offset = 8;                         // Optional: byte offset of result in the overall output
+  string result_mime = 9;                    // e.g. "application/json;schema=embedding_v1","audio/wav","image/png"
+  string result_schema = 10;                 // Optional: explicit schema name (e.g., "bbox_v1","mask_rle_v1")
+}
+
+// ---- Service contract ----
+service Inference {
+  // Bidirectional stream: client sends chunks; server returns incremental/final results; ordered but non-blocking
+  rpc Infer(stream InferRequest) returns (stream InferResponse);
+
+  // Capability declaration (backward compatibility: single capability; use StreamCapabilities for multiple)
+  rpc GetCapabilities(google.protobuf.Empty) returns (Capability);
+
+  // Recommended: server stream returns all capabilities (call at startup or after hot-reload)
+  rpc StreamCapabilities(google.protobuf.Empty) returns (stream Capability);
+
+  // Health probe
+  rpc Health(google.protobuf.Empty) returns (google.protobuf.Empty);
+}

lumen_app/proto/ml_service_pb2.py

@@ -0,0 +1,66 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: proto/ml_service.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,
+    '',
+    'proto/ml_service.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import empty_pb2 as google_dot_protobuf_dot_empty__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x16proto/ml_service.proto\x12\x0ehome_native.v1\x1a\x1bgoogle/protobuf/empty.proto\"Q\n\x05\x45rror\x12\'\n\x04\x63ode\x18\x01 \x01(\x0e\x32\x19.home_native.v1.ErrorCode\x12\x0f\n\x07message\x18\x02 \x01(\t\x12\x0e\n\x06\x64\x65tail\x18\x03 \x01(\t\"\xa4\x01\n\x06IOTask\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x13\n\x0binput_mimes\x18\x02 \x03(\t\x12\x14\n\x0coutput_mimes\x18\x03 \x03(\t\x12\x32\n\x06limits\x18\x04 \x03(\x0b\x32\".home_native.v1.IOTask.LimitsEntry\x1a-\n\x0bLimitsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x98\x02\n\nCapability\x12\x14\n\x0cservice_name\x18\x01 \x01(\t\x12\x11\n\tmodel_ids\x18\x02 \x03(\t\x12\x0f\n\x07runtime\x18\x03 \x01(\t\x12\x17\n\x0fmax_concurrency\x18\x04 \x01(\r\x12\x12\n\nprecisions\x18\x05 \x03(\t\x12\x34\n\x05\x65xtra\x18\x06 \x03(\x0b\x32%.home_native.v1.Capability.ExtraEntry\x12%\n\x05tasks\x18\x07 \x03(\x0b\x32\x16.home_native.v1.IOTask\x12\x18\n\x10protocol_version\x18\x08 \x01(\t\x1a,\n\nExtraEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xea\x01\n\x0cInferRequest\x12\x16\n\x0e\x63orrelation_id\x18\x01 \x01(\t\x12\x0c\n\x04task\x18\x02 \x01(\t\x12\x0f\n\x07payload\x18\x03 \x01(\x0c\x12\x34\n\x04meta\x18\x04 \x03(\x0b\x32&.home_native.v1.InferRequest.MetaEntry\x12\x14\n\x0cpayload_mime\x18\x05 \x01(\t\x12\x0b\n\x03seq\x18\x06 \x01(\x04\x12\r\n\x05total\x18\x07 \x01(\x04\x12\x0e\n\x06offset\x18\x08 \x01(\x04\x1a+\n\tMetaEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\xab\x02\n\rInferResponse\x12\x16\n\x0e\x63orrelation_id\x18\x01 \x01(\t\x12\x10\n\x08is_final\x18\x02 \x01(\x08\x12\x0e\n\x06result\x18\x03 \x01(\x0c\x12\x35\n\x04meta\x18\x04 \x03(\x0b\x32\'.home_native.v1.InferResponse.MetaEntry\x12$\n\x05\x65rror\x18\x05 \x01(\x0b\x32\x15.home_native.v1.Error\x12\x0b\n\x03seq\x18\x06 \x01(\x04\x12\r\n\x05total\x18\x07 \x01(\x04\x12\x0e\n\x06offset\x18\x08 \x01(\x04\x12\x13\n\x0bresult_mime\x18\t \x01(\t\x12\x15\n\rresult_schema\x18\n \x01(\t\x1a+\n\tMetaEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01*\x9f\x01\n\tErrorCode\x12\x1a\n\x16\x45RROR_CODE_UNSPECIFIED\x10\x00\x12\x1f\n\x1b\x45RROR_CODE_INVALID_ARGUMENT\x10\x01\x12\x1a\n\x16\x45RROR_CODE_UNAVAILABLE\x10\x02\x12 \n\x1c\x45RROR_CODE_DEADLINE_EXCEEDED\x10\x03\x12\x17\n\x13\x45RROR_CODE_INTERNAL\x10\x04\x32\xa2\x02\n\tInference\x12H\n\x05Infer\x12\x1c.home_native.v1.InferRequest\x1a\x1d.home_native.v1.InferResponse(\x01\x30\x01\x12\x45\n\x0fGetCapabilities\x12\x16.google.protobuf.Empty\x1a\x1a.home_native.v1.Capability\x12J\n\x12StreamCapabilities\x12\x16.google.protobuf.Empty\x1a\x1a.home_native.v1.Capability0\x01\x12\x38\n\x06Health\x12\x16.google.protobuf.Empty\x1a\x16.google.protobuf.EmptyB\x0eZ\x0cserver/protob\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'proto.ml_service_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\014server/proto'
+  _globals['_IOTASK_LIMITSENTRY']._loaded_options = None
+  _globals['_IOTASK_LIMITSENTRY']._serialized_options = b'8\001'
+  _globals['_CAPABILITY_EXTRAENTRY']._loaded_options = None
+  _globals['_CAPABILITY_EXTRAENTRY']._serialized_options = b'8\001'
+  _globals['_INFERREQUEST_METAENTRY']._loaded_options = None
+  _globals['_INFERREQUEST_METAENTRY']._serialized_options = b'8\001'
+  _globals['_INFERRESPONSE_METAENTRY']._loaded_options = None
+  _globals['_INFERRESPONSE_METAENTRY']._serialized_options = b'8\001'
+  _globals['_ERRORCODE']._serialized_start=1144
+  _globals['_ERRORCODE']._serialized_end=1303
+  _globals['_ERROR']._serialized_start=71
+  _globals['_ERROR']._serialized_end=152
+  _globals['_IOTASK']._serialized_start=155
+  _globals['_IOTASK']._serialized_end=319
+  _globals['_IOTASK_LIMITSENTRY']._serialized_start=274
+  _globals['_IOTASK_LIMITSENTRY']._serialized_end=319
+  _globals['_CAPABILITY']._serialized_start=322
+  _globals['_CAPABILITY']._serialized_end=602
+  _globals['_CAPABILITY_EXTRAENTRY']._serialized_start=558
+  _globals['_CAPABILITY_EXTRAENTRY']._serialized_end=602
+  _globals['_INFERREQUEST']._serialized_start=605
+  _globals['_INFERREQUEST']._serialized_end=839
+  _globals['_INFERREQUEST_METAENTRY']._serialized_start=796
+  _globals['_INFERREQUEST_METAENTRY']._serialized_end=839
+  _globals['_INFERRESPONSE']._serialized_start=842
+  _globals['_INFERRESPONSE']._serialized_end=1141
+  _globals['_INFERRESPONSE_METAENTRY']._serialized_start=796
+  _globals['_INFERRESPONSE_METAENTRY']._serialized_end=839
+  _globals['_INFERENCE']._serialized_start=1306
+  _globals['_INFERENCE']._serialized_end=1596
+# @@protoc_insertion_point(module_scope)