refactoring

Author: tom-sapletta-com

code2logic/llm/__init__.py

@@ -1,15 +1,37 @@
 """LLM client integrations.
 
-Re-exports from parent package for backward compatibility.
+Re-exports from lolm package for unified LLM management.
+Backward compatible with existing code2logic imports.
 """
-from ..llm_clients import (
-    BaseLLMClient, OpenRouterClient, OllamaLocalClient,
-    LiteLLMClient, get_client
+
+# Import from lolm package (shared LLM infrastructure)
+from lolm import (
+    BaseLLMClient,
+    OpenRouterClient,
+    OllamaClient as OllamaLocalClient,  # Alias for backward compatibility
+    LiteLLMClient,
+    get_client,
+    LLMManager,
+    LLMConfig,
+    RECOMMENDED_MODELS,
+    DEFAULT_MODELS,
 )
+
 from ..intent import EnhancedIntentGenerator
 
 __all__ = [
-    'BaseLLMClient', 'OpenRouterClient', 'OllamaLocalClient',
-    'LiteLLMClient', 'get_client',
+    # Core clients (from lolm)
+    'BaseLLMClient',
+    'OpenRouterClient', 
+    'OllamaLocalClient',
+    'LiteLLMClient',
+    'get_client',
+    # Manager
+    'LLMManager',
+    'LLMConfig',
+    # Constants
+    'RECOMMENDED_MODELS',
+    'DEFAULT_MODELS',
+    # Intent
     'EnhancedIntentGenerator',
 ]