Merge pull request #374 from DocShotgun/main

Templating: Support chat_template.jinja
This commit is contained in:
Brian 2025-08-05 01:18:25 -04:00 committed by GitHub
commit 83f778db2d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -218,6 +218,7 @@ async def find_prompt_template(template_name, model_dir: pathlib.Path):
logger.info("Attempting to load a prompt template if present.") logger.info("Attempting to load a prompt template if present.")
find_template_functions = [ find_template_functions = [
lambda: PromptTemplate.from_file(model_dir / "chat_template.jinja"),
lambda: PromptTemplate.from_model_json( lambda: PromptTemplate.from_model_json(
model_dir / "chat_template.json", model_dir / "chat_template.json",
key="chat_template", key="chat_template",