chore: more __init__ stuff

phil65 · Dec 31, 2024 · ea7fdb2 · ea7fdb2
1 parent 597ea7d
commit ea7fdb2
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/src/llmling/__init__.py b/src/llmling/__init__.py
@@ -5,7 +5,6 @@
     LoadedResource,
     default_registry as resource_registry,
 )
-from llmling.config.models import Config
 from llmling.config.runtime import RuntimeConfig
 from llmling.core.exceptions import (
     LLMLingError,
@@ -17,19 +16,32 @@
 )
 from llmling.processors.registry import ProcessorRegistry
 from llmling.tools import LLMCallableTool, ToolError
-from llmling.prompts import PromptMessage, PromptParameter, StaticPrompt
-from llmling.config.models import ConfigModel, GlobalSettings, LLMCapabilitiesConfig
+from llmling.prompts import (
+    PromptMessage,
+    PromptParameter,
+    StaticPrompt,
+    DynamicPrompt,
+    BasePrompt,
+)
+from llmling.config.models import (
+    ConfigModel,
+    GlobalSettings,
+    LLMCapabilitiesConfig,
+    Config,
+)
 from llmling.config.store import ConfigStore
 from llmling.core.baseregistry import BaseRegistry
 
 __version__ = "1.5.5"
 
 __all__ = [
+    "BasePrompt",
     "BaseRegistry",
     "Config",
     "ConfigError",
     "ConfigModel",
     "ConfigStore",
+    "DynamicPrompt",
     "GlobalSettings",
     "LLMCallableTool",
     "LLMCapabilitiesConfig",