run/tool.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

#!/usr/bin/env ruby

require 'json'
require 'pathname'

def parse_argv
  func_file = __FILE__
    func_data = nil

  if func_file.end_with?("tool.rb")
    func_file = ARGV[0]
    func_data = ARGV[1]
  else
    func_file = File.basename(func_file)
    func_data = ARGV[0]
  end

  func_file += '.rb' unless func_file.end_with?(".rb")

  [func_file, func_data]
end

def load_func(func_file)
  func_path = File.expand_path("../tools/rb/#{func_file}", __dir__)

  begin
    require func_path
  rescue LoadError
    puts "Invalid function: #{func_file}"
    exit 1
  end
end

func_file, func_data = parse_argv

if ENV["LLM_FUNCTION_ACTION"] == "declarate"
  load_func(func_file)
  puts JSON.pretty_generate(declarate)
else
  if func_data.nil?
    puts "No json data"
    exit 1
  end

  begin
    args = JSON.parse(func_data)
  rescue JSON::ParserError
    puts "Invalid json data"
    exit 1
  end

  load_func(func_file)
  execute(args)
end