Update version (#1648)

* Add support for JSON and SSE responses to invoke endpoint

* update tensorlake version, indexify server and executor version

* update poetry lock

* fix import

* fix compilation errors

* fix test

Author: diptanu

Cargo.lock

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "indexify"
 # Incremented if any of the components provided in this packages are updated.
-version = "0.4.20"
+version = "0.4.21"
 description = "Open Source Indexify components and helper tools"
 authors = ["Tensorlake Inc. <support@tensorlake.ai>"]
 license = "Apache 2.0"
@@ -29,7 +29,7 @@ psutil = "^7.0.0"
 boto3 = "^1.39.14"
 # Adds function-executor binary, utils lib, sdk used in indexify-cli commands.
 # We need to specify the tensorlake version exactly because pip install doesn't respect poetry.lock files.
-tensorlake = "0.2.32"
+tensorlake = "0.2.33"
 # Uncomment the next line to use local tensorlake package (only for development!)
 # tensorlake = { path = "../tensorlake", develop = true }
 # grpcio is provided by tensorlake

indexify/poetry.lock

@@ -42,7 +42,7 @@ def setUp(self):
 
     def _stream(self) -> tuple[dict[str, int], str]:
         counts = defaultdict(int)
-        events = self.graph.stream(block_until_done=True, count=2)
+        events = self.graph.stream(count=2)
         try:
             while True:
                 event = next(events)

indexify/pyproject.toml

@@ -1,6 +1,6 @@
 [package]
 name = "indexify-server"
-version = "0.3.7"
+version = "0.3.8"
 edition = "2021"
 authors = ["Tensorlake Inc. <support@tensorlake.ai>"]
 license = "Apache-2.0"

indexify/tests/features/test_cacheable.py

@@ -262,14 +262,13 @@ pub async fn invoke_with_object_v1(
 }
 
 #[derive(Debug, Deserialize)]
-pub struct RequestQueryParam {
+pub struct RequestQueryParams {
     pub block_until_finish: Option<bool>,
 }
 
-#[axum::debug_handler]
 pub async fn invoke_with_object(
     Path((namespace, compute_graph)): Path<(String, String)>,
-    Query(params): Query<RequestQueryParam>,
+    Query(params): Query<RequestQueryParams>,
     State(state): State<RouteState>,
     headers: HeaderMap,
     body: Body,
@@ -318,7 +317,6 @@ pub async fn invoke_with_object(
     if should_block {
         rx.replace(state.indexify_state.task_event_stream());
     }
-
     let compute_graph = state
         .indexify_state
         .reader()
@@ -342,7 +340,7 @@ pub async fn invoke_with_object(
         namespace: namespace.clone(),
         compute_graph_name: compute_graph.name.clone(),
         invocation_payload,
-        ctx: graph_invocation_ctx.clone(),
+        ctx: graph_invocation_ctx,
     });
     state
         .indexify_state
@@ -357,13 +355,13 @@ pub async fn invoke_with_object(
 
     let invocation_event_stream =
         create_invocation_progress_stream(id, rx, state, namespace, compute_graph.name).await;
-    Ok(axum::response::Sse::new(invocation_event_stream)
-        .keep_alive(
+    Ok(
+        axum::response::Sse::new(invocation_event_stream).keep_alive(
             axum::response::sse::KeepAlive::new()
                 .interval(Duration::from_secs(1))
                 .text("keep-alive-text"),
-        )
-        .into_response())
+        ),
+    )
 }
 
 /// Stream progress of a request until it is completed