update splitLlamaModels (opendatahub-io#5230)

* update splitLlamaModels

* add unit tests

Author: pnaik1

packages/gen-ai/frontend/src/app/utilities/__tests__/utils.spec.ts

@@ -0,0 +1,51 @@
+import { splitLlamaModelId } from '~/app/utilities/utils';
+
+describe('splitLlamaModelId', () => {
+  it('should split a valid model ID with provider and ID', () => {
+    const result = splitLlamaModelId('vllm-inference-1/facebook-opt-1');
+    expect(result).toEqual({
+      providerId: 'vllm-inference-1',
+      id: 'facebook-opt-1',
+    });
+  });
+
+  it('should handle model ID without slash', () => {
+    const result = splitLlamaModelId('facebook-opt-1');
+    expect(result).toEqual({
+      providerId: '',
+      id: 'facebook-opt-1',
+    });
+  });
+
+  it('should handle empty string', () => {
+    const result = splitLlamaModelId('');
+    expect(result).toEqual({
+      providerId: '',
+      id: '',
+    });
+  });
+
+  it('should handle model ID with multiple slashes (only splits on first)', () => {
+    const result = splitLlamaModelId('vllm-inference-1/facebook/opt-1');
+    expect(result).toEqual({
+      providerId: 'vllm-inference-1',
+      id: 'facebook/opt-1',
+    });
+  });
+
+  it('should handle complex provider and model names', () => {
+    const result = splitLlamaModelId('anthropic-vertex/claude-3-opus-20240229');
+    expect(result).toEqual({
+      providerId: 'anthropic-vertex',
+      id: 'claude-3-opus-20240229',
+    });
+  });
+
+  it('should handle model ID with special characters', () => {
+    const result = splitLlamaModelId('provider-123/model_name-v2.0');
+    expect(result).toEqual({
+      providerId: 'provider-123',
+      id: 'model_name-v2.0',
+    });
+  });
+});

packages/gen-ai/frontend/src/app/utilities/utils.ts

@@ -12,7 +12,12 @@ export const convertAIModelToK8sResource = (model: AIModel): K8sResourceCommon =
 });
 
 export const splitLlamaModelId = (llamaModelId: string): { providerId: string; id: string } => {
-  const [providerId, id] = llamaModelId.split('/');
+  const slashIndex = llamaModelId.indexOf('/');
+  if (slashIndex === -1) {
+    return { providerId: '', id: llamaModelId };
+  }
+  const providerId = llamaModelId.substring(0, slashIndex);
+  const id = llamaModelId.substring(slashIndex + 1);
   if (!providerId || !id) {
     return { providerId: '', id: llamaModelId };
   }