|
50 | 50 | "config": openapi_config, |
51 | 51 | }, |
52 | 52 | "embedder": { |
53 | | - "backend": "universal_api", |
| 53 | + "backend": "ollama", |
54 | 54 | "config": { |
55 | | - "provider": "openai", |
56 | | - "api_key": "sk-xxxxxxxxx", |
57 | | - "model_name_or_path": "text-embedding-3-large", |
58 | | - "base_url": "http://xxx.xxx.xxx.xxx:xxxx/v1" |
| 55 | + "model_name_or_path": "nomic-embed-text:latest", |
59 | 56 | }, |
60 | 57 | }, |
61 | 58 | "chunker": { |
|
108 | 105 | "config": { |
109 | 106 | "uri": neo4j_uri, |
110 | 107 | "user": "neo4j", |
111 | | - "password": "iaarlichunyu", |
| 108 | + "password": "12345678", |
112 | 109 | "db_name": f"db{user_name.replace('-', '')}", |
113 | 110 | "auto_create": True, |
114 | 111 | }, |
115 | 112 | }, |
116 | 113 | "embedder": { |
117 | | - "backend": "universal_api", |
| 114 | + "backend": "ollama", |
118 | 115 | "config": { |
119 | | - "provider": "openai", |
120 | | - "api_key": "sk-xxxxxxxxx", |
121 | | - "model_name_or_path": "text-embedding-3-large", |
122 | | - "base_url": "http://xxx.xxx.xxx.xxx:xxxx/v1" |
| 116 | + "model_name_or_path": "nomic-embed-text:latest", |
123 | 117 | }, |
124 | 118 | }, |
125 | | - "reorganize": True, |
126 | 119 | }, |
127 | 120 | }, |
128 | 121 | "act_mem": {}, |
@@ -266,12 +259,11 @@ def filter_memory_data(memories_data): |
266 | 259 | # 7. Add document |
267 | 260 | print(f"🚀 [{datetime.now().strftime('%H:%M:%S')}] Starting to add document...") |
268 | 261 | start_time = time.time() |
269 | | - |
270 | 262 | ## 7.1 add pdf for ./tmp/data if use doc mem mos.add(doc_path="./tmp/data/") |
271 | | -mos.add(doc_path="examples/data/docs/test/") |
| 263 | +start_time = time.time() |
| 264 | + |
272 | 265 | get_all_results = mos.get_all() |
273 | 266 | filtered_results = filter_memory_data(get_all_results) |
274 | | - |
275 | 267 | print(f"Get all results after add doc: {filtered_results}") |
276 | 268 |
|
277 | 269 | print( |
|
0 commit comments