FIX: improve url identification for paint function response

merefield · Jun 19, 2024 · 19bf579 · 19bf579
1 parent 395e9f6
commit 19bf579
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/lib/discourse_chatbot/bots/open_ai_bot_rag.rb b/lib/discourse_chatbot/bots/open_ai_bot_rag.rb
@@ -313,7 +313,16 @@ def legal_urls?(res, post_ids_found, topic_ids_found)
 
     private
 
-    def image_url?(url)
+    def image_url?(string)
+      # Regular expression to find URLs
+      url_regex = /\bhttps?:\/\/[^\s]+/
+
+      # Check if the string contains more than one URL or other text
+      urls = string.scan(url_regex)
+      return false unless urls.length == 1 && string.strip == urls[0]
+
+      # Proceed with the existing logic if only one URL is found
+      url = urls[0]
       image_extensions = %w[.jpg .jpeg .png .gif .bmp .tiff .webp]
 
       uri = URI.parse(url)

diff --git a/plugin.rb b/plugin.rb
@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 # name: discourse-chatbot
 # about: a plugin that allows you to have a conversation with a configurable chatbot in Discourse Chat, Topics and Private Messages
-# version: 0.9.35
+# version: 0.9.36
 # authors: merefield
 # url: https://github.com/merefield/discourse-chatbot