diff options
| author | Martin Bielik <martin.bielik@instea.sk> | 2024-12-15 22:24:38 +0100 |
|---|---|---|
| committer | Martin Bielik <martin.bielik@instea.sk> | 2024-12-15 22:42:21 +0100 |
| commit | f26bee941bf9e5a5452ee0d75e7f2f2ea3c5216a (patch) | |
| tree | 3257544771acbcf32feeea89ff6fab991b09f404 /py/utils.py | |
| parent | 6bf889156f2ca8cecdc14ff8a882e4ed043d152e (diff) | |
| download | vim-ai-f26bee941bf9e5a5452ee0d75e7f2f2ea3c5216a.tar.gz | |
refactoring: make prompt in python
Diffstat (limited to 'py/utils.py')
| -rw-r--r-- | py/utils.py | 24 |
1 files changed, 0 insertions, 24 deletions
diff --git a/py/utils.py b/py/utils.py index fc888ab..ae32bc2 100644 --- a/py/utils.py +++ b/py/utils.py @@ -44,30 +44,6 @@ def load_api_key(config_token_file_path): return (api_key, org_id) -def strip_roles(prompt): - chunks = re.split(r'[ :]+', prompt) - roles = [] - for chunk in chunks: - if not chunk.startswith("/"): - break - roles.append(chunk) - if not roles: - return prompt - last_role = roles[-1] - return prompt[prompt.index(last_role) + len(last_role):].strip() - -def make_prompt(raw_prompt, role_prompt): - prompt = raw_prompt.strip() - prompt = strip_roles(prompt) - - if not role_prompt: - return prompt - - delim = '' if prompt.startswith(':') else ':\n' - prompt = role_prompt + delim + prompt - - return prompt - def make_config(config): options = config['options'] # initial prompt can be both a string and a list of strings, normalize it to list |