mirror of
https://github.com/dogkeeper886/ollama37.git
synced 2025-12-10 15:57:04 +00:00
20 lines
620 B
Python
20 lines
620 B
Python
from os import path
|
|
from difflib import SequenceMatcher
|
|
from jinja2 import Environment, PackageLoader
|
|
|
|
|
|
def template(name, prompt):
|
|
best_ratio = 0
|
|
best_template = ''
|
|
|
|
environment = Environment(loader=PackageLoader(__name__, 'templates'))
|
|
for template in environment.list_templates():
|
|
base, _ = path.splitext(template)
|
|
ratio = SequenceMatcher(None, path.basename(name).lower(), base).ratio()
|
|
if ratio > best_ratio:
|
|
best_ratio = ratio
|
|
best_template = template
|
|
|
|
template = environment.get_template(best_template)
|
|
return template.render(prompt=prompt)
|