|
1 | 1 | import os |
2 | 2 |
|
3 | | -from dotenv import load_dotenv |
4 | | - |
5 | 3 | from memu.app import MemoryService |
6 | 4 |
|
7 | | -load_dotenv() |
8 | | - |
9 | 5 |
|
10 | 6 | async def main(): |
11 | 7 | """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") |
14 | 11 | file_path = os.path.abspath("example/example_conversation.json") |
15 | 12 |
|
16 | 13 | print("\n" + "=" * 60) |
17 | 14 | print("[INMEMORY] Starting test...") |
18 | 15 | print("=" * 60) |
19 | 16 |
|
20 | 17 | 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 | + # }, |
37 | 32 | database_config={ |
38 | 33 | "metadata_store": {"provider": "inmemory"}, |
39 | 34 | }, |
|
0 commit comments