feat: add optional language and frameworks fields to recpies

Signed-off-by: Philippe Martin <[email protected]>
containers · Dec 24, 2024 · c51c8a0 · c51c8a0
1 parent 358fc37
commit c51c8a0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/packages/backend/src/utils/catalogUtils.spec.ts b/packages/backend/src/utils/catalogUtils.spec.ts
@@ -145,6 +145,8 @@ describe('sanitize', () => {
           readme: '',
           recommended: ['hf.instructlab.granite-7b-lab-GGUF', 'hf.instructlab.merlinite-7b-lab-GGUF'],
           backend: 'llama-cpp',
+          language: 'lang1',
+          frameworks: ['fw1'],
         },
       ],
       models: [
@@ -168,6 +170,8 @@ describe('sanitize', () => {
     expect(catalog.version).equals(CatalogFormat.CURRENT);
     expect(catalog.models[0].backend).toBeUndefined();
     expect(catalog.models[0].name).equals('Mistral-7B-Instruct-v0.3-Q4_K_M');
+    expect(catalog.recipes[0].language).equals('lang1');
+    expect(catalog.recipes[0].frameworks).toStrictEqual(['fw1']);
   });
 });
 

diff --git a/packages/backend/src/utils/catalogUtils.ts b/packages/backend/src/utils/catalogUtils.ts
@@ -144,6 +144,8 @@ export function sanitizeRecipe(recipe: unknown): Recipe {
       basedir: 'basedir' in recipe && typeof recipe.basedir === 'string' ? recipe.basedir : undefined,
       recommended: 'recommended' in recipe && isStringArray(recipe.recommended) ? recipe.recommended : undefined,
       backend: 'backend' in recipe && typeof recipe.backend === 'string' ? recipe.backend : undefined,
+      language: 'language' in recipe && typeof recipe.language === 'string' ? recipe.language : undefined,
+      frameworks: 'frameworks' in recipe && isStringArray(recipe.frameworks) ? recipe.frameworks : undefined,
     };
   throw new Error('invalid recipe format');
 }