simonw--llm
Simon Willison
26 行
608 B
Python
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"
|