summaryrefslogtreecommitdiff
path: root/py/utils.py
diff options
context:
space:
mode:
authorMartin Bielik <martin.bielik@instea.sk>2024-12-15 22:24:38 +0100
committerMartin Bielik <martin.bielik@instea.sk>2024-12-15 22:42:21 +0100
commitf26bee941bf9e5a5452ee0d75e7f2f2ea3c5216a (patch)
tree3257544771acbcf32feeea89ff6fab991b09f404 /py/utils.py
parent6bf889156f2ca8cecdc14ff8a882e4ed043d152e (diff)
downloadvim-ai-f26bee941bf9e5a5452ee0d75e7f2f2ea3c5216a.tar.gz
refactoring: make prompt in python
Diffstat (limited to 'py/utils.py')
-rw-r--r--py/utils.py24
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