zep-cloud 3.4.0__tar.gz → 3.4.2__tar.gz

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zep-cloud
-Version: 3.4.0
+Version: 3.4.2
 Summary: 
 Requires-Python: >=3.9.0,<4.0
 Classifier: Intended Audience :: Developers

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/pyproject.toml

@@ -3,7 +3,7 @@ name = "zep-cloud"
 
 [tool.poetry]
 name = "zep-cloud"
-version = "3.4.0"
+version = "3.4.2"
 description = ""
 readme = "README.md"
 authors = []

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/src/zep_cloud/core/client_wrapper.py

@@ -22,10 +22,10 @@ class BaseClientWrapper:
 
     def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
-            "User-Agent": "zep-cloud/3.4.0",
+            "User-Agent": "zep-cloud/3.4.1",
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "zep-cloud",
-            "X-Fern-SDK-Version": "3.4.0",
+            "X-Fern-SDK-Version": "3.4.1",
             **(self.get_custom_headers() or {}),
         }
         headers["Authorization"] = f"Api-Key {self.api_key}"

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/src/zep_cloud/graph/utils.py

@@ -1,20 +1,21 @@
 from datetime import datetime
-from typing import List
+from typing import List, Optional
 
+from dateutil import parser as dateutil_parser
 from zep_cloud import EntityEdge, EntityNode, Episode
 
 DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
 
 
-def parse_iso_datetime(iso_string: str) -> datetime:
-    """Parse ISO datetime string, handling Z suffix for UTC."""
+def parse_iso_datetime(iso_string: str) -> Optional[datetime]:
+    """Parse ISO datetime string using dateutil parser."""
+    if not iso_string:
+        return None
+    
     try:
-        return datetime.fromisoformat(iso_string)
-    except ValueError:
-        # Handle Z suffix for Python 3.9 compatibility
-        if iso_string.endswith('Z'):
-            return datetime.fromisoformat(iso_string[:-1] + '+00:00')
-        raise
+        return dateutil_parser.isoparse(iso_string)
+    except (ValueError, TypeError):
+        return None
 
 TEMPLATE_STRING = """
 FACTS and ENTITIES{episodes_header} represent relevant context to the current conversation.
@@ -52,9 +53,13 @@ def format_edge_date_range(edge: EntityEdge) -> str:
     invalid_at = "present"
 
     if edge.valid_at is not None:
-        valid_at = parse_iso_datetime(edge.valid_at).strftime(DATE_FORMAT)
+        parsed_valid_at = parse_iso_datetime(edge.valid_at)
+        if parsed_valid_at is not None:
+            valid_at = parsed_valid_at.strftime(DATE_FORMAT)
     if edge.invalid_at is not None:
-        invalid_at = parse_iso_datetime(edge.invalid_at).strftime(DATE_FORMAT)
+        parsed_invalid_at = parse_iso_datetime(edge.invalid_at)
+        if parsed_invalid_at is not None:
+            invalid_at = parsed_invalid_at.strftime(DATE_FORMAT)
 
     return f"{valid_at} - {invalid_at}"
 
@@ -114,7 +119,8 @@ def compose_context_string(edges: List[EntityEdge], nodes: List[EntityNode], epi
             elif hasattr(episode, 'role_type') and episode.role_type:
                 role_prefix = f"({episode.role_type}): "
 
-            timestamp = parse_iso_datetime(episode.created_at).strftime(DATE_FORMAT)
+            parsed_timestamp = parse_iso_datetime(episode.created_at)
+            timestamp = parsed_timestamp.strftime(DATE_FORMAT) if parsed_timestamp is not None else "date unknown"
             
             episode_str = f"  - {role_prefix}{episode.content} ({timestamp})"
             episodes_list.append(episode_str)

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/src/zep_cloud/thread/client.py

@@ -159,12 +159,12 @@ class ThreadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ThreadContextResponse:
         """
-        Returns most relevant context for a given thread.
+        Returns most relevant context from the user graph (including memory from any/all past threads) based on the content of the past few messages of the given thread.
 
         Parameters
         ----------
         thread_id : str
-            The ID of the thread for which to retrieve context.
+            The ID of the current thread (for which context is being retrieved).
 
         min_rating : typing.Optional[float]
             The minimum rating by which to filter relevant facts.
@@ -272,7 +272,7 @@ class ThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -334,7 +334,7 @@ class ThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -537,12 +537,12 @@ class AsyncThreadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ThreadContextResponse:
         """
-        Returns most relevant context for a given thread.
+        Returns most relevant context from the user graph (including memory from any/all past threads) based on the content of the past few messages of the given thread.
 
         Parameters
         ----------
         thread_id : str
-            The ID of the thread for which to retrieve context.
+            The ID of the current thread (for which context is being retrieved).
 
         min_rating : typing.Optional[float]
             The minimum rating by which to filter relevant facts.
@@ -666,7 +666,7 @@ class AsyncThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -736,7 +736,7 @@ class AsyncThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/src/zep_cloud/thread/raw_client.py

@@ -268,12 +268,12 @@ class RawThreadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[ThreadContextResponse]:
         """
-        Returns most relevant context for a given thread.
+        Returns most relevant context from the user graph (including memory from any/all past threads) based on the content of the past few messages of the given thread.
 
         Parameters
         ----------
         thread_id : str
-            The ID of the thread for which to retrieve context.
+            The ID of the current thread (for which context is being retrieved).
 
         min_rating : typing.Optional[float]
             The minimum rating by which to filter relevant facts.
@@ -450,7 +450,7 @@ class RawThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -532,7 +532,7 @@ class RawThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -829,12 +829,12 @@ class AsyncRawThreadClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> AsyncHttpResponse[ThreadContextResponse]:
         """
-        Returns most relevant context for a given thread.
+        Returns most relevant context from the user graph (including memory from any/all past threads) based on the content of the past few messages of the given thread.
 
         Parameters
         ----------
         thread_id : str
-            The ID of the thread for which to retrieve context.
+            The ID of the current thread (for which context is being retrieved).
 
         min_rating : typing.Optional[float]
             The minimum rating by which to filter relevant facts.
@@ -1011,7 +1011,7 @@ class AsyncRawThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1093,7 +1093,7 @@ class AsyncRawThreadClient:
             that are added to a user's graph.
 
         return_context : typing.Optional[bool]
-            Optionally return memory context relevant to the most recent messages.
+            Optionally return context block relevant to the most recent messages.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/src/zep_cloud/types/add_thread_messages_request.py

@@ -23,7 +23,7 @@ class AddThreadMessagesRequest(UniversalBaseModel):
 
     return_context: typing.Optional[bool] = pydantic.Field(default=None)
     """
-    Optionally return memory context relevant to the most recent messages.
+    Optionally return context block relevant to the most recent messages.
     """
 
     if IS_PYDANTIC_V2:

{zep_cloud-3.4.0 → zep_cloud-3.4.2}/src/zep_cloud/types/thread_context_response.py

@@ -9,7 +9,7 @@ from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 class ThreadContextResponse(UniversalBaseModel):
     context: typing.Optional[str] = pydantic.Field(default=None)
     """
-    Memory context containing relevant facts and entities for the session. Can be put into the prompt directly.
+    Context block containing relevant facts, entities, and messages/episodes from the user graph. Meant to be replaced in the system prompt on every chat turn.
     """
 
     if IS_PYDANTIC_V2: