gem "openai-toolable"

gem install openai-toolable

require "openai/toolable"
require "openai"

# Thiết lập API Key
OPENAI_API_KEY = ENV["OPENAI_API_KEY"]

# Định nghĩa một tool với tham số yêu cầu
weather_tool = Openai::Toolable::ToolFactory.build(
  name: "get_weather",
  type: "function",
  description: "Get the current weather in a given location",
  parameters: [
    { name: "location", type: :string, description: "The city and state, e.g. San Francisco, CA", required: true },
    { name: "unit", type: :string, description: "The unit of temperature, e.g. celsius or fahrenheit", required: true }
  ]
)

# Tạo handler cho tool và đăng ký tool
tool_handler = Openai::Toolable::ToolHandler.new
tool_handler.register(
  name: "get_weather",
  lambda: ->(location:, unit:) { puts "Getting the weather in #{location} (#{unit})..." }
)

# Khởi tạo client OpenAI và gọi API GPT với công cụ
client = OpenAI::Client.new(api_key: OPENAI_API_KEY)

begin
  response = client.chat.completions.create(
    model: "gpt-4o-mini",
    messages: [{ role: "user", content: "What's the weather like in Boston?" }],
    tools: [weather_tool.to_json],
    tool_choice: "auto"
  )
  # Xử lý phản hồi và gọi tool tương ứng
  tool_handler.handle(response: response)
rescue StandardError => e
  puts "An error occurred: #{e.message}"
end