feat: adds embeddings support for parasail

Author: sammaji

core/providers/parasail/parasail.go

@@ -164,9 +164,23 @@ func (provider *ParasailProvider) ResponsesStream(ctx *schemas.BifrostContext, p
 	)
 }
 
-// Embedding is not supported by the Parasail provider.
+// Embedding performs an embedding request to the Parasail API.
+// Uses Parasail's OpenAI-compatible embeddings endpoint.
 func (provider *ParasailProvider) Embedding(ctx *schemas.BifrostContext, key schemas.Key, request *schemas.BifrostEmbeddingRequest) (*schemas.BifrostEmbeddingResponse, *schemas.BifrostError) {
-	return nil, providerUtils.NewUnsupportedOperationError(schemas.EmbeddingRequest, provider.GetProviderKey())
+	return openai.HandleOpenAIEmbeddingRequest(
+		ctx,
+		provider.client,
+		provider.networkConfig.BaseURL+providerUtils.GetPathFromContext(ctx, "/v1/embeddings"),
+		request,
+		openai.BearerAuthHeader(key),
+		provider.networkConfig.ExtraHeaders,
+		provider.GetProviderKey(),
+		providerUtils.ShouldSendBackRawRequest(ctx, provider.sendBackRawRequest),
+		providerUtils.ShouldSendBackRawResponse(ctx, provider.sendBackRawResponse),
+		nil,
+		nil,
+		provider.logger,
+	)
 }
 
 // Speech is not supported by the Parasail provider.

core/providers/parasail/parasail_test.go

@@ -27,7 +27,7 @@ func TestParasail(t *testing.T) {
 		Provider:       schemas.Parasail,
 		ChatModel:      "parasail-llama-33-70b-fp8",
 		TextModel:      "", // Parasail doesn't support text completion
-		EmbeddingModel: "", // Parasail doesn't support embedding
+		EmbeddingModel: "parasail-bge-m3",
 		Scenarios: llmtests.TestScenarios{
 			TextCompletion:        false, // Not supported
 			SimpleChat:            true,
@@ -42,7 +42,7 @@ func TestParasail(t *testing.T) {
 			ImageBase64:           false, // Not supported yet
 			MultipleImages:        false, // Not supported yet
 			CompleteEnd2End:       true,
-			Embedding:             false, // Not supported yet
+			Embedding:             true,
 			ListModels:            true,
 		},
 	}

docs/providers/supported-providers/overview.mdx

@@ -35,7 +35,7 @@ The following table summarizes which operations are supported by each provider v
 | Ollama (`ollama/<model>`)            | ✅     | ✅   | ✅            | ✅   | ✅            | ✅        | ✅                 | ❌     | ❌              | ❌         | ❌                  | ❌              | ✅         | ❌  | ❌           | ❌  | ❌           | ❌    | ❌    | ❌           | ❌     | ❌  | ❌    | ❌          | ❌         | ❌          | ❌                   |
 | OpenAI (`openai/<model>`)            | ✅     | ✅   | ✅            | ✅   | ✅            | ✅        | ✅                 | ✅     | ✅              | ✅         | ✅                  | ✅              | ✅         | ✅  | ✅           | ✅  | ✅           | ✅    | ✅    | ✅           | ❌     | ❌  | ✅    | ✅          | ✅         | ✅          | ✅                   |
 | OpenRouter (`openrouter/<model>`)    | ✅     | ✅   | ✅            | ✅   | ✅            | ✅        | ✅                 | ❌     | ❌              | ❌         | ❌                  | ❌              | ✅         | ❌  | ❌           | ❌  | ❌           | ❌    | ❌    | ❌           | ❌     | ❌  | ❌    | ❌          | ❌         | ❌          | ❌                   |
-| Parasail (`parasail/<model>`)        | ✅     | ❌   | ❌            | ✅   | ✅            | ✅        | ✅                 | ❌     | ❌              | ❌         | ❌                  | ❌              | ❌         | ❌  | ❌           | ❌  | ❌           | ❌    | ❌    | ❌           | ❌     | ❌  | ❌    | ❌          | ❌         | ❌          | ❌                   |
+| Parasail (`parasail/<model>`)        | ✅     | ❌   | ❌            | ✅   | ✅            | ✅        | ✅                 | ❌     | ❌              | ❌         | ❌                  | ❌              | ✅         | ❌  | ❌           | ❌  | ❌           | ❌    | ❌    | ❌           | ❌     | ❌  | ❌    | ❌          | ❌         | ❌          | ❌                   |
 | Perplexity (`perplexity/<model>`)    | ❌     | ❌   | ❌            | ✅   | ✅            | ✅        | ✅                 | ❌     | ❌              | ❌         | ❌                  | ❌              | ❌         | ❌  | ❌           | ❌  | ❌           | ❌    | ❌    | ❌           | ❌     | ❌  | ❌    | ❌          | ❌         | ❌          | ❌                   |
 | Replicate (`replicate/<model>`)      | ✅     | ✅   | ✅            | ✅   | ✅            | ✅        | ✅                 | ✅     | ✅              | ✅         | ✅                  | ❌              | ❌         | ❌  | ❌           | ❌  | ❌           | ✅    | ❌    | ❌           | ❌     | ❌  | ✅    | ❌          | ❌         | ❌          | ❌                   |
 | Runware (`runware/<model>`)          | ❌     | ❌   | ❌            | ❌   | ❌            | ❌        | ❌                 | ✅     | ❌              | ✅         | ❌                  | ❌              | ❌         | ❌  | ❌           | ❌  | ❌           | ❌    | ❌    | ❌           | ❌     | ❌  | ✅    | ❌          | ❌         | ❌          | ❌                   |

docs/providers/supported-providers/parasail.mdx

@@ -20,16 +20,16 @@ Parasail is an **OpenAI-compatible provider** offering high-performance inferenc
 | Chat Completions | ✅ | ✅ | `/v1/chat/completions` |
 | Responses API | ✅ | ✅ | `/v1/chat/completions` |
 | List Models | ✅ | - | `/v1/models` |
+| Embeddings | ✅ | - | `/v1/embeddings` |
 | Text Completions | ❌ | ❌ | - |
-| Embeddings | ❌ | ❌ | - |
 | Image Generation | ❌ | ❌ | - |
 | Speech (TTS) | ❌ | ❌ | - |
 | Transcriptions (STT) | ❌ | ❌ | - |
 | Files | ❌ | ❌ | - |
 | Batch | ❌ | ❌ | - |
 
 <Note>
-**Unsupported Operations** (❌): Text Completions, Embeddings, Image Generation, Speech, Transcriptions, Files, and Batch are not supported by the upstream Parasail API. These return `UnsupportedOperationError`.
+**Unsupported Operations** (❌): Text Completions, Image Generation, Speech, Transcriptions, Files, and Batch are not supported by the upstream Parasail API. These return `UnsupportedOperationError`.
 </Note>
 
 ## Setup & Configuration
@@ -139,12 +139,19 @@ Lists available Parasail models with capabilities and context information.
 
 ---
 
+# 4. Embeddings
+
+Parasail's OpenAI-compatible `/v1/embeddings` endpoint, used through the standard embeddings interface. Accepts a single string or an array of strings, and `encoding_format` of `float` or `base64`.
+
+Embedding models are addressed either by Parasail's own ID or by the underlying HuggingFace ID, for example `parasail-bge-m3` or `BAAI/bge-m3`. Which ones your key can reach comes from `/v1/models`.
+
+---
+
 ## Unsupported Features
 
 | Feature | Reason |
 |---------|--------|
 | Text Completions | Not offered by Parasail API |
-| Embedding | Not offered by Parasail API |
 | Image Generation | Not offered by Parasail API |
 | Speech/TTS | Not offered by Parasail API |
 | Transcription/STT | Not offered by Parasail API |