indexify 0.3.12__tar.gz → 0.3.13__tar.gz

{indexify-0.3.12 → indexify-0.3.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: indexify
-Version: 0.3.12
+Version: 0.3.13
 Summary: Open Source Indexify components and helper tools
 Home-page: https://github.com/tensorlakeai/indexify
 License: Apache 2.0

{indexify-0.3.12 → indexify-0.3.13}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "indexify"
 # Incremented if any of the components provided in this packages are updated.
-version = "0.3.12"
+version = "0.3.13"
 description = "Open Source Indexify components and helper tools"
 authors = ["Tensorlake Inc. <support@tensorlake.ai>"]
 license = "Apache 2.0"

{indexify-0.3.12 → indexify-0.3.13}/src/indexify/executor/function_executor/function_executor.py

@@ -3,12 +3,15 @@ from typing import Any, Optional
 
 import grpc
 from tensorlake.function_executor.proto.function_executor_pb2 import (
+    InfoRequest,
+    InfoResponse,
     InitializeRequest,
     InitializeResponse,
 )
 from tensorlake.function_executor.proto.function_executor_pb2_grpc import (
     FunctionExecutorStub,
 )
+from tensorlake.function_executor.proto.message_validator import MessageValidator
 from tensorlake.utils.http_client import get_httpx_client
 
 from .health_checker import HealthChecker
@@ -36,7 +39,10 @@ from .metrics.function_executor import (
     metric_destroys,
     metric_establish_channel_errors,
     metric_establish_channel_latency,
+    metric_function_executor_infos,
     metric_function_executors_count,
+    metric_get_info_rpc_errors,
+    metric_get_info_rpc_latency,
     metric_initialize_rpc_errors,
     metric_initialize_rpc_latency,
 )
@@ -96,6 +102,7 @@ class FunctionExecutor:
                 await self._create_server(config)
                 await self._establish_channel()
                 stub: FunctionExecutorStub = FunctionExecutorStub(self._channel)
+                await _collect_server_info(stub)
                 await _initialize_server(stub, initialize_request)
                 await self._create_invocation_state_client(
                     stub=stub,
@@ -262,9 +269,29 @@ class FunctionExecutor:
             self._health_checker = None
 
 
+async def _collect_server_info(stub: FunctionExecutorStub) -> None:
+    with (
+        metric_get_info_rpc_errors.count_exceptions(),
+        metric_get_info_rpc_latency.time(),
+    ):
+        info: InfoResponse = await stub.get_info(InfoRequest())
+        validator = MessageValidator(info)
+        validator.required_field("version")
+        validator.required_field("sdk_version")
+        validator.required_field("sdk_language")
+        validator.required_field("sdk_language_version")
+
+        metric_function_executor_infos.labels(
+            version=info.version,
+            sdk_version=info.sdk_version,
+            sdk_language=info.sdk_language,
+            sdk_language_version=info.sdk_language_version,
+        ).inc()
+
+
 async def _initialize_server(
     stub: FunctionExecutorStub, initialize_request: InitializeRequest
-):
+) -> None:
     with (
         metric_initialize_rpc_errors.count_exceptions(),
         metric_initialize_rpc_latency.time(),

{indexify-0.3.12 → indexify-0.3.13}/src/indexify/executor/function_executor/metrics/function_executor.py

@@ -78,6 +78,22 @@ metric_destroy_channel_errors: prometheus_client.Counter = prometheus_client.Cou
     "Number of Function Executor channel destruction errors",
 )
 
+# FE get_info RPC metrics.
+metric_get_info_rpc_latency: prometheus_client.Histogram = (
+    latency_metric_for_fast_operation(
+        "function_executor_get_info_rpc", "Function Executor get_info RPC"
+    )
+)
+metric_get_info_rpc_errors: prometheus_client.Counter = prometheus_client.Counter(
+    "function_executor_get_info_rpc_errors",
+    "Number of Function Executor get_info RPC errors",
+)
+metric_function_executor_infos: prometheus_client.Counter = prometheus_client.Counter(
+    "function_executor_infos",
+    "Number of Function Executors with particular info",
+    ["version", "sdk_version", "sdk_language", "sdk_language_version"],
+)
+
 # FE initialization RPC metrics.
 metric_initialize_rpc_latency: prometheus_client.Histogram = (
     latency_metric_for_customer_controlled_operation(