Remove image from default

agixt/Interactions.py

@@ -644,6 +644,7 @@ async def run(
                     agent_settings["image_provider"] != "None"
                     and agent_settings["image_provider"] != ""
                     and agent_settings["image_provider"] != None
+                    and agent_settings["image_provider"] != "default"
                 ):
                     img_gen_prompt = f"Users message: {user_input} \n\n{'The user uploaded an image, one does not need generated unless the user is specifically asking.' if images else ''} **The assistant is acting as sentiment analysis expert and only responds with a concise YES or NO answer on if the user would like an image as visual or a picture generated. No other explanation is needed!**\nWould the user potentially like an image generated based on their message?\nAssistant: "
                     logging.info(f"[IMG] Decision maker prompt: {img_gen_prompt}")

agixt/providers/default.py

@@ -12,21 +12,18 @@
 # tts: gTTS
 # transcription: faster-whisper
 # translation: faster-whisper
-# image: huggingface (Requires API Key)
 
 
 class DefaultProvider:
     def __init__(
         self,
         AI_MODEL: str = "mixtral-8x7b",
-        HUGGINGFACE_API_KEY: str = "",
         **kwargs,
     ):
         self.AI_MODEL = AI_MODEL if AI_MODEL else "mixtral-8x7b"
         self.AI_TEMPERATURE = 0.7
         self.AI_TOP_P = 0.7
         self.MAX_TOKENS = 16000
-        self.HUGGINGFACE_API_KEY = HUGGINGFACE_API_KEY
         self.TRANSCRIPTION_MODEL = (
             "base"
             if "TRANSCRIPTION_MODEL" not in kwargs
@@ -44,7 +41,6 @@ def services():
             "tts",
             "transcription",
             "translation",
-            "image",
         ]
 
     async def inference(self, prompt, tokens: int = 0, images: list = []):
@@ -85,8 +81,3 @@ async def translate_audio(self, audio_path: str):
             audio_path=audio_path,
             translate=True,
         )
-
-    async def generate_image(self, prompt: str):
-        return await HuggingfaceProvider(
-            HUGGINGFACE_API_KEY=self.HUGGINGFACE_API_KEY
-        ).generate_image(prompt=prompt)