Update main.ts

main.ts

@@ -38,15 +38,22 @@ class GoogleAIService {
 
   constructor() {
     this.apiKeys = [];
-    this.apiKeys = Deno.env.get(`GOOGLE_AI_KEYS`).split(',').map(s => s.trim());
+    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++;
+    }
+
     if (this.apiKeys.length === 0) {
-      throw new Error("No Google AI API keys found in environment variables (e.g., GOOGLE_AI_KEYS)");
+      throw new Error("No Google AI API keys found in environment variables (e.g., GOOGLE_AI_KEY_1, GOOGLE_AI_KEY)");
     }
   }
 
   private getNextApiKey(): string {
     const key = this.apiKeys[this.currentKeyIndex];
-    console.log(key)
     this.currentKeyIndex = (this.currentKeyIndex + 1) % this.apiKeys.length;
     return key;
   }
@@ -122,8 +129,8 @@ class GoogleAIService {
   }
 
   public isVisionModel = (modelName: string): boolean => modelName.toLowerCase().includes('vision') || modelName.toLowerCase().includes('pro');
-  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 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 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');
 
@@ -456,19 +463,22 @@ 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");
@@ -495,7 +505,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 += `![image](${result.imageUrl})`;
+        if (result.imageUrl) response += `${inputImage ? 'Edited' : 'Generated'} image:\\\\n${result.imageUrl}`;
         return response || `Image processing complete.`;
       } catch (error) {
         return `Image processing failed: ${error.message}`;
@@ -591,22 +601,17 @@ 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[] = [];
@@ -619,6 +624,7 @@ class OpenAICompatibleServer {
           }
         });
       }
+      
       let responseText: string;
 
       // Routing logic based on keywords and content types