SDK regeneration

.fern/metadata.json

@@ -11,6 +11,6 @@
       "exported_class_name": "Zep"
     }
   },
-  "originGitCommit": "4135f93a8bc65a8ec8f29a261ce0e721b714cc57",
-  "sdkVersion": "4.0.0a1"
+  "originGitCommit": "f5d305f39ed149d17c3af28e057c82f3801d28f2",
+  "sdkVersion": "4.0.0a2"
 }

pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "zep-cloud"
 dynamic = ["version"]
-version = "4.0.0a1"
+version = "4.0.0a2"
 
 [tool.poetry]
 name = "zep-cloud"
-version = "4.0.0a1"
+version = "4.0.0a2"
 description = ""
 readme = "README.md"
 authors = []

src/zep_cloud/__init__.py

@@ -19,7 +19,9 @@
         AsyncResult,
         Batch,
         BatchItem,
+        BatchItemKind,
         BatchItemPage,
+        BatchItemStatus,
         BatchItemsResponse,
         BatchPage,
         CloneGraphResult,
@@ -46,6 +48,7 @@
         ErrorBody,
         Graph,
         GraphContextResponse,
+        GraphDataType,
         GraphDeleteResult,
         GraphPage,
         Instructions,
@@ -64,6 +67,7 @@
         Ontology,
         ProcessBatchResult,
         Project,
+        RoleType,
         SearchRequest,
         SubgraphResponse,
         Task,
@@ -100,7 +104,9 @@
     "BadRequestError": ".errors",
     "Batch": ".types",
     "BatchItem": ".types",
+    "BatchItemKind": ".types",
     "BatchItemPage": ".types",
+    "BatchItemStatus": ".types",
     "BatchItemsResponse": ".types",
     "BatchPage": ".types",
     "CloneGraphResult": ".types",
@@ -127,6 +133,7 @@
     "ErrorBody": ".types",
     "Graph": ".types",
     "GraphContextResponse": ".types",
+    "GraphDataType": ".types",
     "GraphDeleteResult": ".types",
     "GraphPage": ".types",
     "Instructions": ".types",
@@ -146,6 +153,7 @@
     "Ontology": ".types",
     "ProcessBatchResult": ".types",
     "Project": ".types",
+    "RoleType": ".types",
     "SearchRequest": ".types",
     "SubgraphResponse": ".types",
     "Task": ".types",
@@ -212,7 +220,9 @@ def __dir__():
     "BadRequestError",
     "Batch",
     "BatchItem",
+    "BatchItemKind",
     "BatchItemPage",
+    "BatchItemStatus",
     "BatchItemsResponse",
     "BatchPage",
     "CloneGraphResult",
@@ -239,6 +249,7 @@ def __dir__():
     "ErrorBody",
     "Graph",
     "GraphContextResponse",
+    "GraphDataType",
     "GraphDeleteResult",
     "GraphPage",
     "Instructions",
@@ -258,6 +269,7 @@ def __dir__():
     "Ontology",
     "ProcessBatchResult",
     "Project",
+    "RoleType",
     "SearchRequest",
     "SubgraphResponse",
     "Task",

src/zep_cloud/core/client_wrapper.py

@@ -27,12 +27,12 @@ def get_headers(self) -> typing.Dict[str, str]:
         import platform
 
         headers: typing.Dict[str, str] = {
-            "User-Agent": "zep-cloud/4.0.0a1",
+            "User-Agent": "zep-cloud/4.0.0a2",
             "X-Fern-Language": "Python",
             "X-Fern-Runtime": f"python/{platform.python_version()}",
             "X-Fern-Platform": f"{platform.system().lower()}/{platform.release()}",
             "X-Fern-SDK-Name": "zep-cloud",
-            "X-Fern-SDK-Version": "4.0.0a1",
+            "X-Fern-SDK-Version": "4.0.0a2",
             **(self.get_custom_headers() or {}),
         }
         headers["Authorization"] = f"Api-Key {self.api_key}"

src/zep_cloud/types/__init__.py

@@ -18,7 +18,9 @@
     from .async_result import AsyncResult
     from .batch import Batch
     from .batch_item import BatchItem
+    from .batch_item_kind import BatchItemKind
     from .batch_item_page import BatchItemPage
+    from .batch_item_status import BatchItemStatus
     from .batch_items_response import BatchItemsResponse
     from .batch_page import BatchPage
     from .clone_graph_result import CloneGraphResult
@@ -45,6 +47,7 @@
     from .error_body import ErrorBody
     from .graph import Graph
     from .graph_context_response import GraphContextResponse
+    from .graph_data_type import GraphDataType
     from .graph_delete_result import GraphDeleteResult
     from .graph_page import GraphPage
     from .instructions import Instructions
@@ -63,6 +66,7 @@
     from .ontology import Ontology
     from .process_batch_result import ProcessBatchResult
     from .project import Project
+    from .role_type import RoleType
     from .search_request import SearchRequest
     from .subgraph_response import SubgraphResponse
     from .task import Task
@@ -91,7 +95,9 @@
     "AsyncResult": ".async_result",
     "Batch": ".batch",
     "BatchItem": ".batch_item",
+    "BatchItemKind": ".batch_item_kind",
     "BatchItemPage": ".batch_item_page",
+    "BatchItemStatus": ".batch_item_status",
     "BatchItemsResponse": ".batch_items_response",
     "BatchPage": ".batch_page",
     "CloneGraphResult": ".clone_graph_result",
@@ -118,6 +124,7 @@
     "ErrorBody": ".error_body",
     "Graph": ".graph",
     "GraphContextResponse": ".graph_context_response",
+    "GraphDataType": ".graph_data_type",
     "GraphDeleteResult": ".graph_delete_result",
     "GraphPage": ".graph_page",
     "Instructions": ".instructions",
@@ -136,6 +143,7 @@
     "Ontology": ".ontology",
     "ProcessBatchResult": ".process_batch_result",
     "Project": ".project",
+    "RoleType": ".role_type",
     "SearchRequest": ".search_request",
     "SubgraphResponse": ".subgraph_response",
     "Task": ".task",
@@ -188,7 +196,9 @@ def __dir__():
     "AsyncResult",
     "Batch",
     "BatchItem",
+    "BatchItemKind",
     "BatchItemPage",
+    "BatchItemStatus",
     "BatchItemsResponse",
     "BatchPage",
     "CloneGraphResult",
@@ -215,6 +225,7 @@ def __dir__():
     "ErrorBody",
     "Graph",
     "GraphContextResponse",
+    "GraphDataType",
     "GraphDeleteResult",
     "GraphPage",
     "Instructions",
@@ -233,6 +244,7 @@ def __dir__():
     "Ontology",
     "ProcessBatchResult",
     "Project",
+    "RoleType",
     "SearchRequest",
     "SubgraphResponse",
     "Task",

src/zep_cloud/types/add_message.py

@@ -6,13 +6,14 @@
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .role_type import RoleType
 
 
 class AddMessage(UniversalBaseModel):
     content: typing.Optional[str] = None
     metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
     name: typing.Optional[str] = None
-    role: typing.Optional[str] = None
+    role: typing.Optional[RoleType] = None
     uuid_: typing_extensions.Annotated[
         typing.Optional[str], FieldMetadata(alias="uuid"), pydantic.Field(alias="uuid")
     ] = None

src/zep_cloud/types/batch_item.py

@@ -6,6 +6,8 @@
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .batch_item_kind import BatchItemKind
+from .batch_item_status import BatchItemStatus
 
 
 class BatchItem(UniversalBaseModel):
@@ -14,9 +16,9 @@ class BatchItem(UniversalBaseModel):
     graph_uuid: typing.Optional[str] = None
     sequence_index: typing.Optional[int] = None
     source_uuid: typing.Optional[str] = None
-    status: typing.Optional[str] = None
+    status: typing.Optional[BatchItemStatus] = None
     thread_uuid: typing.Optional[str] = None
-    type: typing.Optional[str] = None
+    type: typing.Optional[BatchItemKind] = None
     uuid_: typing_extensions.Annotated[
         typing.Optional[str], FieldMetadata(alias="uuid"), pydantic.Field(alias="uuid")
     ] = None

src/zep_cloud/types/batch_item_kind.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+BatchItemKind = typing.Union[typing.Literal["graph_episode", "thread_message"], typing.Any]

src/zep_cloud/types/batch_item_status.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+BatchItemStatus = typing.Union[
+    typing.Literal["pending", "queued", "processing", "succeeded", "failed", "skipped", "canceled"], typing.Any
+]

src/zep_cloud/types/context_episode.py

@@ -6,6 +6,8 @@
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .graph_data_type import GraphDataType
+from .role_type import RoleType
 
 
 class ContextEpisode(UniversalBaseModel):
@@ -16,11 +18,11 @@ class ContextEpisode(UniversalBaseModel):
     metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
     processed: typing.Optional[bool] = None
     relevance: typing.Optional[float] = None
-    role: typing.Optional[str] = None
+    role: typing.Optional[RoleType] = None
     role_name: typing.Optional[str] = None
     score: typing.Optional[float] = None
     selection_rank: typing.Optional[int] = None
-    source: typing.Optional[str] = None
+    source: typing.Optional[GraphDataType] = None
     source_description: typing.Optional[str] = None
     thread_uuid: typing.Optional[str] = None
     uuid_: typing_extensions.Annotated[

src/zep_cloud/types/episode.py

@@ -6,6 +6,8 @@
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .graph_data_type import GraphDataType
+from .role_type import RoleType
 
 
 class Episode(UniversalBaseModel):
@@ -16,10 +18,10 @@ class Episode(UniversalBaseModel):
     metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
     processed: typing.Optional[bool] = None
     relevance: typing.Optional[float] = None
-    role: typing.Optional[str] = None
+    role: typing.Optional[RoleType] = None
     role_name: typing.Optional[str] = None
     score: typing.Optional[float] = None
-    source: typing.Optional[str] = None
+    source: typing.Optional[GraphDataType] = None
     source_description: typing.Optional[str] = None
     thread_uuid: typing.Optional[str] = None
     uuid_: typing_extensions.Annotated[

src/zep_cloud/types/graph_data_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+GraphDataType = typing.Union[typing.Literal["text", "json", "message", "fact_triple"], typing.Any]

src/zep_cloud/types/message.py

@@ -6,6 +6,7 @@
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .role_type import RoleType
 
 
 class Message(UniversalBaseModel):
@@ -14,7 +15,7 @@ class Message(UniversalBaseModel):
     metadata: typing.Optional[typing.Dict[str, typing.Any]] = None
     name: typing.Optional[str] = None
     processed: typing.Optional[bool] = None
-    role: typing.Optional[str] = None
+    role: typing.Optional[RoleType] = None
     thread_uuid: typing.Optional[str] = None
     uuid_: typing_extensions.Annotated[
         typing.Optional[str], FieldMetadata(alias="uuid"), pydantic.Field(alias="uuid")

src/zep_cloud/types/role_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+RoleType = typing.Union[typing.Literal["system", "assistant", "user", "function", "tool"], typing.Any]