feat: add OpenAI-compatible embeddings endpoint

Cargo.toml

@@ -20,7 +20,7 @@ path = "src/main.rs"
 
 [dependencies]
 # OpenSecret SDK 
-opensecret = "0.2.3"
+opensecret = "0.2.7"
 
 # Web server
 axum = { version = "0.8.4", features = ["http2", "macros"] }

src/lib.rs

@@ -2,7 +2,7 @@ pub mod config;
 pub mod proxy;
 
 pub use config::{Config, OpenAIError, OpenAIErrorDetails};
-use proxy::{create_chat_completion, health_check, list_models, ProxyState};
+use proxy::{create_chat_completion, create_embeddings, health_check, list_models, ProxyState};
 
 use axum::{
     http::Method,
@@ -28,6 +28,7 @@ pub fn create_app(config: Config) -> Router {
         // OpenAI-compatible endpoints
         .route("/v1/models", get(list_models))
         .route("/v1/chat/completions", post(create_chat_completion))
+        .route("/v1/embeddings", post(create_embeddings))
         .with_state(state)
         .layer(
             ServiceBuilder::new().layer(

src/proxy.rs

@@ -7,8 +7,8 @@ use axum::{
 };
 use futures::Stream;
 use opensecret::{
-    ChatCompletionChunk, ChatCompletionRequest, ModelsResponse, OpenSecretClient,
-    Result as OpenSecretResult,
+    ChatCompletionChunk, ChatCompletionRequest, EmbeddingRequest, EmbeddingResponse,
+    ModelsResponse, OpenSecretClient, Result as OpenSecretResult,
 };
 use std::{convert::Infallible, sync::Arc};
 use tracing::{debug, error};
@@ -198,3 +198,35 @@ fn create_sse_stream(
         yield Ok(done_event);
     }
 }
+
+pub async fn create_embeddings(
+    State(state): State<Arc<ProxyState>>,
+    headers: HeaderMap,
+    Json(request): Json<EmbeddingRequest>,
+) -> Result<Json<EmbeddingResponse>, (StatusCode, Json<OpenAIError>)> {
+    let api_key = extract_api_key(&headers, &state.config.default_api_key)
+        .map_err(|e| (StatusCode::UNAUTHORIZED, Json(e)))?;
+
+    debug!("Embeddings request for model: {}", request.model);
+
+    let client = create_client_with_auth(&state.config.backend_url, &api_key)
+        .await
+        .map_err(|e| (StatusCode::INTERNAL_SERVER_ERROR, Json(e)))?;
+
+    let response = client.create_embeddings(request).await.map_err(|e| {
+        error!("Failed to create embeddings: {}", e);
+        (
+            StatusCode::INTERNAL_SERVER_ERROR,
+            Json(OpenAIError::server_error(format!(
+                "Failed to create embeddings: {}",
+                e
+            ))),
+        )
+    })?;
+
+    debug!(
+        "Successfully created embeddings with {} vectors",
+        response.data.len()
+    );
+    Ok(Json(response))
+}