Added ClaudeAPI

classes/DataSet_OpenAIChatImage.py

@@ -3,6 +3,7 @@ import io
 from PIL import Image
 import numpy as np
 from openai import OpenAI
+import os
 
 class DataSet_OpenAIChatImage:
 
@@ -18,7 +19,7 @@ class DataSet_OpenAIChatImage:
                 "prompt": ("STRING", {"multiline": True, "default": ""}),
                 "model": (["gpt-4o","gpt-4", "gpt-4-32k", "gpt-3.5-turbo", "gpt-4-0125-preview", "gpt-4-turbo-preview", "gpt-4-1106-preview", "gpt-4-0613"], {"default": "gpt-4o"}),
                 "api_url": ("STRING", {"multiline": False, "default": "https://api.openai.com/v1"}),
-                "api_key": ("STRING", {"multiline": False}),
+                # "api_key": ("STRING", {"multiline": False}),
                 "token_length": ("INT", {"default": 1024})
             }
         }
@@ -35,9 +36,10 @@ class DataSet_OpenAIChatImage:
         img.save(buffered, format="PNG")
         return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
-    def generate(self, image, image_detail, model, api_url, api_key, prompt, token_length):
+    def generate(self, image, image_detail, model, api_url, prompt, token_length):
         try:
 
+            api_key = os.environ.get("OPENAI_API_KEY")
             ai = OpenAI(api_key=api_key, base_url=api_url)
             base64img = self.to_base64(image)
             if not api_key: