Merge pull request #117 from merefield/extensible_functions

FEATURE: ability to extend with functions from other plugins

lib/discourse_chatbot/bots/open_ai_bot_rag.rb

@@ -1,6 +1,27 @@
 # frozen_string_literal: true
 require "openai"
 
+BUILT_IN_FUNCTIONS = ["DiscourseChatbot::StockDataFunction",
+"DiscourseChatbot::GetCoordsOfLocationDescriptionFunction",
+"DiscourseChatbot::GetDistanceBetweenLocationsFunction",
+"DiscourseChatbot::ForumTopicSearchFromTopicLocationFunction",
+"DiscourseChatbot::ForumTopicSearchFromUserLocationFunction",
+"DiscourseChatbot::ForumTopicSearchFromLocationFunction",
+"DiscourseChatbot::ForumGetUserAddressFunction",
+"DiscourseChatbot::ForumUserSearchFromTopicLocationFunction",
+"DiscourseChatbot::ForumUserSearchFromUserLocationFunction",
+"DiscourseChatbot::ForumUserSearchFromLocationFunction",
+"DiscourseChatbot::ForumUserDistanceFromLocationFunction",
+"DiscourseChatbot::ForumSearchFunction",
+"DiscourseChatbot::PaintFunction",
+"DiscourseChatbot::VisionFunction",
+"DiscourseChatbot::WikipediaFunction",
+"DiscourseChatbot::WebSearchFunction",
+"DiscourseChatbot::WebCrawlerFunction",
+"DiscourseChatbot::NewsFunction",
+"DiscourseChatbot::EscalateToStaffFunction",
+"DiscourseChatbot::CalculatorFunction"]
+
 module ::DiscourseChatbot
 
   class OpenAiBotRag < OpenAIBotBase
@@ -93,6 +114,12 @@ def merge_functions(opts)
       functions << web_search_function if !(SiteSetting.chatbot_serp_api_key.blank? && SiteSetting.chatbot_jina_api_token.blank?)
       functions << stock_data_function if !SiteSetting.chatbot_marketstack_key.blank?
 
+      if ::DiscourseChatbot::Function.descendants.count > BUILT_IN_FUNCTIONS.count
+        ::DiscourseChatbot::Function.descendants.each do |func|
+          functions << func.new if !BUILT_IN_FUNCTIONS.include?(func.to_s)
+        end
+      end
+
       @functions = parse_functions(functions)
       @tools = @functions.map { |func| { "type": "function", "function": func } }
       @func_mapping = create_func_mapping(functions)

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: 1.0.0
+# version: 1.0.1
 # authors: merefield
 # url: https://github.com/merefield/discourse-chatbot