Return episode as a result of graph.add (#238)

* SDK regeneration

* SDK regeneration

* chore: Version bump

* SDK regeneration

* SDK regeneration

* SDK regeneration

* SDK regeneration

* SDK regeneration

* SDK regeneration

* SDK regeneration

* SDK regeneration

* chore: Version bump

* poetry lock

* chore: version bump

* SDK regeneration

* chore: Version bump

* SDK regeneration

---------

Co-authored-by: fern-api <115122769+fern-api[bot]@users.noreply.github.com>

Author: paul-paliychuk

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zep-cloud"
-version = "2.6.0"
+version = "2.6.1"
 description = ""
 readme = "README.md"
 authors = []

src/zep_cloud/core/client_wrapper.py

@@ -17,7 +17,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "zep-cloud",
-            "X-Fern-SDK-Version": "2.6.0",
+            "X-Fern-SDK-Version": "2.6.1",
         }
         headers["Authorization"] = f"Api-Key {self.api_key}"
         return headers

src/zep_cloud/graph/client.py

@@ -11,12 +11,12 @@
 from ..errors.internal_server_error import InternalServerError
 from ..types.add_triple_response import AddTripleResponse
 from ..types.api_error import ApiError as types_api_error_ApiError
+from ..types.episode import Episode
 from ..types.graph_data_type import GraphDataType
 from ..types.graph_search_results import GraphSearchResults
 from ..types.graph_search_scope import GraphSearchScope
 from ..types.reranker import Reranker
 from ..types.search_filters import SearchFilters
-from ..types.success_response import SuccessResponse
 from .edge.client import AsyncEdgeClient, EdgeClient
 from .episode.client import AsyncEpisodeClient, EpisodeClient
 from .node.client import AsyncNodeClient, NodeClient
@@ -40,7 +40,7 @@ def add(
         type: typing.Optional[GraphDataType] = OMIT,
         user_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None
-    ) -> SuccessResponse:
+    ) -> Episode:
         """
         Add data to the graph. Note: each subscription tier has different limits on the amount of data that can be added to the graph please refer to the pricing page for more information.
 
@@ -59,8 +59,8 @@ def add(
 
         Returns
         -------
-        SuccessResponse
-            Data added
+        Episode
+            Added episode
 
         Examples
         --------
@@ -80,7 +80,7 @@ def add(
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(SuccessResponse, _response.json())  # type: ignore
+                return pydantic_v1.parse_obj_as(Episode, _response.json())  # type: ignore
             if _response.status_code == 400:
                 raise BadRequestError(
                     pydantic_v1.parse_obj_as(types_api_error_ApiError, _response.json())  # type: ignore
@@ -341,7 +341,7 @@ async def add(
         type: typing.Optional[GraphDataType] = OMIT,
         user_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None
-    ) -> SuccessResponse:
+    ) -> Episode:
         """
         Add data to the graph. Note: each subscription tier has different limits on the amount of data that can be added to the graph please refer to the pricing page for more information.
 
@@ -360,8 +360,8 @@ async def add(
 
         Returns
         -------
-        SuccessResponse
-            Data added
+        Episode
+            Added episode
 
         Examples
         --------
@@ -389,7 +389,7 @@ async def main() -> None:
         )
         try:
             if 200 <= _response.status_code < 300:
-                return pydantic_v1.parse_obj_as(SuccessResponse, _response.json())  # type: ignore
+                return pydantic_v1.parse_obj_as(Episode, _response.json())  # type: ignore
             if _response.status_code == 400:
                 raise BadRequestError(
                     pydantic_v1.parse_obj_as(types_api_error_ApiError, _response.json())  # type: ignore