Hopiu/llm
1
0
Fork 0
mirror of https://github.com/Hopiu/llm.git synced 2026-04-07 15:00:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Move default plugins into llm/default_plugins

Browse source
This commit is contained in:
Simon Willison 2023-07-01 11:10:30 -07:00
parent c4513068fb
commit 1b3f14fe89
4 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
llm/default_plugins/__init__.py Normal file
View file

6
llm/openai_models.py → llm/default_plugins/openai_models.py
View file

@ -1,6 +1,6 @@
from . import Model, Prompt, OptionsError, Response, hookimpl
from .errors import NeedsKeyException
from .utils import dicts_to_table_string
from llm import Model, Prompt, OptionsError, Response, hookimpl
from llm.errors import NeedsKeyException
from llm.utils import dicts_to_table_string
import click
import datetime
from typing import Optional

4
llm/vertex_models.py → llm/default_plugins/vertex_models.py
View file

@ -1,5 +1,5 @@
from . import Model, Prompt, Response, hookimpl
from .errors import NeedsKeyException
from llm import Model, Prompt, Response, hookimpl
from llm.errors import NeedsKeyException
import requests

5
llm/plugins.py
View file

@ -5,7 +5,10 @@ from typing import Dict, List
from . import hookspecs
from .models import ModelWithAliases, Model
DEFAULT_PLUGINS = ("llm.openai_models", "llm.vertex_models")
DEFAULT_PLUGINS = (
"llm.default_plugins.openai_models",
"llm.default_plugins.vertex_models",
)
pm = pluggy.PluginManager("llm")
pm.add_hookspecs(hookspecs)