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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
import vim
import os
from config import make_config
dirname = os.path.dirname(__file__)
def default_eval_mock(cmd):
match cmd:
case 'g:vim_ai_debug_log_file':
return '/tmp/vim_ai_debug.log'
case 'g:vim_ai_roles_config_file':
return dirname + '/resources/roles.ini'
case _:
return None
default_config = {
"options": {
"model": "gpt-4o",
"endpoint_url": "https://api.openai.com/v1/chat/completions",
"max_tokens": "0",
"max_completion_tokens": "0",
"temperature": "1",
"request_timeout": "20",
"stream": "1",
"enable_auth": "1",
"token_file_path": "",
"selection_boundary": "",
"initial_prompt": "You are a general assistant.",
},
"ui": {
"open_chat_command": "preset_below",
"scratch_buffer_keep_open": "0",
"populate_options": "0",
"code_syntax_enabled": "1",
"paste_mode": "1",
},
}
def make_input_mock(mocker, input_options):
def eval_mock(cmd):
if cmd == 'l:input':
return input_options
return default_eval_mock(cmd)
mocker.patch('vim.eval', eval_mock)
def test_default_config(mocker):
make_input_mock(mocker, {
'config_default': default_config,
'config_extension': {},
'instruction': 'hello',
'command_type': 'chat',
})
command_spy = mocker.spy(vim, "command")
actual_output = make_config('l:input', 'l:output')
expected_output = {
'config': default_config,
'role_prompt': '',
}
command_spy.assert_called_once_with(f"let l:output={expected_output}")
assert expected_output == actual_output
def test_param_config(mocker):
make_input_mock(mocker, {
'config_default': default_config,
'config_extension': {
'options': {
'max_tokens': '1000',
},
},
'instruction': 'hello',
'command_type': 'chat',
})
actual_config = make_config('l:input', 'l:output')['config']
assert '1000' == actual_config['options']['max_tokens']
assert 'gpt-4o' == actual_config['options']['model']
def test_role_config(mocker):
make_input_mock(mocker, {
'config_default': default_config,
'config_extension': {},
'instruction': '/test-role-simple',
'command_type': 'chat',
})
config = make_config('l:input', 'l:output')
actual_config = config['config']
actual_role_prompt = config['role_prompt']
assert 'o1-preview' == actual_config['options']['model']
assert 'simple role prompt' == actual_role_prompt
def test_role_config_different_commands(mocker):
make_input_mock(mocker, {
'config_default': default_config,
'config_extension': {},
'instruction': '/test-role hello',
'command_type': 'chat',
})
config = make_config('l:input', 'l:output')
actual_config = config['config']
actual_role_prompt = config['role_prompt']
assert 'model-common' == actual_config['options']['model']
assert 'https://localhost/chat' == actual_config['options']['endpoint_url']
assert '0' == actual_config['ui']['paste_mode']
assert 'preset_tab' == actual_config['ui']['open_chat_command']
assert '' == actual_role_prompt
make_input_mock(mocker, {
'config_default': default_config,
'config_extension': {},
'instruction': '/test-role hello',
'command_type': 'complete',
})
config = make_config('l:input', 'l:output')
actual_config = config['config']
actual_role_prompt = config['role_prompt']
assert 'model-common' == actual_config['options']['model']
assert 'https://localhost/complete' == actual_config['options']['endpoint_url']
assert '0' == actual_config['ui']['paste_mode']
assert '' == actual_role_prompt
def test_multiple_role_configs(mocker):
make_input_mock(mocker, {
'config_default': default_config,
'config_extension': {},
'instruction': '/test-role /test-role-simple hello',
'command_type': 'chat',
})
config = make_config('l:input', 'l:output')
actual_config = config['config']
actual_role_prompt = config['role_prompt']
assert 'o1-preview' == actual_config['options']['model']
assert 'https://localhost/chat' == actual_config['options']['endpoint_url']
assert 'simple role prompt' == actual_role_prompt
|
