Skip to content

Commit 30f17ce

Browse files
committed
fix: tests
1 parent 4237441 commit 30f17ce

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/ai/AkamaiKnowledgeBaseCR.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ describe('AkamaiKnowledgeBaseCR', () => {
6868
expect(kbCR.spec.pipelineParameters).toEqual({
6969
url: 'https://docs.example.com',
7070
table_name: 'test-kb',
71-
embedding_model: 'nvidia/text-embedding-ada-002',
71+
embedding_model: 'text-embedding-ada-002',
7272
embedding_api_base: 'http://embedding-model.ai.svc.cluster.local',
7373
embed_dim: 1536,
7474
embed_batch_size: expect.any(Number),
@@ -133,7 +133,7 @@ describe('AkamaiKnowledgeBaseCR', () => {
133133
expect(response.kind).toBe('AkamaiKnowledgeBase')
134134
expect(response.metadata.name).toBe('test-kb')
135135
expect(response.metadata.labels?.['apl.io/teamId']).toBe('team-123')
136-
expect(response.spec.modelName).toBe('nvidia/text-embedding-ada-002')
136+
expect(response.spec.modelName).toBe('text-embedding-ada-002')
137137
expect(response.spec.sourceUrl).toBe('https://docs.example.com')
138138
expect(response.status).toEqual({})
139139
})
@@ -216,7 +216,7 @@ describe('AkamaiKnowledgeBaseCR', () => {
216216
const result = await AkamaiKnowledgeBaseCR.create('team-123', 'test-kb', 'cluster-name', mockKnowledgeBaseRequest)
217217

218218
expect(result).toBeInstanceOf(AkamaiKnowledgeBaseCR)
219-
expect(result.spec.pipelineParameters.embedding_model).toBe('nvidia/text-embedding-ada-002')
219+
expect(result.spec.pipelineParameters.embedding_model).toBe('text-embedding-ada-002')
220220
})
221221

222222
test('should handle AI models fetch error', async () => {

0 commit comments

Comments
 (0)