Revert "SDK regeneration"

This reverts commit c647733.

Author: paul-paliychuk

reference.md

@@ -1,15 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
-# isort: skip_file
-
 from .types import (
-    AddThreadMessagesResponse,
+    AddMemoryResponse,
     AddTripleResponse,
     ApiError,
+    ApidataDocument,
+    ApidataDocumentCollection,
+    ApidataDocumentSearchResponse,
+    ApidataDocumentWithScore,
+    ClassifySessionRequest,
     CloneGraphResponse,
     ComparisonOperator,
+    CreateDocumentRequest,
     DateFilter,
     EdgeType,
+    EndSessionResponse,
+    EndSessionsResponse,
     EntityEdge,
     EntityEdgeSourceTarget,
     EntityNode,
@@ -21,45 +27,65 @@
     EpisodeData,
     EpisodeMentions,
     EpisodeResponse,
+    Fact,
     FactRatingExamples,
     FactRatingInstruction,
-    Graph,
+    FactResponse,
+    FactsResponse,
     GraphDataType,
     GraphEdgesRequest,
     GraphNodesRequest,
     GraphSearchResults,
     GraphSearchScope,
+    Group,
+    GroupListResponse,
+    Memory,
+    MemorySearchResult,
     Message,
     MessageListResponse,
-    ModelsFactRatingExamples,
-    ModelsFactRatingInstruction,
+    NewFact,
+    Question,
     Reranker,
     RoleType,
     SearchFilters,
+    SearchScope,
+    SearchType,
+    Session,
+    SessionClassification,
+    SessionListResponse,
+    SessionSearchResponse,
+    SessionSearchResult,
     SuccessResponse,
-    Thread,
-    ThreadContextResponse,
-    ThreadListResponse,
+    Summary,
+    SummaryListResponse,
+    UpdateDocumentListRequest,
     User,
     UserListResponse,
     UserNodeResponse,
 )
-from .errors import BadRequestError, InternalServerError, NotFoundError
-from . import graph, thread, user
-from .client import AsyncZep, Zep
+from .errors import BadRequestError, ConflictError, InternalServerError, NotFoundError, UnauthorizedError
+from . import document, graph, group, memory, user
 from .environment import ZepEnvironment
 from .version import __version__
 
 __all__ = [
-    "AddThreadMessagesResponse",
+    "AddMemoryResponse",
     "AddTripleResponse",
     "ApiError",
-    "AsyncZep",
+    "ApidataDocument",
+    "ApidataDocumentCollection",
+    "ApidataDocumentSearchResponse",
+    "ApidataDocumentWithScore",
     "BadRequestError",
+    "ClassifySessionRequest",
     "CloneGraphResponse",
     "ComparisonOperator",
+    "ConflictError",
+    "CreateDocumentRequest",
     "DateFilter",
     "EdgeType",
+    "EndSessionResponse",
+    "EndSessionsResponse",
     "EntityEdge",
     "EntityEdgeSourceTarget",
     "EntityNode",
@@ -71,34 +97,49 @@
     "EpisodeData",
     "EpisodeMentions",
     "EpisodeResponse",
+    "Fact",
     "FactRatingExamples",
     "FactRatingInstruction",
-    "Graph",
+    "FactResponse",
+    "FactsResponse",
     "GraphDataType",
     "GraphEdgesRequest",
     "GraphNodesRequest",
     "GraphSearchResults",
     "GraphSearchScope",
+    "Group",
+    "GroupListResponse",
     "InternalServerError",
+    "Memory",
+    "MemorySearchResult",
     "Message",
     "MessageListResponse",
-    "ModelsFactRatingExamples",
-    "ModelsFactRatingInstruction",
+    "NewFact",
     "NotFoundError",
+    "Question",
     "Reranker",
     "RoleType",
     "SearchFilters",
+    "SearchScope",
+    "SearchType",
+    "Session",
+    "SessionClassification",
+    "SessionListResponse",
+    "SessionSearchResponse",
+    "SessionSearchResult",
     "SuccessResponse",
-    "Thread",
-    "ThreadContextResponse",
-    "ThreadListResponse",
+    "Summary",
+    "SummaryListResponse",
+    "UnauthorizedError",
+    "UpdateDocumentListRequest",
     "User",
     "UserListResponse",
     "UserNodeResponse",
-    "Zep",
     "ZepEnvironment",
     "__version__",
+    "document",
     "graph",
-    "thread",
+    "group",
+    "memory",
     "user",
 ]

requirements.txt

@@ -4,11 +4,14 @@
 import typing
 
 import httpx
+
 from .core.api_error import ApiError
 from .core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
+from .document.client import AsyncDocumentClient, DocumentClient
 from .environment import ZepEnvironment
 from .graph.client import AsyncGraphClient, GraphClient
-from .thread.client import AsyncThreadClient, ThreadClient
+from .group.client import AsyncGroupClient, GroupClient
+from .memory.client import AsyncMemoryClient, MemoryClient
 from .user.client import AsyncUserClient, UserClient
 
 
@@ -31,9 +34,6 @@ class BaseClient:
 
 
     api_key : typing.Optional[str]
-    headers : typing.Optional[typing.Dict[str, str]]
-        Additional headers to send with every request.
-
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
@@ -45,7 +45,7 @@ class BaseClient:
 
     Examples
     --------
-    from zep_cloud import Zep
+    from zep_cloud.client import Zep
 
     client = Zep(
         api_key="YOUR_API_KEY",
@@ -58,29 +58,27 @@ def __init__(
         base_url: typing.Optional[str] = None,
         environment: ZepEnvironment = ZepEnvironment.DEFAULT,
         api_key: typing.Optional[str] = os.getenv("ZEP_API_KEY"),
-        headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
-        httpx_client: typing.Optional[httpx.Client] = None,
+        httpx_client: typing.Optional[httpx.Client] = None
     ):
-        _defaulted_timeout = (
-            timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
-        )
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
         if api_key is None:
             raise ApiError(body="The client must be instantiated be either passing in api_key or setting ZEP_API_KEY")
         self._client_wrapper = SyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
             api_key=api_key,
-            headers=headers,
             httpx_client=httpx_client
             if httpx_client is not None
             else httpx.Client(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
             if follow_redirects is not None
             else httpx.Client(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.document = DocumentClient(client_wrapper=self._client_wrapper)
         self.graph = GraphClient(client_wrapper=self._client_wrapper)
-        self.thread = ThreadClient(client_wrapper=self._client_wrapper)
+        self.memory = MemoryClient(client_wrapper=self._client_wrapper)
+        self.group = GroupClient(client_wrapper=self._client_wrapper)
         self.user = UserClient(client_wrapper=self._client_wrapper)
 
 
@@ -103,9 +101,6 @@ class AsyncBaseClient:
 
 
     api_key : typing.Optional[str]
-    headers : typing.Optional[typing.Dict[str, str]]
-        Additional headers to send with every request.
-
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
 
@@ -117,7 +112,7 @@ class AsyncBaseClient:
 
     Examples
     --------
-    from zep_cloud import AsyncZep
+    from zep_cloud.client import AsyncZep
 
     client = AsyncZep(
         api_key="YOUR_API_KEY",
@@ -130,29 +125,27 @@ def __init__(
         base_url: typing.Optional[str] = None,
         environment: ZepEnvironment = ZepEnvironment.DEFAULT,
         api_key: typing.Optional[str] = os.getenv("ZEP_API_KEY"),
-        headers: typing.Optional[typing.Dict[str, str]] = None,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
-        httpx_client: typing.Optional[httpx.AsyncClient] = None,
+        httpx_client: typing.Optional[httpx.AsyncClient] = None
     ):
-        _defaulted_timeout = (
-            timeout if timeout is not None else 60 if httpx_client is None else httpx_client.timeout.read
-        )
+        _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
         if api_key is None:
             raise ApiError(body="The client must be instantiated be either passing in api_key or setting ZEP_API_KEY")
         self._client_wrapper = AsyncClientWrapper(
             base_url=_get_base_url(base_url=base_url, environment=environment),
             api_key=api_key,
-            headers=headers,
             httpx_client=httpx_client
             if httpx_client is not None
             else httpx.AsyncClient(timeout=_defaulted_timeout, follow_redirects=follow_redirects)
             if follow_redirects is not None
             else httpx.AsyncClient(timeout=_defaulted_timeout),
             timeout=_defaulted_timeout,
         )
+        self.document = AsyncDocumentClient(client_wrapper=self._client_wrapper)
         self.graph = AsyncGraphClient(client_wrapper=self._client_wrapper)
-        self.thread = AsyncThreadClient(client_wrapper=self._client_wrapper)
+        self.memory = AsyncMemoryClient(client_wrapper=self._client_wrapper)
+        self.group = AsyncGroupClient(client_wrapper=self._client_wrapper)
         self.user = AsyncUserClient(client_wrapper=self._client_wrapper)
 
 

src/zep_cloud/__init__.py

@@ -1,52 +1,30 @@
 # This file was auto-generated by Fern from our API Definition.
 
-# isort: skip_file
-
 from .api_error import ApiError
 from .client_wrapper import AsyncClientWrapper, BaseClientWrapper, SyncClientWrapper
 from .datetime_utils import serialize_datetime
-from .file import File, convert_file_dict_to_httpx_tuples, with_content_type
+from .file import File, convert_file_dict_to_httpx_tuples
 from .http_client import AsyncHttpClient, HttpClient
-from .http_response import AsyncHttpResponse, HttpResponse
 from .jsonable_encoder import jsonable_encoder
-from .pydantic_utilities import (
-    IS_PYDANTIC_V2,
-    UniversalBaseModel,
-    UniversalRootModel,
-    parse_obj_as,
-    universal_field_validator,
-    universal_root_validator,
-    update_forward_refs,
-)
+from .pydantic_utilities import deep_union_pydantic_dicts, pydantic_v1
 from .query_encoder import encode_query
 from .remove_none_from_dict import remove_none_from_dict
 from .request_options import RequestOptions
-from .serialization import FieldMetadata, convert_and_respect_annotation_metadata
 
 __all__ = [
     "ApiError",
     "AsyncClientWrapper",
     "AsyncHttpClient",
-    "AsyncHttpResponse",
     "BaseClientWrapper",
-    "FieldMetadata",
     "File",
     "HttpClient",
-    "HttpResponse",
-    "IS_PYDANTIC_V2",
     "RequestOptions",
     "SyncClientWrapper",
-    "UniversalBaseModel",
-    "UniversalRootModel",
-    "convert_and_respect_annotation_metadata",
     "convert_file_dict_to_httpx_tuples",
+    "deep_union_pydantic_dicts",
     "encode_query",
     "jsonable_encoder",
-    "parse_obj_as",
+    "pydantic_v1",
     "remove_none_from_dict",
     "serialize_datetime",
-    "universal_field_validator",
-    "universal_root_validator",
-    "update_forward_refs",
-    "with_content_type",
 ]

src/zep_cloud/base_client.py

@@ -1,23 +1,15 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from typing import Any, Dict, Optional
+import typing
 
 
 class ApiError(Exception):
-    headers: Optional[Dict[str, str]]
-    status_code: Optional[int]
-    body: Any
+    status_code: typing.Optional[int]
+    body: typing.Any
 
-    def __init__(
-        self,
-        *,
-        headers: Optional[Dict[str, str]] = None,
-        status_code: Optional[int] = None,
-        body: Any = None,
-    ) -> None:
-        self.headers = headers
+    def __init__(self, *, status_code: typing.Optional[int] = None, body: typing.Any = None):
         self.status_code = status_code
         self.body = body
 
     def __str__(self) -> str:
-        return f"headers: {self.headers}, status_code: {self.status_code}, body: {self.body}"
+        return f"status_code: {self.status_code}, body: {self.body}"