File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ,
Original file line number Diff line number Diff line change 11import typing
22
3- from zep_cloud import ApidataEdgeType as EdgeType , ApidataEntityEdgeSourceTarget as EntityEdgeSourceTarget
3+ from zep_cloud import EdgeType , EntityEdgeSourceTarget
44from zep_cloud .core .client_wrapper import AsyncClientWrapper , SyncClientWrapper
55from zep_cloud .external_clients .ontology import (
66 EdgeModel ,
99)
1010from zep_cloud .graph .client import AsyncGraphClient as AsyncBaseGraphClient
1111from 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
1414if typing .TYPE_CHECKING :
1515 from zep_cloud .external_clients .ontology import EntityModel
Original file line number Diff line number Diff line change 22from typing import List , Optional
33
44from 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
77DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
88
Original file line number Diff line number Diff line change 33
44import pytest
55
6- from zep_cloud import GraphitiEntityEdge as EntityEdge , GraphitiEntityNode as EntityNode , ApidataGraphEpisode as Episode
6+ from zep_cloud import EntityEdge , EntityNode , Episode
77from zep_cloud .graph .utils import compose_context_string , format_edge_date_range
88
99
You can’t perform that action at this time.
0 commit comments