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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
# frozen_string_literal: true
require 'yaml'
require_relative '../../../../logic/providers/openai/tools'
RSpec.describe NanoBot::Logic::OpenAI::Tools do
context 'tools' do
let(:cartridge) { load_symbolized('cartridges/tools.yml') }
context 'adapt' do
it 'adapts to OpenAI expected format' do
expect(described_class.adapt(cartridge[:tools][0])).to eq(
{ type: 'function',
function: {
name: 'what-time-is-it',
description: 'Returns the current date and time for a given timezone.',
parameters: {
type: 'object',
properties: {
timezone: {
type: 'string',
description: 'A string representing the timezone that should be used to provide a datetime, following the IANA (Internet Assigned Numbers Authority) Time Zone Database. Examples are "Asia/Tokyo" and "Europe/Paris".'
}
}, required: ['timezone']
}
} }
)
expect(described_class.adapt(cartridge[:tools][1])).to eq(
{ type: 'function',
function: {
name: 'get-current-weather',
description: 'Get the current weather in a given location.',
parameters: {
type: 'object',
properties: {
location: { type: 'string' },
unit: { type: 'string' }
}
}
} }
)
expect(described_class.adapt(cartridge[:tools][2])).to eq(
{ type: 'function',
function: {
name: 'sh',
description: "It has access to computer users' data and can be used to run shell commands, similar to those in a Linux terminal, to extract information. Please be mindful and careful to avoid running dangerous commands on users' computers.",
parameters: {
type: 'object',
properties: {
command: {
type: 'array',
description: 'An array of strings that represents a shell command along with its arguments or options. For instance, `["df", "-h"]` executes the `df -h` command, where each array element specifies either the command itself or an associated argument/option.',
items: { type: 'string' }
}
}
}
} }
)
expect(described_class.adapt(cartridge[:tools][3])).to eq(
{ type: 'function',
function: {
name: 'clock',
description: 'Returns the current date and time.',
parameters: { type: 'object', properties: {} }
} }
)
end
end
context 'prepare' do
let(:tools) { load_symbolized('providers/openai/tools.yml') }
it 'prepare tools to be executed' do
expect(described_class.prepare(cartridge[:tools], tools)).to eq(
[{ id: 'call_XYZ',
name: 'get-current-weather',
label: 'get-current-weather',
type: 'function',
parameters: { 'location' => 'Tokyo, Japan' },
source: { fennel: "(let [{:location location :unit unit} parameters]\n (.. \"Here is the weather in \" location \", in \" unit \": 35.8°C.\"))\n" } },
{ id: 'call_ZYX',
name: 'what-time-is-it',
label: 'what-time-is-it',
type: 'function',
parameters: {},
source: { fennel: "(os.date)\n" } }]
)
end
end
end
end
|
