Make sure to test the new stable models: hard coding -002 #53 #56

Open-Multi-Modal-Personal-Assistant · Sep 30, 2024 · 0cb3e2a · 0cb3e2a
1 parent c32c463
commit 0cb3e2a
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/lib/ai/service/ai_service.dart b/lib/ai/service/ai_service.dart
@@ -40,7 +40,7 @@ class AiService with FirebaseMixin, ToolsMixin {
     final preferences = GetIt.I.get<PreferencesService>();
     final modelType = preferences.fastLlmMode ? 'flash' : 'pro';
     return FirebaseVertexAI.instance.generativeModel(
-      model: 'gemini-1.5-$modelType',
+      model: 'gemini-1.5-$modelType-002',
       safetySettings: [
         SafetySetting(
           HarmCategory.harassment,
@@ -232,12 +232,10 @@ class AiService with FirebaseMixin, ToolsMixin {
             await fileRef.getDownloadURL();
           } catch (e) {
             // Not uploaded yet
-            debugPrint('Not uploaded yet? $e');
             await fileRef.putFile(mediumFile.file);
           }
 
           final fileUri = 'gs://$bucket/${fileRef.fullPath}';
-          debugPrint('Storage URI: $fileUri');
           parts.add(FileData(mediumFile.mimeType, fileUri));
         }
       }