c1groupy 0.4.1__tar.gz → 0.4.2__tar.gz

{c1groupy-0.4.1 → c1groupy-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: c1groupy
-Version: 0.4.1
+Version: 0.4.2
 Summary: C1G company Python module for utilities and project setup
 License-Expression: MIT
 License-File: LICENCE

{c1groupy-0.4.1 → c1groupy-0.4.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "c1groupy"
-version = "0.4.1"
+version = "0.4.2"
 description = "C1G company Python module for utilities and project setup"
 authors = [
     {name = "Tim M Schendzielorz",email = "tim.schendzielorz@googlemail.com"},
@@ -40,3 +40,9 @@ packages = [{include = "c1gpy", from = "src"}]
 [build-system]
 requires = ["poetry-core>=2.0.0,<3.0.0"]
 build-backend = "poetry.core.masonry.api"
+
+[dependency-groups]
+dev = [
+    "pytest>=9.0.2",
+    "pytest-asyncio>=1.3.0",
+]

{c1groupy-0.4.1 → c1groupy-0.4.2}/src/c1gpy/llm_handler/handler.py

@@ -92,7 +92,8 @@ class LLMHandler:
     def __init__(
         self,
         model_name: str,
-        api_key: str,
+        api_key: str | None = None,
+        *,
         langfuse_prompt_name: str,
         langfuse_config: LangfuseConfig,
         langfuse_prompt_label: str = "production",

{c1groupy-0.4.1 → c1groupy-0.4.2}/src/c1gpy/llm_handler/provider_registry.py

@@ -8,6 +8,7 @@ Factories use lazy imports so optional provider dependencies are only required
 when that provider is used.
 """
 
+import os
 from enum import Enum
 from typing import Any, Callable
 
@@ -24,6 +25,29 @@ class ModelProvider(Enum):
     GOOGLE = "google"
 
 
+_ENV_API_KEY_VARS: dict[ModelProvider, tuple[str, ...]] = {
+    ModelProvider.OPENAI: ("OPENAI_API_KEY",),
+    ModelProvider.ANTHROPIC: ("ANTHROPIC_API_KEY",),
+    ModelProvider.GOOGLE: ("GOOGLE_API_KEY", "GEMINI_API_KEY"),
+}
+
+
+def _resolve_api_key(provider: ModelProvider, api_key: str | None) -> str:
+    if api_key is not None and api_key.strip() != "":
+        return api_key
+
+    for env_var in _ENV_API_KEY_VARS.get(provider, ()):
+        env_value = os.getenv(env_var)
+        if env_value:
+            return env_value
+
+    raise ProviderError(
+        "Missing API key. Provide 'api_key' or set one of these environment variables: "
+        f"{_ENV_API_KEY_VARS.get(provider, ())}",
+        provider=provider.value,
+    )
+
+
 # Registry of provider detection rules and LLM constructors.
 _PROVIDER_REGISTRY: dict[
     ModelProvider,
@@ -69,7 +93,7 @@ def detect_provider(model_name: str) -> ModelProvider:
 def create_llm(
     provider: ModelProvider,
     model_name: str,
-    api_key: str,
+    api_key: str | None = None,
     **kwargs: Any,
 ) -> BaseChatModel:
     """Construct a LangChain chat model for the given provider."""
@@ -81,7 +105,8 @@ def create_llm(
             provider=str(provider),
         )
 
-    return factory(model_name, api_key, kwargs)
+    resolved_api_key = _resolve_api_key(provider, api_key)
+    return factory(model_name, resolved_api_key, kwargs)
 
 
 def _create_openai_llm(