mirror of
https://github.com/dogkeeper886/ollama37.git
synced 2025-12-11 08:17:03 +00:00
13 lines
432 B
Python
13 lines
432 B
Python
from os import path
|
|
from difflib import get_close_matches
|
|
from jinja2 import Environment, PackageLoader
|
|
|
|
|
|
def template(name, prompt):
|
|
environment = Environment(loader=PackageLoader(__name__, 'templates'))
|
|
best_templates = get_close_matches(
|
|
path.basename(name), environment.list_templates(), n=1, cutoff=0
|
|
)
|
|
template = environment.get_template(best_templates.pop())
|
|
return template.render(prompt=prompt)
|