File size: 2,669 Bytes
cd7c282 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
"""Chat Client Factory for unified provider selection."""
from typing import Any
import structlog
from agent_framework import BaseChatClient
from agent_framework.openai import OpenAIChatClient
from src.clients.huggingface import HuggingFaceChatClient
from src.utils.config import settings
logger = structlog.get_logger()
def get_chat_client(
provider: str | None = None,
api_key: str | None = None,
model_id: str | None = None,
**kwargs: Any,
) -> BaseChatClient:
"""
Factory for creating chat clients.
Auto-detection priority:
1. Explicit provider parameter
2. OpenAI key (Best Function Calling)
3. Gemini key (Best Context/Cost)
4. HuggingFace (Free Fallback)
Args:
provider: Force specific provider ("openai", "gemini", "huggingface")
api_key: Override API key for the provider
model_id: Override default model ID
**kwargs: Additional arguments for the client
Returns:
Configured BaseChatClient instance (Namespace Neutral)
Raises:
ValueError: If an unsupported provider is explicitly requested
NotImplementedError: If Gemini is explicitly requested (not yet implemented)
"""
# Normalize provider to lowercase for case-insensitive matching
normalized = provider.lower() if provider is not None else None
# Validate explicit provider requests early
valid_providers = (None, "openai", "gemini", "huggingface")
if normalized not in valid_providers:
raise ValueError(f"Unsupported provider: {provider!r}")
# 1. OpenAI (Standard / Paid Tier)
if normalized == "openai" or (normalized is None and settings.has_openai_key):
logger.info("Using OpenAI Chat Client")
return OpenAIChatClient(
model_id=model_id or settings.openai_model,
api_key=api_key or settings.openai_api_key,
**kwargs,
)
# 2. Gemini (High Performance / Alternative)
if normalized == "gemini":
# Explicit request for Gemini - fail loudly
raise NotImplementedError("Gemini client not yet implemented (Planned Phase 4)")
if normalized is None and settings.has_gemini_key:
# Implicit (has key but not explicit) - log warning and fall through
logger.warning("Gemini key detected but client not yet implemented; falling back")
# 3. HuggingFace (Free Fallback)
# This is the default if no other keys are present
logger.info("Using HuggingFace Chat Client (Free Tier)")
return HuggingFaceChatClient(
model_id=model_id or settings.huggingface_model,
api_key=api_key or settings.hf_token,
**kwargs,
)
|