Spaces:
Sleeping
Sleeping
| """ | |
| AIモデルアダプターのファクトリー関数 | |
| 環境変数に基づいて適切なモデルを自動選択 | |
| """ | |
| import os | |
| from typing import Optional | |
| from .base import BaseAI | |
| from .transformers_ai import TransformersAI | |
| from .openai_ai import OpenAIAI | |
| from .anthropic_ai import AnthropicAI | |
| from .google_ai import GoogleAI | |
| def get_ai_model(model_type: Optional[str] = None, **kwargs) -> BaseAI: | |
| """ | |
| 環境変数または引数に基づいて適切なAIモデルを取得 | |
| Args: | |
| model_type: モデルタイプ（"transformers", "openai", "anthropic", "google"） | |
| Noneの場合は環境変数MODEL_TYPEから取得 | |
| **kwargs: 各モデル固有の引数 | |
| - transformers: model_path | |
| - openai: model_name, api_key | |
| - anthropic: model_name, api_key | |
| - google: model_name, api_key | |
| Returns: | |
| BaseAI: 選択されたモデルのインスタンス | |
| Examples: | |
| # 環境変数から自動選択 | |
| ai = get_ai_model() | |
| # 明示的に指定 | |
| ai = get_ai_model("transformers", model_path="Qwen/Qwen2.5-3B-Instruct") | |
| ai = get_ai_model("openai", model_name="gpt-4", api_key="sk-...") | |
| """ | |
| # モデルタイプを決定 | |
| if model_type is None: | |
| model_type = os.getenv("MODEL_TYPE", "transformers") | |
| model_type = model_type.lower() | |
| # モデルタイプに応じて適切なクラスを返す | |
| if model_type == "transformers": | |
| model_path = kwargs.get("model_path") or os.getenv("HF_MODEL_REPO", "meta-llama/Llama-3.2-3B-Instruct") | |
| return TransformersAI.get_model(model_path=model_path) | |
| elif model_type == "openai": | |
| model_name = kwargs.get("model_name") or os.getenv("OPENAI_MODEL", "gpt-4") | |
| api_key = kwargs.get("api_key") or os.getenv("OPENAI_API_KEY") | |
| return OpenAIAI.get_model(model_name=model_name, api_key=api_key) | |
| elif model_type == "anthropic": | |
| model_name = kwargs.get("model_name") or os.getenv("ANTHROPIC_MODEL", "claude-3-5-sonnet-20241022") | |
| api_key = kwargs.get("api_key") or os.getenv("ANTHROPIC_API_KEY") | |
| return AnthropicAI.get_model(model_name=model_name, api_key=api_key) | |
| elif model_type == "google": | |
| model_name = kwargs.get("model_name") or os.getenv("GOOGLE_MODEL", "gemini-pro") | |
| api_key = kwargs.get("api_key") or os.getenv("GOOGLE_API_KEY") | |
| return GoogleAI.get_model(model_name=model_name, api_key=api_key) | |
| else: | |
| raise ValueError( | |
| f"不明なモデルタイプ: {model_type}. " | |
| f"サポートされているタイプ: transformers, openai, anthropic, google" | |
| ) | |
| # 後方互換性のため、BaseAIもエクスポート | |
| __all__ = [ | |
| "BaseAI", | |
| "TransformersAI", | |
| "OpenAIAI", | |
| "AnthropicAI", | |
| "GoogleAI", | |
| "get_ai_model", | |
| ] | |