zrb 1.6.7__py3-none-any.whl → 1.6.8__py3-none-any.whl

zrb/llm_config.py

@@ -68,7 +68,9 @@ class LLMConfig:
         default_history_summarization_threshold: int | None = None,
         default_enrich_context: bool | None = None,
         default_context_enrichment_threshold: int | None = None,
+        default_model: Model | None = None,
         default_model_settings: ModelSettings | None = None,
+        default_model_provider: Provider | None = None,
     ):
         self._default_model_name = default_model_name
         self._default_model_base_url = default_base_url
@@ -87,8 +89,8 @@ class LLMConfig:
             default_context_enrichment_threshold
         )
         self._default_model_settings = default_model_settings
-        self._default_provider = None
-        self._default_model = None
+        self._default_model_provider = default_model_provider
+        self._default_model = default_model
 
     @property
     def default_model_name(self) -> str | None:
@@ -121,8 +123,8 @@ class LLMConfig:
 
     @property
     def default_model_provider(self) -> Provider | str:
-        if self._default_provider is not None:
-            return self._default_provider
+        if self._default_model_provider is not None:
+            return self._default_model_provider
         if self.default_model_base_url is None and self.default_model_api_key is None:
             return "openai"
         from pydantic_ai.providers.openai import OpenAIProvider
@@ -233,8 +235,8 @@ class LLMConfig:
     def set_default_model_base_url(self, model_base_url: str):
         self._default_model_base_url = model_base_url
 
-    def set_default_provider(self, provider: Provider | str):
-        self._default_provider = provider
+    def set_default_model_provider(self, provider: Provider | str):
+        self._default_model_provider = provider
 
     def set_default_model(self, model: Model | str):
         self._default_model = model

zrb/task/llm/config.py

@@ -85,7 +85,7 @@ def get_model(
         if model_base_url is None and model_api_key is None:
             default_model_provider = llm_config.default_model_provider
             if default_model_provider is not None:
-                new_llm_config.set_default_provider(default_model_provider)
+                new_llm_config.set_default_model_provider(default_model_provider)
         return new_llm_config.default_model
     # If it's already a Model instance, return it directly
     if isinstance(model, Model):

{zrb-1.6.7.dist-info → zrb-1.6.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.6.7
+Version: 1.6.8
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later

{zrb-1.6.7.dist-info → zrb-1.6.8.dist-info}/RECORD

@@ -245,7 +245,7 @@ zrb/input/option_input.py,sha256=TQB82ko5odgzkULEizBZi0e9TIHEbIgvdP0AR3RhA74,213
 zrb/input/password_input.py,sha256=szBojWxSP9QJecgsgA87OIYwQrY2AQ3USIKdDZY6snU,1465
 zrb/input/str_input.py,sha256=NevZHX9rf1g8eMatPyy-kUX3DglrVAQpzvVpKAzf7bA,81
 zrb/input/text_input.py,sha256=6T3MngWdUs0u0ZVs5Dl11w5KS7nN1RkgrIR_zKumzPM,3695
-zrb/llm_config.py,sha256=Jy3VEzs22MuH6mYQHNQBuFkegtQZBf5wHHPdqVjN_FQ,10286
+zrb/llm_config.py,sha256=Twe7-32OxvzCSPFOY-cV3SQGPSqCcjO1ZmiYvSjNpCk,10443
 zrb/runner/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/runner/cli.py,sha256=2xy4cPGqDZOm4EBGcP4WtKBXWjk74O2whDS3XarX0XQ,6925
 zrb/runner/common_util.py,sha256=JDMcwvQ8cxnv9kQrAoKVLA40Q1omfv-u5_d5MvvwHeE,1373
@@ -318,7 +318,7 @@ zrb/task/cmd_task.py,sha256=3JFkWZEhyrQAwbQJs2pgICBmkohUR9T-hjXw82JyNtA,10720
 zrb/task/http_check.py,sha256=Gf5rOB2Se2EdizuN9rp65HpGmfZkGc-clIAlHmPVehs,2565
 zrb/task/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/task/llm/agent.py,sha256=6wGSsw03GdY_fj12CsJh7wxB6BnE13N8RYXaWfbiUsk,5451
-zrb/task/llm/config.py,sha256=cVEeb-hdN2gxkxO4wbS0qwNfx5yg8UvAbmhzasJojvQ,3452
+zrb/task/llm/config.py,sha256=oGxHYMdIvhASnKwNuMPwcdeJiFfS0tNskzGHakpfpQU,3458
 zrb/task/llm/context.py,sha256=U9a8lxa2ikz6my0Sd5vpO763legHrMHyvBjbrqNmv0Y,3838
 zrb/task/llm/context_enrichment.py,sha256=BKiWTorUrwDCMxghFXua3d-CHA3faUaCc3am7I3YGCg,6446
 zrb/task/llm/error.py,sha256=27DQXSG8SH1-XuvXFdZQKzP39wZDWmd_YnSTz6DJKKI,3690
@@ -369,7 +369,7 @@ zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=VGISej2KQZERpornK-8X7bysp4JydMrMUTnG8B0-liI,20708
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.6.7.dist-info/METADATA,sha256=EFCJigILumuwW-EMlmrEnJwDluClLCEpni55anTdAkc,8385
-zrb-1.6.7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.6.7.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.6.7.dist-info/RECORD,,
+zrb-1.6.8.dist-info/METADATA,sha256=-DMAZAILCbphje8LYMrIIbFRFbMDF7MNsK8MMOBehM0,8385
+zrb-1.6.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.6.8.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.6.8.dist-info/RECORD,,