Skip to content

Commit b855acf

Browse files
committed
feat: Add geminiOptions to configuration tests and update model dimension assertions
1 parent 9377f4e commit b855acf

File tree

2 files changed

+36
-3
lines changed

2 files changed

+36
-3
lines changed

src/services/code-index/__tests__/config-manager.test.ts

Lines changed: 33 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,13 @@ describe("CodeIndexConfigManager", () => {
3737
modelId: undefined,
3838
openAiOptions: { openAiNativeApiKey: "" },
3939
ollamaOptions: { ollamaBaseUrl: "" },
40+
geminiOptions: {
41+
apiModelId: undefined,
42+
geminiApiKey: "",
43+
geminiEmbeddingDimension: undefined,
44+
geminiEmbeddingTaskType: "CODE_RETRIEVAL_QUERY",
45+
rateLimitSeconds: undefined,
46+
},
4047
qdrantUrl: "http://localhost:6333",
4148
qdrantApiKey: "",
4249
searchMinScore: 0.4,
@@ -68,6 +75,19 @@ describe("CodeIndexConfigManager", () => {
6875
modelId: "text-embedding-3-large",
6976
openAiOptions: { openAiNativeApiKey: "test-openai-key" },
7077
ollamaOptions: { ollamaBaseUrl: "" },
78+
geminiOptions: {
79+
apiModelId: "text-embedding-3-large",
80+
geminiApiKey: "",
81+
geminiEmbeddingDimension: undefined,
82+
geminiEmbeddingTaskType: "CODE_RETRIEVAL_QUERY",
83+
rateLimitSeconds: {
84+
codebaseIndexEnabled: true,
85+
codebaseIndexQdrantUrl: "http://qdrant.local",
86+
codebaseIndexEmbedderProvider: "openai",
87+
codebaseIndexEmbedderBaseUrl: "",
88+
codebaseIndexEmbedderModelId: "text-embedding-3-large",
89+
},
90+
},
7191
qdrantUrl: "http://qdrant.local",
7292
qdrantApiKey: "test-qdrant-key",
7393
searchMinScore: 0.4,
@@ -485,6 +505,19 @@ describe("CodeIndexConfigManager", () => {
485505
modelId: "text-embedding-3-large",
486506
openAiOptions: { openAiNativeApiKey: "test-openai-key" },
487507
ollamaOptions: { ollamaBaseUrl: undefined },
508+
geminiEmbeddingDimension: undefined,
509+
geminiOptions: {
510+
apiModelId: "text-embedding-3-large",
511+
geminiApiKey: "",
512+
geminiEmbeddingDimension: undefined,
513+
geminiEmbeddingTaskType: "CODE_RETRIEVAL_QUERY",
514+
rateLimitSeconds: {
515+
codebaseIndexEnabled: true,
516+
codebaseIndexQdrantUrl: "http://qdrant.local",
517+
codebaseIndexEmbedderProvider: "openai",
518+
codebaseIndexEmbedderModelId: "text-embedding-3-large",
519+
},
520+
},
488521
qdrantUrl: "http://qdrant.local",
489522
qdrantApiKey: "test-qdrant-key",
490523
searchMinScore: 0.4,

src/services/code-index/__tests__/service-factory.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ describe("CodeIndexServiceFactory", () => {
194194
factory.createVectorStore()
195195

196196
// Assert
197-
expect(mockGetModelDimension).toHaveBeenCalledWith("openai", testModelId)
197+
expect(mockGetModelDimension).toHaveBeenCalledWith("openai", testModelId, undefined)
198198
expect(MockedQdrantVectorStore).toHaveBeenCalledWith(
199199
"/test/workspace",
200200
"http://localhost:6333",
@@ -219,7 +219,7 @@ describe("CodeIndexServiceFactory", () => {
219219
factory.createVectorStore()
220220

221221
// Assert
222-
expect(mockGetModelDimension).toHaveBeenCalledWith("ollama", testModelId)
222+
expect(mockGetModelDimension).toHaveBeenCalledWith("ollama", testModelId, undefined)
223223
expect(MockedQdrantVectorStore).toHaveBeenCalledWith(
224224
"/test/workspace",
225225
"http://localhost:6333",
@@ -243,7 +243,7 @@ describe("CodeIndexServiceFactory", () => {
243243
factory.createVectorStore()
244244

245245
// Assert
246-
expect(mockGetModelDimension).toHaveBeenCalledWith("openai", "default-model")
246+
expect(mockGetModelDimension).toHaveBeenCalledWith("openai", "default-model", undefined)
247247
expect(MockedQdrantVectorStore).toHaveBeenCalledWith(
248248
"/test/workspace",
249249
"http://localhost:6333",

0 commit comments

Comments
 (0)