polyllm-core 0.2.0__tar.gz → 0.3.0__tar.gz

{polyllm_core-0.2.0 → polyllm_core-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polyllm-core
-Version: 0.2.0
+Version: 0.3.0
 Summary: PolyLLM Python core: unified protocol, registry, and parameter policy.
 Author: PolyLLM contributors
 License-Expression: MIT

{polyllm_core-0.2.0 → polyllm_core-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "polyllm-core"
-version = "0.2.0"
+version = "0.3.0"
 description = "PolyLLM Python core: unified protocol, registry, and parameter policy."
 readme = "README.md"
 requires-python = ">=3.10"

{polyllm_core-0.2.0 → polyllm_core-0.3.0}/src/polyllm_core/client.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from dataclasses import replace
 from typing import Any, AsyncIterator, Mapping
 
 from .config import resolve_provider_runtime
@@ -148,7 +149,8 @@ def create_llm(config: ClientConfig | Mapping[str, Any] | None = None, **kwargs:
     if not isinstance(config, ClientConfig):
         raise PolyLLMError("INVALID_CONFIG", "create_llm 配置必须是 ClientConfig 或 dict")
     config = replace_provider_configs(config)
-    resolve_runtime_configs(config)
+    resolved = resolve_runtime_configs(config)
+    config = replace(config, providers=resolved)
     return PolyLLM(config)
 
 
@@ -162,8 +164,6 @@ def _merge_request(
 
 
 def replace_provider_configs(config: ClientConfig) -> ClientConfig:
-    from dataclasses import replace
-
     providers = {
         provider: value if isinstance(value, ProviderConfig) else ProviderConfig(**value)
         for provider, value in config.providers.items()
@@ -226,8 +226,15 @@ def normalize_tool_call(value: Any) -> UnifiedToolCall:
 
 
 def resolve_runtime_configs(config: ClientConfig) -> dict[str, ProviderConfig]:
+    plugins_by_provider = {plugin.provider: plugin for plugin in config.plugins}
     resolved: dict[str, ProviderConfig] = {}
     for provider, provider_config in config.providers.items():
+        plugin = plugins_by_provider.get(provider)
+        if plugin is not None and plugin.allow_missing_api_key and not (
+            provider_config.api_key or provider_config.api_key_env
+        ):
+            resolved[provider] = provider_config
+            continue
         runtime = resolve_provider_runtime(provider, provider_config)
         resolved[provider] = ProviderConfig(
             api_key=runtime.api_key,

{polyllm_core-0.2.0 → polyllm_core-0.3.0}/src/polyllm_core/types.py

@@ -151,6 +151,9 @@ class ChatContext:
 class LLMPlugin(ABC):
     provider: str
     models: tuple[ModelSpec, ...]
+    #: 插件声明"无 API Key 也能工作"（例如本地服务，如 Ollama）。
+    #: 设置为 True 时，`create_llm` 在构造阶段不为该 provider 强制校验 API Key。
+    allow_missing_api_key: bool = False
 
     @abstractmethod
     async def chat(

{polyllm_core-0.2.0 → polyllm_core-0.3.0}/src/polyllm_core.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: polyllm-core
-Version: 0.2.0
+Version: 0.3.0
 Summary: PolyLLM Python core: unified protocol, registry, and parameter policy.
 Author: PolyLLM contributors
 License-Expression: MIT

{polyllm_core-0.2.0 → polyllm_core-0.3.0}/tests/test_core.py

@@ -142,6 +142,27 @@ class CoreTests(unittest.TestCase):
         response = asyncio.run(client.chat("test:mini", messages=[{"role": "user", "content": "hi"}]))
         self.assertEqual(response.model, "test-model")
 
+    def test_runtime_resolved_at_construction_not_request_time(self):
+        import os
+        os.environ["POLYLLM_TEST_KEY"] = "from-env"
+        client = create_llm(
+            plugins=[TestPlugin()],
+            providers={"test": {"api_key_env": "POLYLLM_TEST_KEY"}},
+        )
+        os.environ.pop("POLYLLM_TEST_KEY")
+        # 构造时已把环境变量解析写回 provider config，请求时不再读环境变量
+        self.assertEqual(client._config.providers["test"].api_key, "from-env")
+        captured = {}
+
+        class CapturingPlugin(TestPlugin):
+            async def chat(self, request, config, context):
+                captured["api_key"] = config.api_key
+                return await super().chat(request, config, context)
+
+        client3 = create_llm(plugins=[CapturingPlugin()], providers={"test": {"api_key": "k"}})
+        asyncio.run(client3.chat("test-model", messages=[{"role": "user", "content": "hi"}]))
+        self.assertEqual(captured["api_key"], "k")
+
     def test_unlisted_model_resolution(self):
         client = create_llm(
             plugins=[TestPlugin()],