"Extract_text_generation_utilities_into_modular_structure"

backend/services/llm_providers/textgen_utils/api_key_utils.py

@@ -0,0 +1,26 @@
+"""
+API Key Utilities Module
+
+This module contains API key-related utility functions extracted from main_text_generation.py
+to resolve merge conflicts and improve maintainability.
+"""
+
+from typing import Optional
+from loguru import logger
+
+from ..tenant_provider_config import tenant_provider_config_resolver
+
+
+def get_api_key(gpt_provider: str, user_id: Optional[str] = None) -> Optional[str]:
+    """Get API key for the specified provider."""
+    try:
+        provider_mapping = {
+            "google": "gemini",
+            "huggingface": "huggingface"
+        }
+        mapped_provider = provider_mapping.get(gpt_provider, gpt_provider)
+        key, _source = tenant_provider_config_resolver.resolve_provider_key(mapped_provider, user_id=user_id)
+        return key
+    except Exception as e:
+        logger.error(f"[get_api_key] Error getting API key for {gpt_provider}: {str(e)}")
+        return None