Task management API

pyproject.toml

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

reference.md

@@ -767,6 +767,7 @@ client = Zep(
 client.graph.add_fact_triple(
     fact="fact",
     fact_name="fact_name",
+    source_node_name="source_node_name",
     target_node_name="target_node_name",
 )
 
@@ -800,55 +801,55 @@ client.graph.add_fact_triple(
 <dl>
 <dd>
 
-**target_node_name:** `str` — The name of the target node to add
+**source_node_name:** `str` — The name of the source node to add
 
 </dd>
 </dl>
 
 <dl>
 <dd>
 
-**created_at:** `typing.Optional[str]` — The timestamp of the message
+**target_node_name:** `str` — The name of the target node to add
 
 </dd>
 </dl>
 
 <dl>
 <dd>
 
-**expired_at:** `typing.Optional[str]` — The time (if any) at which the edge expires
+**created_at:** `typing.Optional[str]` — The timestamp of the message
 
 </dd>
 </dl>
 
 <dl>
 <dd>
 
-**fact_uuid:** `typing.Optional[str]` — The uuid of the edge to add
+**expired_at:** `typing.Optional[str]` — The time (if any) at which the edge expires
 
 </dd>
 </dl>
 
 <dl>
 <dd>
 
-**graph_id:** `typing.Optional[str]` 
+**fact_uuid:** `typing.Optional[str]` — The uuid of the edge to add
 
 </dd>
 </dl>
 
 <dl>
 <dd>
 
-**invalid_at:** `typing.Optional[str]` — The time (if any) at which the fact stops being true
+**graph_id:** `typing.Optional[str]` 
 
 </dd>
 </dl>
 
 <dl>
 <dd>
 
-**source_node_name:** `typing.Optional[str]` — The name of the source node to add
+**invalid_at:** `typing.Optional[str]` — The time (if any) at which the fact stops being true
 
 </dd>
 </dl>
@@ -1630,6 +1631,77 @@ client.project.get()
 </dl>
 
 
+</dd>
+</dl>
+</details>
+
+## Task
+<details><summary><code>client.task.<a href="src/zep_cloud/task/client.py">get</a>(...)</code></summary>
+<dl>
+<dd>
+
+#### 📝 Description
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+Gets a task by its ID
+</dd>
+</dl>
+</dd>
+</dl>
+
+#### 🔌 Usage
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+```python
+from zep_cloud import Zep
+
+client = Zep(
+    api_key="YOUR_API_KEY",
+)
+client.task.get(
+    task_id="task_id",
+)
+
+```
+</dd>
+</dl>
+</dd>
+</dl>
+
+#### ⚙️ Parameters
+
+<dl>
+<dd>
+
+<dl>
+<dd>
+
+**task_id:** `str` — Task ID
+
+</dd>
+</dl>
+
+<dl>
+<dd>
+
+**request_options:** `typing.Optional[RequestOptions]` — Request-specific configuration.
+
+</dd>
+</dl>
+</dd>
+</dl>
+
+
 </dd>
 </dl>
 </details>

src/zep_cloud/__init__.py

@@ -25,6 +25,7 @@
     EpisodeResponse,
     FactRatingExamples,
     FactRatingInstruction,
+    GetTaskResponse,
     Graph,
     GraphDataType,
     GraphEdgesRequest,
@@ -44,6 +45,8 @@
     RoleType,
     SearchFilters,
     SuccessResponse,
+    TaskErrorResponse,
+    TaskProgress,
     Thread,
     ThreadContextResponse,
     ThreadListResponse,
@@ -53,7 +56,7 @@
     UserNodeResponse,
 )
 from .errors import BadRequestError, InternalServerError, NotFoundError
-from . import context, graph, project, thread, user
+from . import context, graph, project, task, thread, user
 from .client import AsyncZep, Zep
 from .environment import ZepEnvironment
 from .thread import ThreadGetUserContextRequestMode
@@ -84,6 +87,7 @@
     "EpisodeResponse",
     "FactRatingExamples",
     "FactRatingInstruction",
+    "GetTaskResponse",
     "Graph",
     "GraphDataType",
     "GraphEdgesRequest",
@@ -105,6 +109,8 @@
     "RoleType",
     "SearchFilters",
     "SuccessResponse",
+    "TaskErrorResponse",
+    "TaskProgress",
     "Thread",
     "ThreadContextResponse",
     "ThreadGetUserContextRequestMode",
@@ -119,6 +125,7 @@
     "context",
     "graph",
     "project",
+    "task",
     "thread",
     "user",
 ]

src/zep_cloud/base_client.py

@@ -10,6 +10,7 @@
 from .environment import ZepEnvironment
 from .graph.client import AsyncGraphClient, GraphClient
 from .project.client import AsyncProjectClient, ProjectClient
+from .task.client import AsyncTaskClient, TaskClient
 from .thread.client import AsyncThreadClient, ThreadClient
 from .user.client import AsyncUserClient, UserClient
 
@@ -84,6 +85,7 @@ def __init__(
         self.context = ContextClient(client_wrapper=self._client_wrapper)
         self.graph = GraphClient(client_wrapper=self._client_wrapper)
         self.project = ProjectClient(client_wrapper=self._client_wrapper)
+        self.task = TaskClient(client_wrapper=self._client_wrapper)
         self.thread = ThreadClient(client_wrapper=self._client_wrapper)
         self.user = UserClient(client_wrapper=self._client_wrapper)
 
@@ -158,6 +160,7 @@ def __init__(
         self.context = AsyncContextClient(client_wrapper=self._client_wrapper)
         self.graph = AsyncGraphClient(client_wrapper=self._client_wrapper)
         self.project = AsyncProjectClient(client_wrapper=self._client_wrapper)
+        self.task = AsyncTaskClient(client_wrapper=self._client_wrapper)
         self.thread = AsyncThreadClient(client_wrapper=self._client_wrapper)
         self.user = AsyncUserClient(client_wrapper=self._client_wrapper)
 

src/zep_cloud/core/client_wrapper.py

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

src/zep_cloud/graph/client.py

@@ -256,13 +256,13 @@ def add_fact_triple(
         *,
         fact: str,
         fact_name: str,
+        source_node_name: str,
         target_node_name: str,
         created_at: typing.Optional[str] = OMIT,
         expired_at: typing.Optional[str] = OMIT,
         fact_uuid: typing.Optional[str] = OMIT,
         graph_id: typing.Optional[str] = OMIT,
         invalid_at: typing.Optional[str] = OMIT,
-        source_node_name: typing.Optional[str] = OMIT,
         source_node_summary: typing.Optional[str] = OMIT,
         source_node_uuid: typing.Optional[str] = OMIT,
         target_node_summary: typing.Optional[str] = OMIT,
@@ -282,6 +282,9 @@ def add_fact_triple(
         fact_name : str
             The name of the edge to add. Should be all caps using snake case (eg RELATES_TO)
 
+        source_node_name : str
+            The name of the source node to add
+
         target_node_name : str
             The name of the target node to add
 
@@ -299,9 +302,6 @@ def add_fact_triple(
         invalid_at : typing.Optional[str]
             The time (if any) at which the fact stops being true
 
-        source_node_name : typing.Optional[str]
-            The name of the source node to add
-
         source_node_summary : typing.Optional[str]
             The summary of the source node to add
 
@@ -337,19 +337,20 @@ def add_fact_triple(
         client.graph.add_fact_triple(
             fact="fact",
             fact_name="fact_name",
+            source_node_name="source_node_name",
             target_node_name="target_node_name",
         )
         """
         _response = self._raw_client.add_fact_triple(
             fact=fact,
             fact_name=fact_name,
+            source_node_name=source_node_name,
             target_node_name=target_node_name,
             created_at=created_at,
             expired_at=expired_at,
             fact_uuid=fact_uuid,
             graph_id=graph_id,
             invalid_at=invalid_at,
-            source_node_name=source_node_name,
             source_node_summary=source_node_summary,
             source_node_uuid=source_node_uuid,
             target_node_summary=target_node_summary,
@@ -972,13 +973,13 @@ async def add_fact_triple(
         *,
         fact: str,
         fact_name: str,
+        source_node_name: str,
         target_node_name: str,
         created_at: typing.Optional[str] = OMIT,
         expired_at: typing.Optional[str] = OMIT,
         fact_uuid: typing.Optional[str] = OMIT,
         graph_id: typing.Optional[str] = OMIT,
         invalid_at: typing.Optional[str] = OMIT,
-        source_node_name: typing.Optional[str] = OMIT,
         source_node_summary: typing.Optional[str] = OMIT,
         source_node_uuid: typing.Optional[str] = OMIT,
         target_node_summary: typing.Optional[str] = OMIT,
@@ -998,6 +999,9 @@ async def add_fact_triple(
         fact_name : str
             The name of the edge to add. Should be all caps using snake case (eg RELATES_TO)
 
+        source_node_name : str
+            The name of the source node to add
+
         target_node_name : str
             The name of the target node to add
 
@@ -1015,9 +1019,6 @@ async def add_fact_triple(
         invalid_at : typing.Optional[str]
             The time (if any) at which the fact stops being true
 
-        source_node_name : typing.Optional[str]
-            The name of the source node to add
-
         source_node_summary : typing.Optional[str]
             The summary of the source node to add
 
@@ -1058,6 +1059,7 @@ async def main() -> None:
             await client.graph.add_fact_triple(
                 fact="fact",
                 fact_name="fact_name",
+                source_node_name="source_node_name",
                 target_node_name="target_node_name",
             )
 
@@ -1067,13 +1069,13 @@ async def main() -> None:
         _response = await self._raw_client.add_fact_triple(
             fact=fact,
             fact_name=fact_name,
+            source_node_name=source_node_name,
             target_node_name=target_node_name,
             created_at=created_at,
             expired_at=expired_at,
             fact_uuid=fact_uuid,
             graph_id=graph_id,
             invalid_at=invalid_at,
-            source_node_name=source_node_name,
             source_node_summary=source_node_summary,
             source_node_uuid=source_node_uuid,
             target_node_summary=target_node_summary,