zrb 1.9.11__py3-none-any.whl → 1.9.13__py3-none-any.whl

zrb/task/llm/error.py

@@ -1,18 +1,23 @@
 import json
 from typing import TYPE_CHECKING, Optional
 
-from pydantic import BaseModel
-
 if TYPE_CHECKING:
     from openai import APIError
 
 
 # Define a structured error model for tool execution failures
-class ToolExecutionError(BaseModel):
-    tool_name: str
-    error_type: str
-    message: str
-    details: Optional[str] = None
+class ToolExecutionError:
+    def __init__(
+        self,
+        tool_name: str,
+        error_type: str,
+        message: str,
+        details: Optional[str] = None,
+    ):
+        self.tool_name = tool_name
+        self.error_type = error_type
+        self.message = message
+        self.details = details
 
 
 def extract_api_error_details(error: "APIError") -> str:

zrb/task/llm/history.py

@@ -4,8 +4,6 @@ from collections.abc import Callable
 from copy import deepcopy
 from typing import Any, Optional
 
-from pydantic import BaseModel, Field
-
 from zrb.attr.type import StrAttr
 from zrb.context.any_context import AnyContext
 from zrb.context.any_shared_context import AnySharedContext
@@ -16,19 +14,31 @@ from zrb.util.run import run_async
 
 
 # Define the new ConversationHistoryData model
-class ConversationHistoryData(BaseModel):
-    long_term_context: str = Field(
-        default="",
-        description="A markdown-formatted string containing curated, long-term context.",
-    )
-    conversation_summary: str = Field(
-        default="",
-        description="A free-text summary of the conversation history.",
-    )
-    history: ListOfDict = Field(
-        default_factory=list,
-        description="The recent, un-summarized conversation history.",
-    )
+class ConversationHistoryData:
+    def __init__(
+        self,
+        long_term_context: str = "",
+        conversation_summary: str = "",
+        history: Optional[ListOfDict] = None,
+        messages: Optional[ListOfDict] = None,
+    ):
+        self.long_term_context = long_term_context
+        self.conversation_summary = conversation_summary
+        self.history = (
+            history
+            if history is not None
+            else (messages if messages is not None else [])
+        )
+
+    def to_dict(self) -> dict[str, Any]:
+        return {
+            "long_term_context": self.long_term_context,
+            "conversation_summary": self.conversation_summary,
+            "history": self.history,
+        }
+
+    def model_dump_json(self, indent: int = 2) -> str:
+        return json.dumps(self.to_dict(), indent=indent)
 
     @classmethod
     async def read_from_sources(
@@ -81,7 +91,11 @@ class ConversationHistoryData(BaseModel):
                 return data  # Already a valid instance
             if isinstance(data, dict):
                 # This handles both the new format and the old {'context': ..., 'history': ...}
-                return cls.model_validate(data)
+                return cls(
+                    long_term_context=data.get("long_term_context", ""),
+                    conversation_summary=data.get("conversation_summary", ""),
+                    history=data.get("history", data.get("messages")),
+                )
             elif isinstance(data, list):
                 # Handle very old format (just a list) - wrap it
                 ctx.log_warning(
@@ -183,7 +197,7 @@ async def write_conversation_history(
         ctx, conversation_history_file_attr, render_history_file
     )
     if history_file != "":
-        write_file(history_file, history_data.model_dump_json(indent=2))
+        write_file(history_file, json.dumps(history_data.to_dict(), indent=2))
 
 
 def replace_system_prompt_in_history_list(

zrb/util/git_diff_model.py

@@ -1,7 +1,10 @@
-from pydantic import BaseModel
-
-
-class DiffResult(BaseModel):
-    created: list[str]
-    removed: list[str]
-    updated: list[str]
+class DiffResult:
+    def __init__(
+        self,
+        created: list[str] | None = None,
+        removed: list[str] | None = None,
+        updated: list[str] | None = None,
+    ):
+        self.created = created if created is not None else []
+        self.removed = removed if removed is not None else []
+        self.updated = updated if updated is not None else []

zrb/util/git_subtree_model.py

@@ -1,11 +1,10 @@
-from pydantic import BaseModel
+class SingleSubTreeConfig:
+    def __init__(self, repo_url: str, branch: str, prefix: str):
+        self.repo_url = repo_url
+        self.branch = branch
+        self.prefix = prefix
 
 
-class SingleSubTreeConfig(BaseModel):
-    repo_url: str
-    branch: str
-    prefix: str
-
-
-class SubTreeConfig(BaseModel):
-    data: dict[str, SingleSubTreeConfig]
+class SubTreeConfig:
+    def __init__(self, data: dict[str, SingleSubTreeConfig]):
+        self.data = data

zrb/util/todo_model.py

@@ -1,27 +1,33 @@
 import datetime
+import re
 
-from pydantic import BaseModel, Field, model_validator
 
-
-class TodoTaskModel(BaseModel):
-    priority: str | None = Field("D", pattern=r"^[A-Z]$")  # Priority like A, B, ...
-    completed: bool = False  # True if completed, False otherwise
-    description: str  # Main task description
-    projects: list[str] = []  # List of projects (e.g., +Project)
-    contexts: list[str] = []  # List of contexts (e.g., @Context)
-    keyval: dict[str, str] = {}  # Special key (e.g., due:2016-05-30)
-    creation_date: datetime.date | None = None  # Creation date
-    completion_date: datetime.date | None = None  # Completion date
-
-    @model_validator(mode="before")
-    def validate_dates(cls, values):
-        completion_date = values.get("completion_date")
-        creation_date = values.get("creation_date")
+class TodoTaskModel:
+    def __init__(
+        self,
+        description: str,
+        priority: str | None = "D",
+        completed: bool = False,
+        projects: list[str] | None = None,
+        contexts: list[str] | None = None,
+        keyval: dict[str, str] | None = None,
+        creation_date: datetime.date | None = None,
+        completion_date: datetime.date | None = None,
+    ):
+        if priority is not None and not re.match(r"^[A-Z]$", priority):
+            raise ValueError("Invalid priority format")
         if completion_date and not creation_date:
             raise ValueError(
                 "creation_date must be specified if completion_date is set."
             )
-        return values
+        self.priority = priority
+        self.completed = completed
+        self.description = description
+        self.projects = projects if projects is not None else []
+        self.contexts = contexts if contexts is not None else []
+        self.keyval = keyval if keyval is not None else {}
+        self.creation_date = creation_date
+        self.completion_date = completion_date
 
     def get_additional_info_length(self):
         """

{zrb-1.9.11.dist-info → zrb-1.9.13.dist-info}/METADATA

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

{zrb-1.9.11.dist-info → zrb-1.9.13.dist-info}/RECORD

@@ -341,8 +341,8 @@ zrb/task/llm/agent.py,sha256=BZHbz-YXgSdm1tTwGMR_maqcd3yMFGSdzLyDjuxT_XI,6702
 zrb/task/llm/config.py,sha256=TlyH925_fboIlK2Ixf34tynmenqs9s9rfsnPs4jff78,3490
 zrb/task/llm/context.py,sha256=LGGQ_mb1dWorfshHjGgXEW_pRweGj-6MZcIUFq3AHy4,2213
 zrb/task/llm/context_enrichment.py,sha256=djY4fE9C0zUxJPrrb2xDboBXr_2kPUS_b4HjqslVpHg,6051
-zrb/task/llm/error.py,sha256=s5PSK3ibGupMzOM0P81nstHWrMr3205H0FSDwfhWUDA,3662
-zrb/task/llm/history.py,sha256=cpaNqoEzsNAgzZGPPdohA8U5nTrVWmZ3P1Y5RTtXDgc,7986
+zrb/task/llm/error.py,sha256=0YE8Unv8H1HPb9QzaCBbykEBKtr8JyUWt6dywPuNOUg,3812
+zrb/task/llm/history.py,sha256=dqCwJYRoMVrvyMHb4M6-KLcwAUF4tbDCP7qgD1h694s,8540
 zrb/task/llm/history_summarization.py,sha256=V0G1BiISnxxmD8040PrvT0_dfqGE7zbLtk74KUpuqig,6050
 zrb/task/llm/print_node.py,sha256=zocTKi9gZDxl2I6KNu095TmMc13Yip6SNuWYnswS680,4060
 zrb/task/llm/prompt.py,sha256=qhR8qS8RgaQ23D3amaHSHnBNv_NOnFB_1uxaQNc8KFw,3417
@@ -378,9 +378,9 @@ zrb/util/codemod/modify_module.py,sha256=2mzi_NxJ-kNFo5G0U_Rqb3JoYQl1s6Izt7b_wAl
 zrb/util/cron.py,sha256=UWqqhhM7DDTPx6wjCIdBndnVh3NRu-sdKazp8aZkXh8,3833
 zrb/util/file.py,sha256=tm_8qn0vEM8Hz46yXUcvFHfsLtQNqidQaEuB85xqhFE,2806
 zrb/util/git.py,sha256=GS08OqE-gs6mN-_VqACN82DbI81nzG3FiXeim_-jVAU,8193
-zrb/util/git_diff_model.py,sha256=Fthl2nyBz3D5fIfbMCPEFBWfXUB6qA4rQWApKaimaI8,131
+zrb/util/git_diff_model.py,sha256=Tg2q6oxQ5chryThzjs0cLcKFGM03CnqvusTo_Ug_oX4,369
 zrb/util/git_subtree.py,sha256=AyQWCWEi2EIzEpYXRnYN55157KMUql0WHj70QNw5PHU,4612
-zrb/util/git_subtree_model.py,sha256=mDMUG6mju1_-XXpIE41EqFdu_gmDeeJ3OAxqcJunY7g,196
+zrb/util/git_subtree_model.py,sha256=GB0SJxROdjObMEKRrm6cHV2Fsda3aqEb2JxVwo1sfnM,293
 zrb/util/group.py,sha256=T82yr3qg9I5k10VPXkMyrIRIqyfzadSH813bqzwKEPI,4718
 zrb/util/init_path.py,sha256=9eN7CkWNGhDBpjTQs2j9YHVMzui7Y8DEb1WP4aTPzeo,659
 zrb/util/load.py,sha256=DK0KYSlu48HCoGPqnW1IxnE3pHrZSPCstfz8Fjyqqv8,2140
@@ -390,10 +390,10 @@ zrb/util/string/conversion.py,sha256=sMmstzbrNgLvWAQukqoXz45JtsNpJrniudAtzJaQlYw
 zrb/util/string/format.py,sha256=MwWGAwSdtOgR_2uz-JCXlg_q-uRYUUI-G8CGkfdgqik,1198
 zrb/util/string/name.py,sha256=SXEfxJ1-tDOzHqmSV8kvepRVyMqs2XdV_vyoh_9XUu0,1584
 zrb/util/todo.py,sha256=r9_KYF2-hLKMNjsp6AFK9zivykMrywd-kJ4bCwfdafI,19323
-zrb/util/todo_model.py,sha256=0SJ8aLYfJAscDOk5JsH7pXP3h1rAG91VMCS20-c2Y6A,1576
+zrb/util/todo_model.py,sha256=hhzAX-uFl5rsg7iVX1ULlJOfBtblwQ_ieNUxBWfc-Os,1670
 zrb/xcom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 zrb/xcom/xcom.py,sha256=o79rxR9wphnShrcIushA0Qt71d_p3ZTxjNf7x9hJB78,1571
-zrb-1.9.11.dist-info/METADATA,sha256=UzGGLF7ae4d-kprWIxmTmvvRRmTDWBJ4jxfsYPhxSaM,9778
-zrb-1.9.11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-zrb-1.9.11.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
-zrb-1.9.11.dist-info/RECORD,,
+zrb-1.9.13.dist-info/METADATA,sha256=6Yn22mBzufYdh4OQW8OnKZd2gO7ppZF_5sb74h1fyaM,9778
+zrb-1.9.13.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.9.13.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.9.13.dist-info/RECORD,,