simonw--llm
Simon Willison
a571a4e948
* Docs and shape of register_fragment_loaders hook, refs #863 * Update docs for fragment loaders returning a list of FragmentString * Support multiple fragments with same content, closes #888 * Call the pm.hook.register_fragment_loaders hook * Test for register_fragment_loaders hook * Rename FragmentString to Fragment Closes #863
31 行
718 B
Python
31 行
718 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"
|
|
|
|
|
|
@hookspec
|
|
def register_fragment_loaders(register):
|
|
"Register additional fragment loaders with prefixes"
|