From 041664358c18fba4a608d7969df7dd6c3a7222e1 Mon Sep 17 00:00:00 2001
From: Roo Code <roomote@roocode.com>
Date: Wed, 30 Jul 2025 14:41:47 +0000
Subject: [PATCH] fix: complete Jina provider support in type definitions and
 model profiles

- Add "jina" to EmbedderProvider type definitions in 5 files
- Add Jina model profiles (jina-embeddings-v4, jina-embeddings-v3, jina-clip-v2)
- Add Jina case to getDefaultModelId function
- Update Zod schema in packages/types to include "jina"
- Fixes TypeScript compilation errors for Jina embedding provider
---
 packages/types/src/codebase-index.ts           |  4 +++-
 src/services/code-index/interfaces/embedder.ts |  2 +-
 src/services/code-index/interfaces/manager.ts  |  2 +-
 src/shared/WebviewMessage.ts                   |  2 +-
 src/shared/embeddingModels.ts                  | 10 +++++++++-
 5 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/packages/types/src/codebase-index.ts b/packages/types/src/codebase-index.ts
index 89d5b168d7..a914224e9f 100644
--- a/packages/types/src/codebase-index.ts
+++ b/packages/types/src/codebase-index.ts
@@ -21,7 +21,9 @@ export const CODEBASE_INDEX_DEFAULTS = {
 export const codebaseIndexConfigSchema = z.object({
 	codebaseIndexEnabled: z.boolean().optional(),
 	codebaseIndexQdrantUrl: z.string().optional(),
-	codebaseIndexEmbedderProvider: z.enum(["openai", "ollama", "openai-compatible", "gemini", "mistral"]).optional(),
+	codebaseIndexEmbedderProvider: z
+		.enum(["openai", "ollama", "openai-compatible", "gemini", "mistral", "jina"])
+		.optional(),
 	codebaseIndexEmbedderBaseUrl: z.string().optional(),
 	codebaseIndexEmbedderModelId: z.string().optional(),
 	codebaseIndexEmbedderModelDimension: z.number().optional(),
diff --git a/src/services/code-index/interfaces/embedder.ts b/src/services/code-index/interfaces/embedder.ts
index c5653ea2b7..87a41c5b39 100644
--- a/src/services/code-index/interfaces/embedder.ts
+++ b/src/services/code-index/interfaces/embedder.ts
@@ -28,7 +28,7 @@ export interface EmbeddingResponse {
 	}
 }
 
-export type AvailableEmbedders = "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral"
+export type AvailableEmbedders = "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral" | "jina"
 
 export interface EmbedderInfo {
 	name: AvailableEmbedders
diff --git a/src/services/code-index/interfaces/manager.ts b/src/services/code-index/interfaces/manager.ts
index fd3b2bfdda..1fa3817203 100644
--- a/src/services/code-index/interfaces/manager.ts
+++ b/src/services/code-index/interfaces/manager.ts
@@ -70,7 +70,7 @@ export interface ICodeIndexManager {
 }
 
 export type IndexingState = "Standby" | "Indexing" | "Indexed" | "Error"
-export type EmbedderProvider = "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral"
+export type EmbedderProvider = "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral" | "jina"
 
 export interface IndexProgressUpdate {
 	systemStatus: IndexingState
diff --git a/src/shared/WebviewMessage.ts b/src/shared/WebviewMessage.ts
index a91d1af7ba..bf8d2bc387 100644
--- a/src/shared/WebviewMessage.ts
+++ b/src/shared/WebviewMessage.ts
@@ -255,7 +255,7 @@ export interface WebviewMessage {
 		// Global state settings
 		codebaseIndexEnabled: boolean
 		codebaseIndexQdrantUrl: string
-		codebaseIndexEmbedderProvider: "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral"
+		codebaseIndexEmbedderProvider: "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral" | "jina"
 		codebaseIndexEmbedderBaseUrl?: string
 		codebaseIndexEmbedderModelId: string
 		codebaseIndexEmbedderModelDimension?: number // Generic dimension for all providers
diff --git a/src/shared/embeddingModels.ts b/src/shared/embeddingModels.ts
index a3cd61e659..52d476c39a 100644
--- a/src/shared/embeddingModels.ts
+++ b/src/shared/embeddingModels.ts
@@ -2,7 +2,7 @@
  * Defines profiles for different embedding models, including their dimensions.
  */
 
-export type EmbedderProvider = "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral" // Add other providers as needed
+export type EmbedderProvider = "openai" | "ollama" | "openai-compatible" | "gemini" | "mistral" | "jina" // Add other providers as needed
 
 export interface EmbeddingModelProfile {
 	dimension: number
@@ -53,6 +53,11 @@ export const EMBEDDING_MODEL_PROFILES: EmbeddingModelProfiles = {
 	mistral: {
 		"codestral-embed-2505": { dimension: 1536, scoreThreshold: 0.4 },
 	},
+	jina: {
+		"jina-embeddings-v4": { dimension: 1024, scoreThreshold: 0.4 },
+		"jina-embeddings-v3": { dimension: 1024, scoreThreshold: 0.4 },
+		"jina-clip-v2": { dimension: 768, scoreThreshold: 0.4 },
+	},
 }
 
 /**
@@ -143,6 +148,9 @@ export function getDefaultModelId(provider: EmbedderProvider): string {
 		case "mistral":
 			return "codestral-embed-2505"
 
+		case "jina":
+			return "jina-embeddings-v4"
+
 		default:
 			// Fallback for unknown providers
 			console.warn(`Unknown provider for default model ID: ${provider}. Falling back to OpenAI default.`)