34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
# services/llm/llm_factory.py
|
||
|
|
"""
|
||
|
|
Factory class for creating LLM services
|
||
|
|
"""
|
||
|
|
from typing import Optional
|
||
|
|
from .openai_service import OpenAIService
|
||
|
|
from .ollama_service import OllamaService
|
||
|
|
from .grok_service import GrokService
|
||
|
|
|
||
|
|
class LLMFactory:
|
||
|
|
"""Factory class for creating LLM service instances"""
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def create_service(service_type: str, **kwargs) -> Optional['LLMService']:
|
||
|
|
"""
|
||
|
|
Create an instance of the specified LLM service
|
||
|
|
|
||
|
|
Args:
|
||
|
|
service_type: Type of LLM service ("openai", "ollama", "grok")
|
||
|
|
**kwargs: Additional arguments for service initialization
|
||
|
|
"""
|
||
|
|
services = {
|
||
|
|
"openai": OpenAIService,
|
||
|
|
"ollama": OllamaService,
|
||
|
|
"grok": GrokService
|
||
|
|
}
|
||
|
|
|
||
|
|
service_class = services.get(service_type.lower())
|
||
|
|
if service_class:
|
||
|
|
return service_class(**kwargs)
|
||
|
|
else:
|
||
|
|
print(f"Unknown service type: {service_type}")
|
||
|
|
return None
|