Skip to content

Commit 0c8f829

Browse files
Revert "fix: update imports for Fern-regenerated type names"
This reverts commit d4a6e71.
1 parent bc76826 commit 0c8f829

4 files changed

Lines changed: 4 additions & 8 deletions

File tree

‎src/zep_cloud/client.py‎

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,6 @@ def __init__(
2222
env_api_url = os.getenv("ZEP_API_URL")
2323
if env_api_url:
2424
base_url = f"{env_api_url}/api/v2"
25-
if api_key is None:
26-
raise ValueError("api_key is required. Set ZEP_API_KEY env var or pass api_key.")
2725
super().__init__(
2826
base_url=base_url,
2927
environment=environment,
@@ -49,8 +47,6 @@ def __init__(
4947
env_api_url = os.getenv("ZEP_API_URL")
5048
if env_api_url:
5149
base_url = f"{env_api_url}/api/v2"
52-
if api_key is None:
53-
raise ValueError("api_key is required. Set ZEP_API_KEY env var or pass api_key.")
5450
super().__init__(
5551
base_url=base_url,
5652
environment=environment,

‎src/zep_cloud/external_clients/graph.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import typing
22

3-
from zep_cloud import ApidataEdgeType as EdgeType, ApidataEntityEdgeSourceTarget as EntityEdgeSourceTarget
3+
from zep_cloud import EdgeType, EntityEdgeSourceTarget
44
from zep_cloud.core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
55
from zep_cloud.external_clients.ontology import (
66
EdgeModel,
@@ -9,7 +9,7 @@
99
)
1010
from zep_cloud.graph.client import AsyncGraphClient as AsyncBaseGraphClient
1111
from zep_cloud.graph.client import GraphClient as BaseGraphClient
12-
from zep_cloud.types import ApidataEntityType as EntityType
12+
from zep_cloud.types import EntityType
1313

1414
if typing.TYPE_CHECKING:
1515
from zep_cloud.external_clients.ontology import EntityModel

‎src/zep_cloud/graph/utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from typing import List, Optional
33

44
from dateutil import parser as dateutil_parser
5-
from zep_cloud import GraphitiEntityEdge as EntityEdge, GraphitiEntityNode as EntityNode, ApidataGraphEpisode as Episode
5+
from zep_cloud import EntityEdge, EntityNode, Episode
66

77
DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
88

‎tests/graph/test_utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
import pytest
55

6-
from zep_cloud import GraphitiEntityEdge as EntityEdge, GraphitiEntityNode as EntityNode, ApidataGraphEpisode as Episode
6+
from zep_cloud import EntityEdge, EntityNode, Episode
77
from zep_cloud.graph.utils import compose_context_string, format_edge_date_range
88

99

0 commit comments

Comments
 (0)