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

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-1.9.11.dist-info → zrb-1.9.12.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.9.11
+Version: 1.9.12
 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.12.dist-info}/RECORD

@@ -342,7 +342,7 @@ 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/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
@@ -393,7 +393,7 @@ zrb/util/todo.py,sha256=r9_KYF2-hLKMNjsp6AFK9zivykMrywd-kJ4bCwfdafI,19323
 zrb/util/todo_model.py,sha256=0SJ8aLYfJAscDOk5JsH7pXP3h1rAG91VMCS20-c2Y6A,1576
 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.12.dist-info/METADATA,sha256=-DI8RzdASD_NeRQmc2UTza0e2DoLwD-CFQm4YqENtBk,9778
+zrb-1.9.12.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+zrb-1.9.12.dist-info/entry_points.txt,sha256=-Pg3ElWPfnaSM-XvXqCxEAa-wfVI6BEgcs386s8C8v8,46
+zrb-1.9.12.dist-info/RECORD,,