move to contained directory

This commit is contained in:
Bruce MacDonald
2023-06-27 12:08:52 -04:00
parent 41419f7577
commit 1fcf31b8c4
7 changed files with 279 additions and 177 deletions

View File

@@ -1,20 +0,0 @@
from difflib import SequenceMatcher
import json
with open("model_prompts.json", "r") as f:
model_prompts = json.load(f)
def template(model, prompt):
max_ratio = 0
closest_key = ""
model_name = model.lower()
# Find the specialized prompt with the closest name match
for key in model_prompts.keys():
ratio = SequenceMatcher(None, model_name, key).ratio()
if ratio > max_ratio:
max_ratio = ratio
closest_key = key
# Return the value of the closest match
p = model_prompts.get(closest_key) # TODO: provide a better default template
return p.format(prompt=prompt)