Add handling

Josh-XT · May 1, 2024 · 203a776 · 203a776
1 parent d5c860a
commit 203a776
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/agixt/Interactions.py b/agixt/Interactions.py
@@ -667,10 +667,15 @@ async def run(
                             image_generation_prompt = image_generation_prompt.split(
                                 "```"
                             )[0]
-                        generated_image = await self.agent.generate_image(
-                            prompt=image_generation_prompt
-                        )
-                        self.response = f"{self.response}\n\n![Image generated by {self.agent_name}]({generated_image})"
+                        try:
+                            generated_image = await self.agent.generate_image(
+                                prompt=image_generation_prompt
+                            )
+                            self.response = f"{self.response}\n\n![Image generated by {self.agent_name}]({generated_image})"
+                        except:
+                            logging.warning(
+                                f"[IMG] Failed to generate image for prompt: {image_generation_prompt}"
+                            )
             log_interaction(
                 agent_name=self.agent_name,
                 conversation_name=conversation_name,

diff --git a/agixt/providers/ezlocalai.py b/agixt/providers/ezlocalai.py
@@ -180,7 +180,7 @@ async def text_to_speech(self, text: str):
     async def generate_image(self, prompt: str) -> str:
         filename = f"{uuid.uuid4()}.png"
         image_path = f"./WORKSPACE/{filename}"
-        openai.base_url = self.API_URI if self.API_URI else "https://api.openai.com/v1/"
+        openai.base_url = self.API_URI
         openai.api_key = self.EZLOCALAI_API_KEY
         response = openai.images.generate(
             prompt=prompt,