diff --git a/onelens_backend_client_v2/models.py b/onelens_backend_client_v2/models.py index b908039..dd3a4af 100644 --- a/onelens_backend_client_v2/models.py +++ b/onelens_backend_client_v2/models.py @@ -1,6 +1,6 @@ # generated by datamodel-codegen: # filename: -# 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 index 24ac544..d29ea39 100644 --- 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(