feat(puzzle_api): implement model info tracking and debug logging

glowbom · Dec 28, 2023 · b6cc5f9 · b6cc5f9
1 parent fc69b60
commit b6cc5f9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/app/lib/ai.dart b/app/lib/ai.dart
@@ -1,6 +1,7 @@
 import 'dart:math';
 
 import 'package:glowby/hugging_face_api.dart';
+import 'package:glowby/pulze_ai_api.dart';
 
 import 'openai_api.dart';
 import 'timestamp.dart';
@@ -49,7 +50,7 @@ class Ai {
                   : OpenAI_API.model == 'huggingface'
                       ? HuggingFace_API.model()
                       : OpenAI_API.model == 'pulzeai'
-                          ? 'Powered by Pulze'
+                          ? '${PulzeAI_API.lastUsedModel()}'
                           : '';
       return [
         Message(

diff --git a/app/lib/pulze_ai_api.dart b/app/lib/pulze_ai_api.dart
@@ -31,6 +31,7 @@ class PulzeAI_API {
 ]
 ''';
   static String _model = 'pulze-v0';
+  static String _lastUsedModel = 'pulze-v0';
   static String _systemMessage = '';
   static bool _sendMessages = false;
   static const String _apiKeyKey = 'pulze_ai_api_key';
@@ -84,6 +85,10 @@ class PulzeAI_API {
     return _model;
   }
 
+  static String lastUsedModel() {
+    return _lastUsedModel;
+  }
+
   static void setModel(model) {
     _model = model;
     _secureStorage.write(key: _modelKey, value: _model);
@@ -141,6 +146,10 @@ class PulzeAI_API {
 
       if (response.statusCode == 200) {
         final responseBody = jsonDecode(utf8.decode(response.bodyBytes));
+        // Extracting the model information
+        String modelInfo = responseBody['metadata']['model']['model'];
+        _lastUsedModel = modelInfo;
+
         String receivedResponse =
             responseBody['choices'][0]['text'].toString().trim();
 
@@ -149,6 +158,7 @@ class PulzeAI_API {
 
         if (kDebugMode) {
           print('Generated Text: $generatedText');
+          print('Model Info: $modelInfo');
         }
 
         return generatedText;