Skip to content

Commit 45f447b

Browse files
committed
revert: restore tests/test_inmemory.py removed by mistake
1 parent 17a0478 commit 45f447b

File tree

1 file changed

+17
-22
lines changed

1 file changed

+17
-22
lines changed

tests/test_inmemory.py

Lines changed: 17 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -1,39 +1,34 @@
11
import os
22

3-
from dotenv import load_dotenv
4-
53
from memu.app import MemoryService
64

7-
load_dotenv()
8-
95

106
async def main():
117
"""Test with in-memory storage (default)."""
12-
zenmux_api_key = os.environ.get("ZENMUX_API_KEY")
13-
openai_api_key = os.environ.get("OPENAI_API_KEY")
8+
api_key = os.environ.get("OPENAI_API_KEY")
9+
# dashscope_api_key = os.environ.get("DASHSCOPE_API_KEY")
10+
# voyage_api_key = os.environ.get("VOYAGE_API_KEY")
1411
file_path = os.path.abspath("example/example_conversation.json")
1512

1613
print("\n" + "=" * 60)
1714
print("[INMEMORY] Starting test...")
1815
print("=" * 60)
1916

2017
service = MemoryService(
21-
llm_profiles={
22-
"default": {
23-
"provider": "openai",
24-
"base_url": "https://zenmux.ai/api/v1",
25-
"api_key": zenmux_api_key,
26-
"chat_model": "openai/gpt-4o-mini",
27-
"client_backend": "sdk",
28-
},
29-
"embedding": {
30-
"provider": "openai",
31-
"base_url": "https://api.openai.com/v1",
32-
"api_key": openai_api_key,
33-
"embed_model": "text-embedding-3-small",
34-
"client_backend": "sdk",
35-
},
36-
},
18+
llm_profiles={"default": {"api_key": api_key}},
19+
# llm_profiles={
20+
# "default": {
21+
# "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
22+
# "api_key": dashscope_api_key,
23+
# "chat_model": "qwen3-max",
24+
# "client_backend": "sdk"
25+
# },
26+
# "embedding": {
27+
# "base_url": "https://api.voyageai.com/v1",
28+
# "api_key": voyage_api_key,
29+
# "embed_model": "voyage-3.5-lite"
30+
# }
31+
# },
3732
database_config={
3833
"metadata_store": {"provider": "inmemory"},
3934
},

0 commit comments

Comments
 (0)