10
10
from typing_extensions import override
11
11
import inspect
12
12
import requests
13
- from zep_python .client import AsyncZep
14
- from zep_python .client import Zep
15
- from zep_python .types import Message , RoleType
13
+ from zep_cloud .client import AsyncZep
14
+ from zep_cloud .client import Zep
15
+ from zep_cloud .types import Message , RoleType
16
16
import pandas as pd
17
17
18
18
@@ -70,7 +70,6 @@ def __init__(
70
70
instructions : str ,
71
71
database : Any ,
72
72
zep_api_key : str = None ,
73
- zep_base_url : str = None ,
74
73
perplexity_api_key : str = None ,
75
74
grok_api_key : str = None ,
76
75
gemini_api_key : str = None ,
@@ -85,7 +84,6 @@ def __init__(
85
84
instructions (str): Base behavioral instructions for the AI
86
85
database (Any): Database instance for message/thread storage
87
86
zep_api_key (str, optional): API key for Zep memory integration. Defaults to None
88
- zep_base_url (str, optional): Base URL for Zep API. Defaults to None
89
87
perplexity_api_key (str, optional): API key for Perplexity search. Defaults to None
90
88
grok_api_key (str, optional): API key for X/Twitter search via Grok. Defaults to None
91
89
gemini_api_key (str, optional): API key for Google Gemini. Defaults to None
@@ -113,12 +111,12 @@ def __init__(
113
111
self ._database = database
114
112
self ._accumulated_value_queue = asyncio .Queue ()
115
113
self ._zep = (
116
- AsyncZep (api_key = zep_api_key , base_url = zep_base_url )
114
+ AsyncZep (api_key = zep_api_key )
117
115
if zep_api_key
118
116
else None
119
117
)
120
118
self ._sync_zep = (
121
- Zep (api_key = zep_api_key , base_url = zep_base_url ) if zep_api_key else None
119
+ Zep (api_key = zep_api_key ) if zep_api_key else None
122
120
)
123
121
self ._perplexity_api_key = perplexity_api_key
124
122
self ._grok_api_key = grok_api_key
0 commit comments