added new navira models

astuto-ai · Dec 18, 2024 · 1f646d3 · 1f646d3
1 parent 738d70c
commit 1f646d3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/onelens_backend_client_v2/models.py b/onelens_backend_client_v2/models.py
@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  <stdin>
-#   timestamp: 2024-12-17T10:44:49+00:00
+#   timestamp: 2024-12-18T07:33:26+00:00
 
 from __future__ import annotations
 
@@ -3570,6 +3570,12 @@ class UpdateEmbeddingsRequest(BaseModel):
     )
 
 
+class UpdateEmbeddingsResponse(BaseModel):
+    status: str = Field(
+        ..., description="The status of the embeddings update.", title="Status"
+    )
+
+
 class UpdateFeatureStatusRequest(BaseModel):
     tenant_id: UUID4 = Field(..., title="Tenant Id")
     id: UUID4 = Field(..., title="Id")

diff --git a/onelens_backend_client_v2/rpc/navira_embedding_service_rpc_handler.py b/onelens_backend_client_v2/rpc/navira_embedding_service_rpc_handler.py
@@ -8,6 +8,9 @@
 from onelens_backend_client_v2.models import UpdateEmbeddingsRequest
 
 
+from onelens_backend_client_v2.models import UpdateEmbeddingsResponse
+
+
 from onelens_backend_client_v2.api_client import ApiClient, RequestSerialized
 
 
@@ -34,7 +37,7 @@ def update_embeddings(
         _content_type: Optional[StrictStr] = None,
         _headers: Optional[Dict[StrictStr, Any]] = None,
         _host_index: Annotated[StrictInt, Field(ge=0, le=0)] = 0,
-    ) -> Any:
+    ) -> UpdateEmbeddingsResponse:
         """Update Embeddings
 
 
@@ -72,7 +75,7 @@ def update_embeddings(
         )
 
         _response_types_map: Dict[str, Optional[str]] = {
-            "200": "Any",
+            "200": "UpdateEmbeddingsResponse",
             "422": "HTTPValidationError",
         }
         response_data = self.api_client.call_api(