tcsenpai/ollama
1
0
Fork 0
mirror of https://github.com/tcsenpai/ollama.git synced 2025-06-06 19:25:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #20 from jmorganca/close-matches

Browse Source 
use difflib.get_close_matches
This commit is contained in:
Michael Yang 2023-06-30 11:54:07 -07:00 committed by GitHub
commit cc329b3dd8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ollama/prompt.py
View File

@ -1,19 +1,9 @@
from os import path
from difflib import SequenceMatcher
from difflib import get_close_matches
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)
best_templates = get_close_matches(name, environment.list_templates(), n=1, cutoff=0)
template = environment.get_template(best_templates.pop())
return template.render(prompt=prompt)