implement some coderabbit suggestions

Author: Mindful-Developer

backend/src/modules/ai/services/summarization.service.ts

@@ -9,7 +9,7 @@ import { env } from '../../../shared/config/environment';
 import { getEncoding } from 'js-tiktoken';
 
 export class SummarizationService {
-  private model: GenerativeModel;
+  private readonly defaultModelType: AIModel;
   private prompt: PromptTemplate;
 
   constructor() {
@@ -18,22 +18,20 @@ export class SummarizationService {
       throw new AppError(500, 'Gemini API key not found in environment variables. Please set GEMINI_API_KEY.', 'summarization.service');
     }
 
-    // Initialize Gemini model
-    const defaultModel = AIModel.Gemini20Flash;
-    const genAI = new GoogleGenerativeAI(env.ai.geminiKey);
-    this.model = genAI.getGenerativeModel({
-      model: defaultModel,
-      systemInstruction: SUMMARY_SYSTEM_PROMPT,
-      generationConfig: MODEL_CONFIGS[defaultModel],
-    });
-
-    // Initialize prompt template
+    this.defaultModelType = AIModel.Gemini20Flash;
     this.prompt = PromptTemplate.fromTemplate(SUMMARY_USER_PROMPT);
   }
 
   async summarize(request: SummarizationRequest): Promise<SummarizationResponse> {
     const startTime = Date.now();
-    const model = request.options?.model || AIModel.Gemini15Flash;
+    const modelType = request.options?.model || this.defaultModelType;
+
+    const genAI = new GoogleGenerativeAI(env.ai.geminiKey!);
+    const model = genAI.getGenerativeModel({
+      model: modelType,
+      systemInstruction: SUMMARY_SYSTEM_PROMPT,
+      generationConfig: MODEL_CONFIGS[modelType],
+    });
 
     try {
       // Format the prompt with system and user messages combined
@@ -42,7 +40,7 @@ export class SummarizationService {
       });
 
       // Call the model
-      const result = await this.model.generateContent(formattedPrompt);
+      const result = await model.generateContent(formattedPrompt);
       const summary = result.response.text();
 
       const endTime = Date.now();
@@ -57,7 +55,7 @@ export class SummarizationService {
       return {
         summary,
         metadata: {
-          model,
+          model: modelType,
           processingTime,
           tokenCount
         }

backend/tests/ai-module.spec.ts

@@ -1,7 +1,7 @@
 import axios from 'axios';
 import { describe, it, expect } from '@jest/globals';
 
-const API_URL = 'http://localhost:3000';
+const API_URL = process.env.BACKEND_URL || 'http://localhost:3000';
 
 describe('Summarization Route Tests', () => {
   it('should return a summary for valid input', async () => {

backend/tests/user-module.spec.ts

@@ -2,7 +2,7 @@ import axios from "axios";
 import { describe, expect, afterAll } from "@jest/globals";
 
 // Configuration for tests
-const API_URL = 'http://localhost:3000';
+const API_URL = process.env.BACKEND_URL || 'http://localhost:3000';
 const testUserCredentials = {
     firstName: 'Test',
     lastName: 'User',