Skip to content

Commit df57725

Browse files
use zep cloud rather than zep ce
1 parent 51a604e commit df57725

File tree

3 files changed

+12
-14
lines changed

3 files changed

+12
-14
lines changed

poetry.lock

Lines changed: 6 additions & 6 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ python = ">=3.9,<4.0"
2121
openai = "^1.61.1"
2222
pydantic = "^2.10.6"
2323
motor = "^3.7.0"
24-
zep-python = "^2.0.2"
24+
zep-cloud = "^2.3.1"
2525
requests = "^2.32.3"
2626
pandas = "^2.2.3"
2727
qdrant-client = "^1.13.2"

solana_agent/ai.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,9 @@
1010
from typing_extensions import override
1111
import inspect
1212
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
1616
import pandas as pd
1717

1818

@@ -70,7 +70,6 @@ def __init__(
7070
instructions: str,
7171
database: Any,
7272
zep_api_key: str = None,
73-
zep_base_url: str = None,
7473
perplexity_api_key: str = None,
7574
grok_api_key: str = None,
7675
gemini_api_key: str = None,
@@ -85,7 +84,6 @@ def __init__(
8584
instructions (str): Base behavioral instructions for the AI
8685
database (Any): Database instance for message/thread storage
8786
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
8987
perplexity_api_key (str, optional): API key for Perplexity search. Defaults to None
9088
grok_api_key (str, optional): API key for X/Twitter search via Grok. Defaults to None
9189
gemini_api_key (str, optional): API key for Google Gemini. Defaults to None
@@ -113,12 +111,12 @@ def __init__(
113111
self._database = database
114112
self._accumulated_value_queue = asyncio.Queue()
115113
self._zep = (
116-
AsyncZep(api_key=zep_api_key, base_url=zep_base_url)
114+
AsyncZep(api_key=zep_api_key)
117115
if zep_api_key
118116
else None
119117
)
120118
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
122120
)
123121
self._perplexity_api_key = perplexity_api_key
124122
self._grok_api_key = grok_api_key

0 commit comments

Comments
 (0)