From 03d3b19686482f091967b63fdfb6a31eaa69f6a5 Mon Sep 17 00:00:00 2001
From: Ankit Mahato <ankmahato@gmail.com>
Date: Sun, 31 Aug 2025 10:57:53 +0530
Subject: [PATCH] fixes

---
 .../test/utils.dart/ai_request_utils_test.dart | 18 ++++++++++++------
 .../test/utils.dart/available_models_test.dart |  8 ++++----
 .../test/utils.dart/model_manager_test.dart    |  7 -------
 3 files changed, 16 insertions(+), 17 deletions(-)

diff --git a/packages/genai/test/utils.dart/ai_request_utils_test.dart b/packages/genai/test/utils.dart/ai_request_utils_test.dart
index 080a3b99..1cf55e59 100644
--- a/packages/genai/test/utils.dart/ai_request_utils_test.dart
+++ b/packages/genai/test/utils.dart/ai_request_utils_test.dart
@@ -1,20 +1,26 @@
-import 'dart:async';
-import 'dart:convert';
-
+import 'dart:io' show Platform;
 import 'package:flutter_test/flutter_test.dart';
 import 'package:genai/models/ai_request_model.dart';
 import 'package:genai/interface/consts.dart';
 import 'package:genai/utils/ai_request_utils.dart';
-import 'package:better_networking/better_networking.dart';
 
-const kTestingAPIKey = "XXXXXXXXXXXX";
+const kGeminiApiKey = 'GEMINI_API_KEY';
 
 void main() {
   group('ai_request_utils', () {
     test(
       'executeGenAIRequest should return formatted output on success',
       () async {
-        const model = AIRequestModel(
+        Map<String, String> envVars = Platform.environment;
+        String? kTestingAPIKey;
+        if (envVars.containsKey(kGeminiApiKey)) {
+          kTestingAPIKey = envVars[kGeminiApiKey];
+        } else {
+          throw ArgumentError(
+            '$kGeminiApiKey should be available as an environment variable.',
+          );
+        }
+        var model = AIRequestModel(
           modelApiProvider: ModelAPIProvider.gemini,
           model: 'gemini-2.0-flash',
           url: kGeminiUrl,
diff --git a/packages/genai/test/utils.dart/available_models_test.dart b/packages/genai/test/utils.dart/available_models_test.dart
index c46df7a7..1a6b5d0b 100644
--- a/packages/genai/test/utils.dart/available_models_test.dart
+++ b/packages/genai/test/utils.dart/available_models_test.dart
@@ -31,13 +31,13 @@ void main() {
     });
 
     test('map getter returns map of providers', () {
-      final provider = AIModelProvider(
+      const provider = AIModelProvider(
         providerId: ModelAPIProvider.openai,
         providerName: "OpenAI",
-        models: [const Model(id: "gpt-4", name: "GPT-4")],
+        models: [Model(id: "gpt-4", name: "GPT-4")],
       );
 
-      final available = AvailableModels(
+      const available = AvailableModels(
         version: 1.0,
         modelProviders: [provider],
       );
@@ -56,7 +56,7 @@ void main() {
           providerName: "OpenAI",
         );
 
-        const model = const Model(id: "gpt-4", name: "GPT-4");
+        const model = Model(id: "gpt-4", name: "GPT-4");
         final req = provider.toAiRequestModel(model: model);
 
         expect(req?.model, "gpt-4");
diff --git a/packages/genai/test/utils.dart/model_manager_test.dart b/packages/genai/test/utils.dart/model_manager_test.dart
index effb8683..32939647 100644
--- a/packages/genai/test/utils.dart/model_manager_test.dart
+++ b/packages/genai/test/utils.dart/model_manager_test.dart
@@ -1,4 +1,3 @@
-import 'dart:convert';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:genai/models/available_models.dart';
 import 'package:genai/utils/model_manager.dart';
@@ -11,12 +10,6 @@ void main() {
     });
 
     test('fetchInstalledOllamaModels parses response', () async {
-      final body = jsonEncode({
-        "models": [
-          {"model": "mistral", "name": "Mistral"},
-          {"model": "llama2", "name": "LLaMA 2"},
-        ],
-      });
       final result = await ModelManager.fetchInstalledOllamaModels();
       expect(result, isNotNull);
     });