@botpress/cognitive 0.1.44 → 0.1.46

package/refresh-models.ts

@@ -3,14 +3,13 @@ import * as fs from 'fs'
 import * as path from 'path'
 import { Model } from 'src/schemas.gen'
 
-const builtInModels = ['auto', 'best', 'fast', 'reasoning', 'cheapest', 'balance']
+const builtInModels = ['auto', 'best', 'fast']
 const filteredLifecycles = ['deprecated', 'discontinued']
-const filteredTags = ['deprecated']
 
 const modelsListPath = path.resolve(__dirname, 'src/cognitive-v2', 'models.ts')
 const typesPath = path.resolve(__dirname, 'src/cognitive-v2', 'types.ts')
 
-type RemoteModel = Model & { lifecycle?: string }
+type RemoteModel = Model & { lifecycle?: string; aliases?: string[] }
 
 const toRef = (m: RemoteModel | string | null | undefined): string | null => {
   if (!m) return null
@@ -36,17 +35,6 @@ async function main(): Promise<void> {
     },
   })
 
-  const uniqueTags = [
-    ...new Set(
-      models
-        .map((m) => m.tags)
-        .flat()
-        .filter((t) => !filteredTags.includes(t))
-    ),
-  ]
-
-  console.log(`Unique tags: ${uniqueTags.join(', ')}`)
-
   const modelsObj = models.reduce((acc, m) => ((acc[m.id] = m), acc), {} as Record<string, RemoteModel>)
 
   const defaultModel: RemoteModel = {
@@ -62,7 +50,6 @@ async function main(): Promise<void> {
   const newFile = `import { Model } from 'src/schemas.gen'\n
 export type RemoteModel = Model & { aliases?: string[]; lifecycle: 'live' | 'beta' | 'deprecated' | 'discontinued' }\n
 export const models: Record<string, RemoteModel>  = ${JSON.stringify(modelsObj, null, 2)}\n
-export const knownTags = [${[...builtInModels, ...uniqueTags].map((t) => `'${t}'`).join(', ')}]\n
 export const defaultModel: RemoteModel = ${JSON.stringify(defaultModel, undefined, 2)}
 `
 
@@ -72,6 +59,12 @@ export const defaultModel: RemoteModel = ${JSON.stringify(defaultModel, undefine
 
   const withoutDeprecated = models.filter((m) => !filteredLifecycles.includes(m.lifecycle))
   const refs = Array.from(new Set(withoutDeprecated.map(toRef).filter(Boolean))).sort((a, b) => a.localeCompare(b))
+  const aliases = models.flatMap((m) =>
+    (m.aliases || []).map((a) => {
+      const [provider] = m.id.split(':')
+      return `${provider}:${a}`
+    })
+  )
 
   const content = fs.readFileSync(typesPath, 'utf8')
 
@@ -85,7 +78,7 @@ export const defaultModel: RemoteModel = ${JSON.stringify(defaultModel, undefine
     throw new Error('Could not locate Models union block in models.ts')
   }
 
-  const items = [...builtInModels, ...uniqueTags, ...refs].map((r) => `  | '${r}'`)
+  const items = [...builtInModels, ...refs, ...aliases].map((r) => `  | '${r}'`)
   const unionBlock = ['type Models =', ...items, '  | ({} & string)', ''].join('\n')
 
   const nextContent = content.slice(0, startIdx) + unionBlock + content.slice(endIdx)