zrb 1.6.7__tar.gz → 1.6.9__tar.gz

{zrb-1.6.7 → zrb-1.6.9}/PKG-INFO

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

{zrb-1.6.7 → zrb-1.6.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.6.7"
+version = "1.6.9"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"

{zrb-1.6.7 → zrb-1.6.9}/src/zrb/context/context.py

@@ -89,7 +89,7 @@ class Context(AnyContext):
         return int(self.render(template))
 
     def render_float(self, template: str | float) -> float:
-        if isinstance(template, float):
+        if isinstance(template, float) or isinstance(template, int):
             return template
         return float(self.render(template))
 
@@ -107,6 +107,7 @@ class Context(AnyContext):
         if plain:
             # self.append_to_shared_log(remove_style(message))
             print(message, sep=sep, end=end, file=file, flush=flush)
+            self.append_to_shared_log(remove_style(f"{message}{end}"))
             return
         color = self._color
         icon = self._icon
@@ -120,7 +121,7 @@ class Context(AnyContext):
         now = datetime.datetime.now()
         formatted_time = now.strftime("%y%m%d %H:%M:%S.%f")[:19] + " "
         prefix = f"{formatted_time}{attempt_status} {padded_styled_task_name} ⬤ "
-        self.append_to_shared_log(remove_style(f"{prefix} {message}"))
+        self.append_to_shared_log(remove_style(f"{prefix} {message}{end}"))
         stylized_prefix = stylize(prefix, color=color)
         print(f"{stylized_prefix} {message}", sep=sep, end=end, file=file, flush=flush)
 

{zrb-1.6.7 → zrb-1.6.9}/src/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-1.6.7 → zrb-1.6.9}/src/zrb/runner/web_route/static/resources/session/current-session.js

@@ -25,7 +25,7 @@ const CURRENT_SESSION = {
                 resultTextarea.rows = resultLineCount <= 5 ? resultLineCount : 5;
                 // update text areas
                 resultTextarea.value = data.final_result;
-                logTextarea.value = data.log.join("\n");
+                logTextarea.value = data.log.join("");
                 // logTextarea.scrollTop = logTextarea.scrollHeight;
                 // visualize history
                 this.showCurrentSession(data.task_status, data.finished);

{zrb-1.6.7 → zrb-1.6.9}/src/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 → zrb-1.6.9}/src/zrb/task/llm_task.py

@@ -332,7 +332,7 @@ class LLMTask(BaseTask):
                 user_prompt=user_prompt,
                 history_list=history_list,
             )
-            if agent_run:
+            if agent_run and agent_run.result:
                 new_history_list = json.loads(agent_run.result.all_messages_json())
                 data_to_write = ConversationHistoryData(
                     context=conversation_context,  # Save the final context state