Update main.ts

main.ts

@@ -38,22 +38,15 @@ class GoogleAIService {
 
   constructor() {
     this.apiKeys = [];
-    let i = 1;
-    while (true) {
-      const key = Deno.env.get(`GOOGLE_AI_KEY_${i}`) ||
-                  (i === 1 ? Deno.env.get("GOOGLE_AI_KEY") : null);
-      if (!key) break;
-      this.apiKeys.push(key);
-      i++;
-    }
-
+    this.apiKeys = Deno.env.get(`GOOGLE_AI_KEYS`).split(',').map(s => s.trim());
     if (this.apiKeys.length === 0) {
-      throw new Error("No Google AI API keys found in environment variables (e.g., GOOGLE_AI_KEY_1, GOOGLE_AI_KEY)");
+      throw new Error("No Google AI API keys found in environment variables (e.g., GOOGLE_AI_KEYS)");
     }
   }
 
   private getNextApiKey(): string {
     const key = this.apiKeys[this.currentKeyIndex];
+    console.log(key)
     this.currentKeyIndex = (this.currentKeyIndex + 1) % this.apiKeys.length;
     return key;
   }
@@ -129,8 +122,8 @@ class GoogleAIService {
   }
 
   public isVisionModel = (modelName: string): boolean => modelName.toLowerCase().includes('vision') || modelName.toLowerCase().includes('pro');
-  public isImageGenerationModel = (modelName: string): boolean => modelName.includes('image-generation') || modelName === 'gemini-2.0-flash-preview-image-generation';
-  public isImageEditingModel = (modelName: string): boolean => modelName.includes('image-generation') || modelName === 'gemini-2.0-flash-preview-image-generation';
+  public isImageGenerationModel = (modelName: string): boolean => modelName.includes('image') || modelName === 'gemini-2.0-flash-preview-image-generation' || modelName === 'gemini-2.5-flash-image-preview';
+  public isImageEditingModel = (modelName: string): boolean => modelName.includes('image') || modelName === 'gemini-2.0-flash-preview-image-generation' || modelName === 'gemini-2.5-flash-image-preview';
   public isDocumentModel = (modelName: string): boolean => modelName.toLowerCase().includes('gemini-1.5') || modelName.toLowerCase().includes('pro') || modelName.toLowerCase().includes('flash');
   public isTTSModel = (modelName: string): boolean => modelName.toLowerCase().includes('tts');
 
@@ -463,22 +456,19 @@ class GoogleAIService {
       tools: [{ googleSearch: {} }],
       generationConfig: { temperature: 0.7, maxOutputTokens: maxTokens || 8192 }
     };
-
     const response = await fetch(
       `https://generativelanguage.googleapis.com/v1beta/${fullModelName}:generateContent?key=${apiKey}`,
       { method: "POST", headers: { "Content-Type": "application/json" }, body: JSON.stringify(requestBody) }
     );
-
     if (!response.ok) {
       console.warn(`Google Search API failed: ${response.status}, trying alternative.`);
       return await this.generateContentWithSearchPrompt(messages, modelName, maxTokens);
     }
-
     const data = await response.json();
     if (!data.candidates || data.candidates.length === 0) {
       return await this.generateContentWithSearchPrompt(messages, modelName, maxTokens);
     }
-
+    
     const candidate = data.candidates[0];
     if (candidate.finishReason === "SAFETY") {
       throw new Error("Response blocked due to safety filters");
@@ -505,7 +495,7 @@ class GoogleAIService {
         const result = await this.generateOrEditImageWithGemini(prompt, modelName, inputImage);
         let response = "";
         if (result.text) response += result.text + "\\\\n\\\\n";
-        if (result.imageUrl) response += `${inputImage ? 'Edited' : 'Generated'} image:\\\\n${result.imageUrl}`;
+        if (result.imageUrl) response += `![image](${result.imageUrl})`;
         return response || `Image processing complete.`;
       } catch (error) {
         return `Image processing failed: ${error.message}`;
@@ -601,17 +591,22 @@ class OpenAICompatibleServer {
       const stream = body.stream || false;
       const maxTokens = body.max_tokens || 1048576;
       console.log(`Request for model: ${requestedModel}, stream: ${stream}, max_tokens: ${maxTokens}`);
-
       const lastMessage = body.messages[body.messages.length - 1];
       const content = typeof lastMessage.content === "string"
         ? lastMessage.content
         : (Array.isArray(lastMessage.content) ? lastMessage.content.map(p => p.text || "").join(" ") : "");
-
+      if (content == 'ping'){
+        const responsePayload = {
+          id: `chatcmpl-${Date.now()}`, object: "chat.completion", created: Math.floor(Date.now() / 1000), model: requestedModel,
+          choices: [{ index: 0, message: { role: "assistant", content: "pong" }, finish_reason: "stop" }],
+          usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 }
+        };
+        return new Response(JSON.stringify(responsePayload), { headers: { "Content-Type": "application/json" } });
+      }
       const hasDocument = body.messages.some(msg =>
         Array.isArray(msg.content) &&
         msg.content.some(part => part.type === "document" || this.isDocumentContent(part.document?.url))
       );
-
       const hasImages = body.messages.some(msg => Array.isArray(msg.content) && msg.content.some(part => part.type === "image_url"));
       
       let inputImages: any[] = [];
@@ -624,7 +619,6 @@ class OpenAICompatibleServer {
           }
         });
       }
-      
       let responseText: string;
 
       // Routing logic based on keywords and content types