mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-03 07:49:30 +00:00
d59ed1091b
This change-set connects GPT based chat with the forum it runs on. Allowing it to perform search, lookup tags and categories and summarize topics. The integration is currently restricted to public portions of the forum. Changes made: - Do not run ai reply job for small actions - Improved composable system prompt - Trivial summarizer for topics - Image generator - Google command for searching via Google - Corrected trimming of posts raw (was replacing with numbers) - Bypass of problem specs The feature works best with GPT-4 --------- Co-authored-by: Roman Rizzi <rizziromanalejandro@gmail.com>
62 lines
1.5 KiB
Ruby
62 lines
1.5 KiB
Ruby
#frozen_string_literal: true
|
|
|
|
module DiscourseAi::AiBot::Commands
|
|
class GoogleCommand < Command
|
|
class << self
|
|
def name
|
|
"google"
|
|
end
|
|
|
|
def desc
|
|
"!google SEARCH_QUERY - will search using Google (supports all Google search operators)"
|
|
end
|
|
end
|
|
|
|
def result_name
|
|
"results"
|
|
end
|
|
|
|
def description_args
|
|
{
|
|
count: @last_num_results || 0,
|
|
query: @last_query || "",
|
|
url: "https://google.com/search?q=#{CGI.escape(@last_query || "")}",
|
|
}
|
|
end
|
|
|
|
def process(search_string)
|
|
@last_query = search_string
|
|
api_key = SiteSetting.ai_google_custom_search_api_key
|
|
cx = SiteSetting.ai_google_custom_search_cx
|
|
query = CGI.escape(search_string)
|
|
uri =
|
|
URI("https://www.googleapis.com/customsearch/v1?key=#{api_key}&cx=#{cx}&q=#{query}&num=10")
|
|
body = Net::HTTP.get(uri)
|
|
|
|
parse_search_json(body).to_s
|
|
end
|
|
|
|
def parse_search_json(json_data)
|
|
parsed = JSON.parse(json_data)
|
|
results = parsed["items"]
|
|
|
|
@last_num_results = parsed.dig("searchInformation", "totalResults").to_i
|
|
|
|
formatted_results = []
|
|
|
|
results.each do |result|
|
|
formatted_result = {
|
|
title: result["title"],
|
|
link: result["link"],
|
|
snippet: result["snippet"],
|
|
displayLink: result["displayLink"],
|
|
formattedUrl: result["formattedUrl"],
|
|
}
|
|
formatted_results << formatted_result
|
|
end
|
|
|
|
formatted_results
|
|
end
|
|
end
|
|
end
|