项目文件夹

文件
Simon Willison 6c9a8efb50 register_template_loaders plugin hook, closes #809
* Moved templates CLI commands next to each other
* llm templates loaders command
* Template loader tests
* Documentation for template loaders
2025-03-21 16:46:44 -07:00

26 行
608 B
Python

from pluggy import HookimplMarker
from pluggy import HookspecMarker
hookspec = HookspecMarker("llm")
hookimpl = HookimplMarker("llm")
@hookspec
def register_commands(cli):
"""Register additional CLI commands, e.g. 'llm mycommand ...'"""
@hookspec
def register_models(register):
"Register additional model instances representing LLM models that can be called"
@hookspec
def register_embedding_models(register):
"Register additional model instances that can be used for embedding"
@hookspec
def register_template_loaders(register):
"Register additional template loaders with prefixes"