Fix webapp build avoiding lodash omit (#600)

latitude-dev · Nov 13, 2024 · 4fab0d8 · 4fab0d8
1 parent 4a19bdf
commit 4fab0d8
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/packages/core/src/services/ai/providers/models/index.ts b/packages/core/src/services/ai/providers/models/index.ts
@@ -1,5 +1,3 @@
-import { omit } from 'lodash-es'
-
 import { ProviderApiKey } from '../../../../schema/types'
 
 export enum Providers {
@@ -111,7 +109,11 @@ export function listModelsForProvider({
   if (!models) return {}
 
   if (name && name === latitudeProvider) {
-    return omit(models, DEFAULT_PROVIDER_UNSUPPORTED_MODELS)
+    return Object.fromEntries(
+      Object.entries(models).filter(
+        ([key]) => !DEFAULT_PROVIDER_UNSUPPORTED_MODELS.includes(key),
+      ),
+    )
   }
 
   return models