Skip to content

Instantly share code, notes, and snippets.

@itsderek23
Created May 8, 2023 15:32
Show Gist options
  • Save itsderek23/7fabe8ec9d1874593ccec015b0f92b5c to your computer and use it in GitHub Desktop.
Save itsderek23/7fabe8ec9d1874593ccec015b0f92b5c to your computer and use it in GitHub Desktop.
A boxcar that returns a Google answer box as JSON
# frozen_string_literal: true
require 'google_search_results'
require 'boxcars/boxcar'
require 'boxcars/result'
class GoogleAnswerBox < Boxcars::Boxcar
# The description of this boxcar. Used in the prompt to inform the engine what
# this boxcar can do.
ANSWERBOXDESC = "useful for when you need to answer questions that require realtime data." \
"You should ask targeted questions"
# implements a boxcar that uses the Google SerpAPI to get the answer box data from search results.
# @param name [String] The name of the boxcar. Defaults to classname.
# @param description [String] A description of the boxcar. Defaults to SERPDESC.
# @param serpapi_api_key [String] The API key to use for the SerpAPI. Defaults to Boxcars.configuration.serpapi_api_key.
def initialize(name: "AnswerBox", description: ANSWERBOXDESC, serpapi_api_key: nil)
super(name: name, description: description)
api_key = Boxcars.configuration.serpapi_api_key(serpapi_api_key: serpapi_api_key)
::GoogleSearch.api_key = api_key
end
# Get an answer from Google using the SerpAPI.
# @param question [String] The question to ask Google.
# @return [String] The answer to the question.
def run(question)
search = ::GoogleSearch.new(q: question)
Boxcars::Result.from_text(find_answer(search.get_hash))
end
private
def find_answer(res)
# :answer_box=>{:type=>"local_time", :result=>"8:02 AM", :extensions=>["Sunday, May 7, 2023 (MDT)", "Time in Denver, CO"]}
raise Error, "Got error from SerpAPI: {res[:error]}" if res[:error]
# might be sending results outside of the answer box ...
# see a result for {"type":"organic_result"
return "No answer found" unless res[:answer_box]
return res.dig(:answer_box).to_json[0..1000].gsub("%","%%")
end
end
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment