11import { V1Deployment } from '@kubernetes/client-node'
2- import { transformK8sDeploymentToAplAIModel , getAIModels } from './aiModelHandler'
2+ import { getAIModels , transformK8sDeploymentToAplAIModel } from './aiModelHandler'
33import * as k8s from './k8s'
44
55// Mock the k8s module
@@ -14,7 +14,9 @@ describe('aiModelHandler', () => {
1414 name : 'gpt-4-deployment' ,
1515 namespace : 'ai-models' ,
1616 labels : {
17+ app : 'gpt-4' ,
1718 modelName : 'gpt-4' ,
19+ modelNameTitle : 'GPT-4o-mini' ,
1820 modelType : 'foundation' ,
1921 modelDimension : '1536' ,
2022 } ,
@@ -55,8 +57,8 @@ describe('aiModelHandler', () => {
5557 name : 'gpt-4' ,
5658 } ,
5759 spec : {
58- displayName : 'gpt-4 ' ,
59- modelEndpoint : 'http://gpt-4-deployment .ai-models.svc.cluster.local' ,
60+ displayName : 'GPT-4o-mini ' ,
61+ modelEndpoint : 'http://gpt-4.ai-models.svc.cluster.local/openai/v1 ' ,
6062 modelType : 'foundation' ,
6163 modelDimension : 1536 ,
6264 } ,
@@ -98,7 +100,7 @@ describe('aiModelHandler', () => {
98100 const result = transformK8sDeploymentToAplAIModel ( deploymentWithModelName )
99101
100102 expect ( result . metadata . name ) . toBe ( 'custom-model-name' )
101- expect ( result . spec . displayName ) . toBe ( 'custom-model-name ' )
103+ expect ( result . spec . displayName ) . toBe ( 'GPT-4o-mini ' )
102104 } )
103105
104106 test ( 'should use modelName from labels when deployment name is missing' , ( ) => {
@@ -117,7 +119,7 @@ describe('aiModelHandler', () => {
117119 const result = transformK8sDeploymentToAplAIModel ( deploymentWithoutName )
118120
119121 expect ( result . metadata . name ) . toBe ( 'custom-model-name' )
120- expect ( result . spec . displayName ) . toBe ( 'custom-model-name ' )
122+ expect ( result . spec . displayName ) . toBe ( 'GPT-4o-mini ' )
121123 } )
122124
123125 test ( 'should handle deployment without labels' , ( ) => {
@@ -164,7 +166,7 @@ describe('aiModelHandler', () => {
164166
165167 const result = transformK8sDeploymentToAplAIModel ( deploymentWithoutNamespace )
166168
167- expect ( result . spec . modelEndpoint ) . toBe ( 'http://test-deployment .undefined.svc.cluster.local' )
169+ expect ( result . spec . modelEndpoint ) . toBe ( 'http://gpt-4 .undefined.svc.cluster.local/openai/v1 ' )
168170 } )
169171
170172 test ( 'should handle deployment without status conditions' , ( ) => {
@@ -232,7 +234,7 @@ describe('aiModelHandler', () => {
232234 const result = transformK8sDeploymentToAplAIModel ( deploymentWithoutMetadata )
233235
234236 expect ( result . metadata . name ) . toBe ( '' )
235- expect ( result . spec . modelEndpoint ) . toBe ( 'http://undefined .undefined.svc.cluster.local' )
237+ expect ( result . spec . modelEndpoint ) . toBe ( 'http://.undefined.svc.cluster.local/openai/v1 ' )
236238 } )
237239 } )
238240
0 commit comments