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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
import vim
import re
import os
import configparser
if "PYTEST_VERSION" in os.environ:
from utils import *
context_py_imported = True
def merge_deep_recursive(target, source = {}):
source = source.copy()
for key, value in source.items():
if isinstance(value, dict):
target_child = target.setdefault(key, {})
merge_deep_recursive(target_child, value)
else:
target[key] = value
return target
def merge_deep(objects):
result = {}
for o in objects:
merge_deep_recursive(result, o)
return result
def is_deprecated_role_syntax(roles, role):
deprecated_sections = [
'options', 'options-complete', 'options-edit', 'options-chat',
'ui', 'ui-complete', 'ui-edit', 'ui-chat',
]
for section in deprecated_sections:
if f"{role}.{section}" in roles:
return True
return False
def load_roles_with_deprecated_syntax(roles, role):
prompt = dict(roles[role]).get('prompt', '')
return {
'role_default': {
'prompt': prompt,
'options': dict(roles.get(f"{role}.options", {})),
'ui': dict(roles.get(f"{role}.ui", {})),
},
'role_complete': {
'prompt': prompt,
'options': dict(roles.get(f"{role}.options-complete", {})),
'ui': dict(roles.get(f"{role}.ui-complete", {})),
},
'role_edit': {
'prompt': prompt,
'options': dict(roles.get(f"{role}.options-edit", {})),
'ui': dict(roles.get(f"{role}.ui-edit", {})),
},
'role_chat': {
'prompt': prompt,
'options': dict(roles.get(f"{role}.options-chat", {})),
'ui': dict(roles.get(f"{role}.ui-chat", {})),
},
}
def parse_role_section(role):
result = {}
for key in role.keys():
parts = key.split('.')
structure = parts[:-1]
primitive = parts[-1]
obj = result
for path in structure:
if not path in obj:
obj[path] = {}
obj = obj[path]
obj[primitive] = role.get(key)
return result
def load_role_config(role):
roles = read_role_files()
roles = dict(roles)
enhance_roles_with_custom_function(roles)
postfixes = ["", ".complete", ".edit", ".chat"]
if not any([f"{role}{postfix}" in roles for postfix in postfixes]):
raise Exception(f"Role `{role}` not found")
if is_deprecated_role_syntax(roles, role):
return load_roles_with_deprecated_syntax(roles, role)
return {
'role_default': parse_role_section(roles.get(role, {})),
'role_complete': parse_role_section(roles.get(f"{role}.complete", {})),
'role_edit': parse_role_section(roles.get(f"{role}.edit", {})),
'role_chat': parse_role_section(roles.get(f"{role}.chat", {})),
}
def parse_role_names(prompt):
chunks = re.split(r'[ :]+', prompt)
roles = []
for chunk in chunks:
if not chunk.startswith("/"):
break
roles.append(chunk)
return [raw_role[1:] for raw_role in roles]
def parse_prompt_and_role_config(user_instruction, command_type):
user_instruction = user_instruction.strip()
roles = parse_role_names(user_instruction)
last_role = roles[-1] if roles else ''
user_prompt = user_instruction[user_instruction.index(last_role) + len(last_role):].strip() # strip roles
role_results = [load_role_config(role) for role in [DEFAULT_ROLE_NAME] + roles]
parsed_role = merge_deep(role_results)
config = merge_deep([
parsed_role.get('role_default', {}),
parsed_role.get('role_' + command_type, {}),
])
role_prompt = config.get('prompt', '')
return user_prompt, config
def make_selection_prompt(user_selection, user_prompt, config_prompt, selection_boundary):
if not user_prompt and not config_prompt:
return user_selection
elif user_selection:
if selection_boundary and selection_boundary not in user_selection:
return f"{selection_boundary}\n{user_selection}\n{selection_boundary}"
else:
return user_selection
return ''
def make_prompt(config_prompt, user_prompt, user_selection, selection_boundary):
user_prompt = user_prompt.strip()
delimiter = ":\n" if user_prompt and user_selection else ""
user_selection = make_selection_prompt(user_selection, user_prompt, config_prompt, selection_boundary)
prompt = f"{user_prompt}{delimiter}{user_selection}"
if not config_prompt:
return prompt
delimiter = '' if prompt.startswith(':') else ':\n'
prompt = f"{config_prompt}{delimiter}{prompt}"
return prompt
def make_ai_context(params):
config_default = params['config_default']
config_extension = params['config_extension']
user_instruction = params['user_instruction']
user_selection = params['user_selection']
command_type = params['command_type']
user_prompt, role_config = parse_prompt_and_role_config(user_instruction, command_type)
final_config = merge_deep([config_default, config_extension, role_config])
selection_boundary = final_config['options']['selection_boundary']
config_prompt = final_config.get('prompt', '')
prompt = make_prompt(config_prompt, user_prompt, user_selection, selection_boundary)
return {
'config': final_config,
'prompt': prompt,
}
|
