MemoryOS 1.0.1__py3-none-any.whl → 1.1.2__py3-none-any.whl

{memoryos-1.0.1.dist-info → memoryos-1.1.2.dist-info}/METADATA

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: MemoryOS
-Version: 1.0.1
+Version: 1.1.2
 Summary: Intelligence Begins with Memory
 License: Apache-2.0
+License-File: LICENSE
 Keywords: memory,llm,language model,memoryOS,agent,kv cache,lora
 Author: MemTensor
 Author-email: MemTensor@memtensor.cn
@@ -22,6 +23,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Provides-Extra: all
 Provides-Extra: mem-reader
 Provides-Extra: mem-scheduler
+Provides-Extra: mem-user
 Provides-Extra: tree-mem
 Requires-Dist: chonkie (>=1.0.7,<2.0.0) ; extra == "all"
 Requires-Dist: chonkie (>=1.0.7,<2.0.0) ; extra == "mem-reader"
@@ -35,6 +37,9 @@ Requires-Dist: ollama (>=0.4.8,<0.5.0)
 Requires-Dist: openai (>=1.77.0,<2.0.0)
 Requires-Dist: pika (>=1.3.2,<2.0.0) ; extra == "all"
 Requires-Dist: pika (>=1.3.2,<2.0.0) ; extra == "mem-scheduler"
+Requires-Dist: pymysql (>=1.1.0,<2.0.0)
+Requires-Dist: pymysql (>=1.1.0,<2.0.0) ; extra == "all"
+Requires-Dist: pymysql (>=1.1.0,<2.0.0) ; extra == "mem-user"
 Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
 Requires-Dist: qdrant-client (>=1.14.2,<2.0.0) ; extra == "all"
 Requires-Dist: redis (>=6.2.0,<7.0.0) ; extra == "all"

{memoryos-1.0.1.dist-info → memoryos-1.1.2.dist-info}/RECORD

@@ -1,16 +1,15 @@
-memos/__init__.py,sha256=EKtqQioVRZ2gNPD65cxhBbX-ZhswH1EBGMj2G6roYME,575
-memos/api/config.py,sha256=Gx3g5jPZ_8EUPTTvA7h14tKL7VYe4FI4NybCTuylWuE,24513
-memos/api/context/context.py,sha256=HBMeg3-e-rOJXMU7aR0JCsEIECuefEx8eyFT-1sSG9c,4158
-memos/api/context/context_thread.py,sha256=T3Vdu7m1l8JMmoUBop1ykyrbH-oZ-_6FnymXsawmLis,3121
-memos/api/context/dependencies.py,sha256=byLwmFsocQufcmi5WFg2j6XjWv9k4PCozCV7uRdCO3M,2468
+memos/__init__.py,sha256=UskX96T9J0ZaLp9Ubx5fFFtJcY6qE1BeLumraYRyFCk,575
+memos/api/client.py,sha256=zaXBRlAlGX28b5xWKilal-zoR2DYLRIYxo47Q3LdEFU,4045
+memos/api/config.py,sha256=SCUtLtYOebne73Bj3QHQR29L1JtftSWR4teTpd8A7gA,24732
+memos/api/context/dependencies.py,sha256=G9vfMVEeQczSd7ma4hlXSk2S8K6txjXHVO27R6FlbjQ,1240
 memos/api/exceptions.py,sha256=MfTNXwLwJE7tHwNVTHlJsB3kF14oA0JfA1-S8xrZEus,833
 memos/api/mcp_serve.py,sha256=GtJDZYx4yxKMs4sCddKJbRAjgsbRunr6S2nAT-KeBMs,21270
-memos/api/middleware/request_context.py,sha256=8tkyIQkgBnGRM_X33CaM6x1M4jsprPMmYspTtRT20u8,2834
-memos/api/product_api.py,sha256=6JrQRIGj1UrFxcHL2myA5_4li6Tf5S5ynja5rX3XhYU,1129
-memos/api/product_models.py,sha256=aZJov3NqAaLpRLGPIg-GK6G00X-NvK2QyuOkQdCCmo8,6449
+memos/api/middleware/request_context.py,sha256=RNaeJt1AqmSfCB6lxkp8voyOjum7k6CrMV2w3cZgv10,2088
+memos/api/product_api.py,sha256=OOrrJ8h4BuJWXD8L8wb22E0Zfa73X0PwLC51wF0d5sc,1178
+memos/api/product_models.py,sha256=0yvI2q5Md_j8P4IK7NuqKMs9csm7I0-cFqufvPSrmRM,9292
 memos/api/routers/__init__.py,sha256=nbduRsr-7RkfBIYEBkbYkpB-7GG7QZgNfEQwLHbDXpI,21
-memos/api/routers/product_router.py,sha256=jeosTAJzwWkqo0Ev2eaHQex-LQtyI9LRLhLIvFDiOyw,17160
-memos/api/start_api.py,sha256=rVXMPEQy9r1PiVzHO3LJCf2qSsb0Mv0tklUJMQMBP8A,14776
+memos/api/routers/product_router.py,sha256=63FzAgIpSNijSyRspk5ftHwhGzTM5D19IbP23xLU9C0,17568
+memos/api/start_api.py,sha256=gR1vM7tYKiS1VNv9Jya9gvflhxeobFV4AQHTKCeW95Q,15185
 memos/chunkers/__init__.py,sha256=7lZOTN3e9Yp5XBsDX5wnWJ3tY126cRU9GmfevzJXAtU,67
 memos/chunkers/base.py,sha256=z0rG5vM7FGremQdSZ_3jlTGbsDtlkWAYWdtSAGqpaR4,655
 memos/chunkers/factory.py,sha256=ixpYz41GG3SZW-1ynLvfbhOZ0aGnFi2wUIYT4_Uxh30,693
@@ -20,26 +19,27 @@ memos/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/configs/base.py,sha256=bOOhEU6HGFTq5jne_TXn8Br72rvAfJHYvYllFUPs12A,2588
 memos/configs/chunker.py,sha256=Ulci0MyhS_FkuV95l7Pr9Vei179aEFEUigwVmtmSDfI,1593
 memos/configs/embedder.py,sha256=e_aOkUamW2NYSBEzsKkw0jNqHERy5GU3A2cSH5G5y68,2750
-memos/configs/graph_db.py,sha256=800e8PqiXTaOBX-GQHRP9f3Gx-0Ma7fnO0pdA5WHJ0k,5868
+memos/configs/graph_db.py,sha256=U_pNInRYW8iB-XG_5jLmPEcGKEmsGJmNIHPBN-m_Y80,5961
 memos/configs/internet_retriever.py,sha256=yWU6avRU9Hqox8hmR0EyILUr4x9wDvcKkhwRwT3PROc,3749
 memos/configs/llm.py,sha256=cso9ZBOdPzOAcgypp2nPFHdynswdG9MZZ-qdgOTEXL8,4175
 memos/configs/mem_chat.py,sha256=TjEQHRG1HpLwCBo3hrn5aVK23rykNtV6Be5p4YIg7F4,2571
 memos/configs/mem_cube.py,sha256=k95dvNxusnyS9w6ciWgNkUXwbHa4HTUEUS-PfPbR0BM,3105
 memos/configs/mem_os.py,sha256=FGrBaFWGUdCKkb4MFDRz9hCE7hydLd0XEFgUrqEhQJw,2755
 memos/configs/mem_reader.py,sha256=hTLFdmPhPIY70Fh_s4ptTGJ3LN-B9n111QG1kboxoNk,2307
-memos/configs/mem_scheduler.py,sha256=kLE1JZM2IkzpsEdvJ0D0w3MNbjsqfB_--EvdZomHB5s,7644
+memos/configs/mem_scheduler.py,sha256=NfukLGW-fm7912tCzvN5NeOFQiALT_kE7fuiVAlTDRY,9158
 memos/configs/mem_user.py,sha256=U0UzCvTacUOxeajJMfXAWWy3a0l52B4LUhH_L7U1VQ4,2036
 memos/configs/memory.py,sha256=wWrDvO67evM4Pn-utqneliIsJlap0ddNhdEH1Vn8HbA,7636
 memos/configs/parser.py,sha256=dy-QoevJbCnkJePKgpzR4oziOzYnS4jB6XH-YrpeMns,1145
 memos/configs/reranker.py,sha256=8IbKPPNUcGq9uGhgXcx41eYjDJr1lZKv89UKrA9v-Ck,459
 memos/configs/utils.py,sha256=X9NQ6-xURsZROAdS3WT96phVfHcOHgDPOo2Yq68QoKM,242
 memos/configs/vec_db.py,sha256=Gjhhda94pyTDjyJGe2Z6rVEqH4FtViiwq1-7QWhjarM,2335
+memos/context/context.py,sha256=3Sj0CW4H16MA2VvkGPrYfPVbRB0TmdUkEZfNrGBYyvc,7806
 memos/dependency.py,sha256=T5jhi9J6EQvx41dP8ma6hdCNn86d1amEP1iMASnrVbA,1842
 memos/deprecation.py,sha256=k2i4bBKqIwu20guwJmPPjOm7reOy19GzLwrTMrIqjPk,7408
 memos/embedders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/embedders/ark.py,sha256=i9dXR_lNXklxT8yUj_GiMIBDMR5tc40ltZMmX3CnJGk,3436
 memos/embedders/base.py,sha256=zOYERcaqBVCyMVndcR6vhFC_O2itEuTM058odDfhTx8,449
-memos/embedders/factory.py,sha256=tyWOvL2IZpWe-NMAvhBPsb4GqGcSUjoHJkZ65qaddEs,1027
+memos/embedders/factory.py,sha256=rZm4WP01_mqTbByv5g7Qc4OUCem4KP8YoEndmZv_9Lw,1110
 memos/embedders/ollama.py,sha256=8lHeMCqxej44ZgeDsPGL-5almxRBWsYYYflu5LeAf7w,2361
 memos/embedders/sentence_transformer.py,sha256=QJoW7qMbvAzmMUXzIhMHXVFnDFurfzxdoVUHIlslzec,1647
 memos/embedders/universal_api.py,sha256=BpmLL18y2neEAIE4CdC-1qr0uOEOJ6o7V_zycdkJmZk,1270
@@ -48,65 +48,76 @@ memos/graph_dbs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/graph_dbs/base.py,sha256=0NXdXII-UmX_bqUhBVpFdxRURLAK8sIFFZg51J9QzV0,8293
 memos/graph_dbs/factory.py,sha256=N1NaABiNnYvnsB-L82LrQaLvND5seCabFdSDsD8LoyA,932
 memos/graph_dbs/item.py,sha256=BfK17hA_hHu7I348jWugP_rb9GS5hpKfgqtYGGHBohk,1450
-memos/graph_dbs/nebular.py,sha256=AgoMapzTVSbuIz4-mTwj9aBhepS8PkcxJp9_s4uWitI,63674
-memos/graph_dbs/neo4j.py,sha256=sBlFCibQOtfO8c6bYQl_K9VsphJorfzx97UMQnrZs3E,42491
-memos/graph_dbs/neo4j_community.py,sha256=lWvfSo_-Dt3ST4vmBQiaziZDtQHV9CwY0qmWajOSPsc,11521
+memos/graph_dbs/nebular.py,sha256=hBDs3p2CLUp2vKOVX0VaVzHbFOh1Mfbngcl-Jlv8zS8,64462
+memos/graph_dbs/neo4j.py,sha256=HI8ZgHJ5dnOQgA4EBtrRo-olGstLZErgnfwn0OepIyg,43872
+memos/graph_dbs/neo4j_community.py,sha256=yvXhM7zvjvhdXHVoT_d2w6OaoK2Tqcl-rzAHuw0Wvgg,11851
 memos/hello_world.py,sha256=RV1vXfK1_U_xAvSusqc-4A8wk3yr8WEQ9q88dmBxvnI,3057
 memos/llms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/llms/base.py,sha256=CF60OpXdSLf-B5bHhQTnomMEkrNCg-SHauu1vo8HpeU,798
 memos/llms/deepseek.py,sha256=gXlInNa9Ts5-qw6nF4yD88vk5-UiW6-1C72DlE8Agok,1992
-memos/llms/factory.py,sha256=DVKMc9Pic36teO2fjIVu3qQulcXh8vYl2hPOJYY3vs4,1153
+memos/llms/factory.py,sha256=6mrUKyZ85bE8Bxq9gRPe_tuy_nCafAHuxAPCSUd9hKw,1236
 memos/llms/hf.py,sha256=zWz-7j2VnhOyytBbZnEJyNLezOFdV7vuMF6JRwbz9-g,15280
 memos/llms/hf_singleton.py,sha256=oxG9rSFZT_MhOWImttDVZfRZvxmpwJ8FarjL8NpgpRM,3502
 memos/llms/ollama.py,sha256=r1HMvSvG_m8gFfwLJhTriJdB6VzKEqz9f9JFs2RJGTw,2732
-memos/llms/openai.py,sha256=yOcdG320q0JFLi7ipHDXoPWpvyeOMFwbzrnXtk2xqec,3832
+memos/llms/openai.py,sha256=lrNFCzr-8kbPjny7Vaafvt7EYPr71Ify1GlHZUNMqQk,6500
 memos/llms/qwen.py,sha256=RFVoC_hxFe4dhCQ583fbTcT1AC3jDrKH0r6cDsZ0Cmo,2459
 memos/llms/utils.py,sha256=OcbM9iSpFJpio7sTT5wvxVx-JnqjIx7eSgiRk7dt0ZI,292
 memos/llms/vllm.py,sha256=9ZEgTUyRk86Wp2deQH0o1oW2GbHQ7C3evi-McELyT2Q,5792
-memos/log.py,sha256=bmOqq0lH-FkLOlNnYtppqdchkDTz7AjJuSuGUKhsxXQ,6362
+memos/log.py,sha256=Cpq6hHY8ETqOhC_l5xW-6n-qniCJ1TTrpl4yHyTkopQ,6761
 memos/mem_chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/mem_chat/base.py,sha256=LICB_mwUkdCVKb_r33GEPbrEr-2v3dEnI54cAvhcEew,830
 memos/mem_chat/factory.py,sha256=KKCDG9FrpfY2hD3iJ4GM9x8dN09dyhstP1cOUH_knrY,720
 memos/mem_chat/simple.py,sha256=-DRlE8nKZcGuwEubALK-nUf-wWRbGXkDJXUg8Dxhoyo,7919
 memos/mem_cube/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/mem_cube/base.py,sha256=NoZKz16oO5AAnJ0alcY52Z1c3C1rbGRe8Znz0CCzDnw,864
-memos/mem_cube/general.py,sha256=w1HbDIM7MD6Pu_Pbv4KHgQlxOMaCHvvJk_tmA9rZ_q8,8370
+memos/mem_cube/general.py,sha256=4n7LHe-KmeyZOdl0e4optSC9T7ut8IMcRJQEDC6sYV0,8490
 memos/mem_cube/utils.py,sha256=gDwiRo9VEvaTaRDzjWOGAH13ALRRV9bvUG7A-wy4KTA,5522
 memos/mem_os/client.py,sha256=0M-WRTlQr7fDAYtq4B8dsMR0PfmyvD-ySMhKcW3Umd0,43
-memos/mem_os/core.py,sha256=3PccrYDzheOKDW8_ZAwzXKvR8Rz4JHfvrdG4vSu6Ttg,45543
-memos/mem_os/main.py,sha256=zjgQy80QN70Xt6GHqn5zw-_2jOT6RWMCh9Yx-3zZuyw,24534
-memos/mem_os/product.py,sha256=_tloS0ncPx04qfwKbggNFQzk0VL5Nqx6uHQriAFZYzI,58877
+memos/mem_os/core.py,sha256=4qLdqBtL-1Ix5w5ulDtEK7onY6yX_-Kp4T3vF8IVTzM,47494
+memos/mem_os/main.py,sha256=0L4-jqagpd9-5eHIScsyCTjPlOzsPiEwf9ca9r4LcgA,24526
+memos/mem_os/product.py,sha256=msECenlI3z2On3jA4IhHNAwjFofQiYxWF4l_pR0OjL4,59996
 memos/mem_os/utils/default_config.py,sha256=OdDCyrE3-duSK0vMfO4AelGUwMt14ffHKMrvki0YKlI,12273
 memos/mem_os/utils/format_utils.py,sha256=QaxImCM0E27l2tkRLzWM1s8vSp3MO_LNaY0APHgV35o,49883
 memos/mem_os/utils/reference_utils.py,sha256=RtVwcuG3zFaxtMxY4S1_S8XWKQ7hO5UtXMswORBxyy0,6129
 memos/mem_reader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/mem_reader/base.py,sha256=SSuaD3J88XbHsME1Qa-EAgZ57xApPHPHppMSeeS3JZ4,957
-memos/mem_reader/factory.py,sha256=emKnId9BhScSkqCZyLSLWntixnU3wAMIOVdsRpHldJA,766
+memos/mem_reader/factory.py,sha256=FWsNCHgsHrLb3__DP-uIB9oIZiualVY1Q-tEdP-mzXY,849
 memos/mem_reader/memory.py,sha256=f3fAjrs8Jf6mBZWTgzkEZle7XjDwTJHxM1L2sOb85Tg,13288
-memos/mem_reader/simple_struct.py,sha256=TTYGOVt5WNwRpaHF0TW6VkVC98HbeS9-IA6dpIlHmpY,11864
+memos/mem_reader/simple_struct.py,sha256=5YJ701brNJr7MlIPMji4aJ5BEHUe-tfifWeErxB_WuM,14085
 memos/mem_scheduler/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/mem_scheduler/base_scheduler.py,sha256=wxBmdsJUmiufQLEfqflb4EUs9h9JE1EcMdsr_CA-LKE,23255
+memos/mem_scheduler/analyzer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+memos/mem_scheduler/analyzer/mos_for_test_scheduler.py,sha256=VDkoBDtFsSqVBbtizjcqCkQVc5EWYa-wZBp-VcPs7R8,24283
+memos/mem_scheduler/analyzer/scheduler_for_eval.py,sha256=ThiSlO9QcQW0TH3bj161zZC7OQm0RQxkYmx_b0-j0_I,9984
+memos/mem_scheduler/base_scheduler.py,sha256=-V5HsZX35qcLAOOwimpiML5QVYVvbk5AygtxEQjGiC4,26356
 memos/mem_scheduler/general_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/mem_scheduler/general_modules/base.py,sha256=j2FfGl5mx0CflHGnzu2fFFfZT_WRAMzmmTW_HUxFRgw,2946
-memos/mem_scheduler/general_modules/dispatcher.py,sha256=weDMAumIFJORJnpIm_8epz5d5pZ3XFx23O_0k-UONww,7375
-memos/mem_scheduler/general_modules/misc.py,sha256=qtHx_M3O0p01-bdje5xp4YbTYcU6rpoRsDvPikxOU-s,3574
-memos/mem_scheduler/general_modules/rabbitmq_service.py,sha256=kXfL8Q7eo8H2OMUDhAotbYG9cpL89HDJ5h7A64caKrw,12620
-memos/mem_scheduler/general_modules/redis_service.py,sha256=oHa9EvrpkknGAeygjmc8tExyCJVOU_7SZU6ScMOT9pY,5898
-memos/mem_scheduler/general_modules/retriever.py,sha256=6BiT-rxdXNnlDJjikQVwuy279W0tePo7tbb6j8c8j6s,7849
-memos/mem_scheduler/general_modules/scheduler_logger.py,sha256=8nzdjCd6I4-KRPwLjdf8eXcSQ0i_-xK6rAbQ3j1-GrA,9985
-memos/mem_scheduler/general_scheduler.py,sha256=oPGSvXKzFVYDr0YFnjS42zFYydHOXqlWd1woZ83CptA,14743
+memos/mem_scheduler/general_modules/dispatcher.py,sha256=be8UDzX977rgy6NPrMsG3IzHV9ehq38umb0E7NpcQ6s,7392
+memos/mem_scheduler/general_modules/misc.py,sha256=nQXGJJ6nZMl_ljmXbGIwuLt-9sLHdPkqBvR24ZBaDGs,7087
+memos/mem_scheduler/general_modules/scheduler_logger.py,sha256=F4ZUkc5OBZZF4UwaKybf42UEQxgQz709NybpsUVRjWA,10132
+memos/mem_scheduler/general_scheduler.py,sha256=0JYxo3uqmC9-69N2CHdbLQbXxm_qbLtWe_67BK3hHQ8,12780
+memos/mem_scheduler/memory_manage_modules/__init__.py,sha256=S7IUj-mFYxytsfb072ugQGEpFdnfSsPTcXowsH8Z_Cc,133
+memos/mem_scheduler/memory_manage_modules/memory_filter.py,sha256=K6wI_9ewvVuGY-PMtD6HzVO289ljXshTOTPvXeCd2L4,12440
+memos/mem_scheduler/memory_manage_modules/retriever.py,sha256=Zl5M44lD7pihIwQ0T0iJ5udWL_bNAo-Kvt6jhtdiB1M,8929
 memos/mem_scheduler/monitors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/mem_scheduler/monitors/dispatcher_monitor.py,sha256=xy3fFa_vM5PpVHWweMTh3V3bZmnccWV49x0nBNFAVVw,10930
-memos/mem_scheduler/monitors/general_monitor.py,sha256=pv_T-olU0XTO44-rXgfqX43nmDEp8iq-rwlMyAA0Lxk,12962
-memos/mem_scheduler/mos_for_test_scheduler.py,sha256=kioG5V27V8jfXxFA9OHpKoULNF697eGToVo7z-U3mqQ,5950
-memos/mem_scheduler/scheduler_factory.py,sha256=pw6FKO0EuxFRipMnD22DvcjbM6pdmVfJmRjUvdTaU5E,800
+memos/mem_scheduler/monitors/dispatcher_monitor.py,sha256=gAPuWoeWVm3338qamo8v7nfT0UVpFkpXb35ZshrTKbo,11177
+memos/mem_scheduler/monitors/general_monitor.py,sha256=6JWXGTuv-udyIe_K0l_rMY1ITL8OgiXxnABGim9YFJ0,16718
+memos/mem_scheduler/optimized_scheduler.py,sha256=J6L366nZBe8a9j6_nwvysUSPTr5mGVT1xi6mo7zHN9E,4997
+memos/mem_scheduler/orm_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+memos/mem_scheduler/orm_modules/base_model.py,sha256=oyWIL9ISx5qh759h8OPETOk1NKAmpjMX1ObG-uC1J7k,23880
+memos/mem_scheduler/orm_modules/monitor_models.py,sha256=XUQdznlD29NQlRM-q1MSajaJNmBGL9AVa7o1VHuBWLE,9457
+memos/mem_scheduler/scheduler_factory.py,sha256=fUEt9wlTwyhN0jeZO8fVdkbA0VZnV6zsyycw3acKCPI,922
 memos/mem_scheduler/schemas/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/mem_scheduler/schemas/general_schemas.py,sha256=r9Ew8KHTVxiDTTLKad1jw12uNMyLltFmn2htplSqbhk,1344
 memos/mem_scheduler/schemas/message_schemas.py,sha256=ahTEY6x8gSXszURne1nKgHCkfEXfKOEXaTKTUsQ9IW0,5977
-memos/mem_scheduler/schemas/monitor_schemas.py,sha256=OvFwchhFCvkHnp8k7NYAQQ5Lq7N0cNKA0mSEasNRLkU,12715
+memos/mem_scheduler/schemas/monitor_schemas.py,sha256=wprD-O2LXW4nNVoTT-wR0bx0mH_VP0zW5RBck8NZm7I,15636
 memos/mem_scheduler/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/mem_scheduler/utils/filter_utils.py,sha256=GsEsVjB6zZbF2gITxXN1DWTLoLG-d8A3TFnGzy1uIaQ,6286
+memos/mem_scheduler/utils/config_utils.py,sha256=x0Mp0wuoIQtDZkaXXJUTtesLWFq8eyh-hjkLEm4cOTM,3819
+memos/mem_scheduler/utils/db_utils.py,sha256=Fba2OIDlQsYwydUyHlcZh7VFsUeeIkSpO0ogePOofps,1042
+memos/mem_scheduler/utils/filter_utils.py,sha256=RoWrCf-qk2BJdV2-5ftWYM_UEQqYF6QZWtjPzfHskSc,6299
 memos/mem_scheduler/utils/misc_utils.py,sha256=zOCpwzCrj0mfiW7BoWiTFEPj-dzdcYZnqMDtKDGO0kc,2765
+memos/mem_scheduler/webservice_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+memos/mem_scheduler/webservice_modules/rabbitmq_service.py,sha256=kXfL8Q7eo8H2OMUDhAotbYG9cpL89HDJ5h7A64caKrw,12620
+memos/mem_scheduler/webservice_modules/redis_service.py,sha256=oHa9EvrpkknGAeygjmc8tExyCJVOU_7SZU6ScMOT9pY,5898
 memos/mem_user/factory.py,sha256=zdC-208YYU4dpUaMpICzNbhMTiLaJDxLJoq4bl-6YiU,2956
 memos/mem_user/mysql_persistent_user_manager.py,sha256=fSndxW6MR7u9ImWCPq-gsuFb2TYX1BXF6Kkv1FCS6Fw,9257
 memos/mem_user/mysql_user_manager.py,sha256=Q_3FxNTw8Vf6uPotHk4YUooa8EVo0jBFTFBKJHPm_6o,16331
@@ -117,7 +128,7 @@ memos/memories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/activation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/activation/base.py,sha256=kiC35AOnuofgYMzLxZyVULsfOoVq03BGvi6AXtNvF-I,1151
 memos/memories/activation/item.py,sha256=8oo3odhNnvg36NYLobwkehAHj74mo36IcH2MrlXSCEI,1720
-memos/memories/activation/kv.py,sha256=c3rhOowxqmYnUnVKGT1XqsHQurl44Ht98wCDz42laP8,8903
+memos/memories/activation/kv.py,sha256=A1_dlJYrdcsLz_kJRD3dq614mw1FC4uiA_c6WVF54VU,8904
 memos/memories/activation/vllmkv.py,sha256=QM6Y8a3dhqGa08_5rbE-JTR4EakNgoEEVfLdVcvmgms,7765
 memos/memories/base.py,sha256=Sr-dEuDc982WwdVREQ2nL8L6rUc0KZPTaBJeYdgx8h8,577
 memos/memories/factory.py,sha256=Z49DRF_VHQJI2UNiTJ_1ouPqg_iCcuV_Tcg5fXkKxlM,1418
@@ -128,57 +139,60 @@ memos/memories/parametric/lora.py,sha256=TqSI2OjmFi-XXCeM-MchSwh1sAhOwL7_JnOwSy9
 memos/memories/textual/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/textual/base.py,sha256=5AQ0oR1rLd7yiwIKPxt2OmAlL6mWhwUl_U8zuafoDfI,2827
 memos/memories/textual/general.py,sha256=exHPHejBE6jZNLm8pT951sIko8hAqNdOaHZdnNdfzBA,8877
-memos/memories/textual/item.py,sha256=GR6V0GNn32W-pYtwSvmz5F6kTh39Yp4XrQm-cEghL2I,5259
-memos/memories/textual/naive.py,sha256=Z_gfbxI6cQGJ_raOTQic4fnpo493Xq3yEQ8qDV4xplo,6954
-memos/memories/textual/tree.py,sha256=7RwwWYm5c_F0pNMnQDjcCV01NGb_t-yN_kU8EMZjB9g,14826
+memos/memories/textual/item.py,sha256=uzThiCAKGkIyG0hHL9ywWQNVEq6tr-yMCakoZBW7Dz8,8280
+memos/memories/textual/naive.py,sha256=BjsDp3CdxcQzvKkGr1e_t4qdL_OFemwLjDf37yLRO-4,6964
+memos/memories/textual/tree.py,sha256=Ch-oS_UDXhkYmH8q7pd2QKCF3T2tCyMQOUub-Y3nCu0,16189
 memos/memories/textual/tree_text_memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/memories/textual/tree_text_memory/organize/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/memories/textual/tree_text_memory/organize/handler.py,sha256=l7WuwWvubGzgaOSUBS0VNjSQU2JxtjwLMtKHxpms8NQ,8139
-memos/memories/textual/tree_text_memory/organize/manager.py,sha256=r-0O9hS3y3oQrTaKOulLmbsquk_FW6eEPse6tf7AuVM,9783
-memos/memories/textual/tree_text_memory/organize/relation_reason_detector.py,sha256=udqQEFZb6TiarB-Rya1N7fKiWHtXiLkggh8bFKIlDjU,8816
-memos/memories/textual/tree_text_memory/organize/reorganizer.py,sha256=txBC6uo3fId4RaTJeTJEg4V3Q-jrIP0jBHqVQkjjAqc,22719
+memos/memories/textual/tree_text_memory/organize/handler.py,sha256=zWkmm0ZdQjmuDUbmQkslGvjFrB-CWuq4y2h7uOTeUro,8135
+memos/memories/textual/tree_text_memory/organize/manager.py,sha256=pkFGEi50xs8JA6M-MwB3EjC9vfJwUZ3683cM2L6x-XA,10319
+memos/memories/textual/tree_text_memory/organize/relation_reason_detector.py,sha256=h2vGAp_0l7xd60CIwc0bz5-FNAN9Sju7e_e7vzcj25s,8815
+memos/memories/textual/tree_text_memory/organize/reorganizer.py,sha256=ONktG6-0DPg-PbAudnLfhh0ccah9cdtGr-cJyWxucuI,24581
 memos/memories/textual/tree_text_memory/retrieve/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-memos/memories/textual/tree_text_memory/retrieve/bochasearch.py,sha256=Nh9nbOD3MPcofD4jM0Mihb7cjE17VTvPEsuuMTzisM4,8258
-memos/memories/textual/tree_text_memory/retrieve/internet_retriever.py,sha256=aD2bujlVBWzDa9ZfB9vdB9Hq2erCpc426AVcolItWIM,8656
-memos/memories/textual/tree_text_memory/retrieve/internet_retriever_factory.py,sha256=3-hJwpGrAAag7z772QBvW9ZBvNI_od5GO1nRv5Q3GbQ,3767
+memos/memories/textual/tree_text_memory/retrieve/bochasearch.py,sha256=7IGL3QHLI74ZJmAIPLFrwxqSXkqubHAKiR7q4po5Tpw,8384
+memos/memories/textual/tree_text_memory/retrieve/internet_retriever.py,sha256=RRViVkwQAlXoqH9_0azIfGr2g2DsAybglXlxJqV27Ns,8719
+memos/memories/textual/tree_text_memory/retrieve/internet_retriever_factory.py,sha256=ob1E-iXjCSSbb6KviZWXAO7LiMt5h9C7xi4bYjC_LiY,3850
 memos/memories/textual/tree_text_memory/retrieve/reasoner.py,sha256=5csoGjviFbN9RJ8dm3B6kjvoC8xbPD8UMiGusefHaf0,2228
-memos/memories/textual/tree_text_memory/retrieve/recall.py,sha256=oGTp-14af1ZCHd8zkKP4vZ9F6gb3F9hGPdGp70AtiVA,7841
+memos/memories/textual/tree_text_memory/retrieve/recall.py,sha256=DnK7aak5eeZpMf6pIwFpkW5IZ92FLtVQEnhf8Qan1Cw,9504
 memos/memories/textual/tree_text_memory/retrieve/reranker.py,sha256=GIo_ZqA8Oea-L58Md724ak7Oq2fCnLfBWBUvMWiD7mY,3727
 memos/memories/textual/tree_text_memory/retrieve/retrieval_mid_structs.py,sha256=CvGfdwtOwC78OWKb1nOg_tsPualirsRfbtA58_kxNpA,433
-memos/memories/textual/tree_text_memory/retrieve/searcher.py,sha256=18o-qlcLI0Fv6H-wRYcPf3ez6kcVjUt37TIbzEl7naM,12854
+memos/memories/textual/tree_text_memory/retrieve/searcher.py,sha256=gbfjsA6skRslhreDsFuARRTjyI0aZmhF7GpncnOJoBg,14772
 memos/memories/textual/tree_text_memory/retrieve/task_goal_parser.py,sha256=vqgyczpS5As9zmljX9YF7TL0fgpl_T22FN5qV_AbcS8,3733
 memos/memories/textual/tree_text_memory/retrieve/utils.py,sha256=ISdSJEq_HqtPEqmTRQO6dXx7ioDWrbLi7mrrep4iQDw,2191
-memos/memories/textual/tree_text_memory/retrieve/xinyusearch.py,sha256=x-nlZJ84shgAq9FqQR3YvE6iTMY_B7Pc0YaiGEqM1rc,10518
+memos/memories/textual/tree_text_memory/retrieve/xinyusearch.py,sha256=2NJTtVeLkqbNwCjwWkPw8haLWeUNpSifGa-BSiPpz9g,10611
 memos/memos_tools/dinding_report_bot.py,sha256=kXmooO_wUpwErT6Obql-SQu8lbt0XnM_58znIeQt734,14735
 memos/memos_tools/lockfree_dict.py,sha256=A2hFUSyjTIauiXONYzr7zFZfGlOF9J9cqrfzDNBaw_w,3811
 memos/memos_tools/notification_service.py,sha256=CZ7tZI6m0A063ON0MVEOUb9uontggvrvBLOMzorD0cI,1125
 memos/memos_tools/notification_utils.py,sha256=uHrDOXEW-WUJcoeR7gVhr2BVjNYUxIjz8cCve5F1PPs,3747
-memos/memos_tools/thread_safe_dict.py,sha256=5hhKndKntYIriTIkmaDaehmojuOMhEkVl4WAc2QLfHM,8252
+memos/memos_tools/singleton.py,sha256=k7D0cO65gnhWmYeXdQwQ6baBCSWtjH_T0niGE2ORWUw,6288
+memos/memos_tools/thread_safe_dict.py,sha256=bM0UzyO2-1V0Uq94k828vGWTHUSj7oS4cA6K-DCLPus,8534
+memos/memos_tools/thread_safe_dict_segment.py,sha256=zS9Nu06EThlttTrleZYkNN-bR0a3df-6imYWjPZSXTo,12478
 memos/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/parsers/base.py,sha256=AWgWIZzReDiTqiv6z08_9aG2KHVzc4Bdr0Lowz0mWVI,435
-memos/parsers/factory.py,sha256=hPKTR0wVgMQ5Z9ZL-a9FyHWCpz9UcrG2oSEDYBAoB3g,704
+memos/parsers/factory.py,sha256=yRz-KIPRxBld2KOdHxh92SHQwHIevPzN9nLxqEz_pKY,787
 memos/parsers/markitdown.py,sha256=SDwmZYC6LyiwdU79yuj6mWNMkQx9YrZzZEHU-4JzBwQ,863
 memos/reranker/__init__.py,sha256=bq3LWhVg6LdK50Hx6jOM-eWEaNM0mX4_JuHNfYgUqE0,69
 memos/reranker/base.py,sha256=bl9W-CqtlAs2k7loPvd3TRqq6Illybt6IosqrYw0DHw,573
-memos/reranker/cosine_local.py,sha256=tOqL6WIUxkFriaROV0hKtJQ-b79BGGYD4Ow1VE0x0g0,2970
-memos/reranker/factory.py,sha256=RMBKOLSgmhfvbhCqU8CnQrXR7176zjq04VD6PMDrxuk,1321
-memos/reranker/http_bge.py,sha256=BFlRMqspV856EGinjkI_Vr2oo6YBeF00uDy_eeTT-k0,3254
+memos/reranker/concat.py,sha256=7tZ-3VhkvMkPmUwqyzU4h_0EQ-PMQfdDwNl4B_VHlSA,1768
+memos/reranker/cosine_local.py,sha256=vfDP9FkC0Cw6hvFZOeGlKz3q7J0fpI0Uh9fZFsRFGyY,2988
+memos/reranker/factory.py,sha256=TvumTHbfjLsJs96S0rWHvCsmQJ4naGBMYWLvfJ61QzQ,1505
+memos/reranker/http_bge.py,sha256=fl8HVGqBtPDCw-h84KgoJ0zFJp1vXZKvmS6oiBKnuSk,11357
 memos/reranker/noop.py,sha256=OOy3d_AHydCm5BH4VNHEfq_vDE-D-WBgwSKIp8GQloo,414
 memos/settings.py,sha256=BYOaqhzReu2yjbwrZqx9-uMwoipkj5Clc6VthnBYOvY,225
 memos/templates/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/templates/mem_reader_prompts.py,sha256=B1sWIuZbIQM1zTv-VAWDsHUNLdO1O6fCy9WdhLMq2Aw,26875
-memos/templates/mem_scheduler_prompts.py,sha256=ty_H9TARf5sQogHSJoRYQmrS6w0xs51mMcKNjkExdFc,5034
+memos/templates/mem_scheduler_prompts.py,sha256=GuLTsBSVvilYIU_dZZj7uWFnSSTDhlDxMjNIumRa4wY,14665
 memos/templates/mos_prompts.py,sha256=yqpf8n0ObJPrJjXg4x-Nxsd0_pr0cDhdyCOopI7sgbI,18321
 memos/templates/tree_reorganize_prompts.py,sha256=FJRIwT2A4d5EnoPhNuzDFdMWpbNLlJEIfLZP69RBlDM,12112
-memos/types.py,sha256=N7XBYxDTdc50KEsS6YxHvYgs23ykGsZ-wNnaJBVdVi4,1791
+memos/types.py,sha256=GyeD3nBInBe0TVXjme-mMzhK_TGcWnhnH3HfmVbUM0c,2018
 memos/utils.py,sha256=uB-8_5V16M7yNeAYXOIPaXwxeceUu4d_5yofLA6vbaE,428
 memos/vec_dbs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 memos/vec_dbs/base.py,sha256=gbw7Gb_zs8Yx8kJIx6aTM-Ylsb4qLkVOFB-iwNJW6Sw,3564
 memos/vec_dbs/factory.py,sha256=Noa4caqzPT9b59i2jzdpAHFCSHiMfDmgRox1POkRRfE,710
 memos/vec_dbs/item.py,sha256=mLrcHF0nWtMCUjScBgaeeSqabQ3vJhKr_6wrU_g25ns,1425
 memos/vec_dbs/qdrant.py,sha256=4KAiSZ64Ulz8EeT2o1DpNDRwnKRF4iiaJbZdq-xrn0c,11052
-memoryos-1.0.1.dist-info/LICENSE,sha256=FU-b6N8tVc7dzUZGyNjUIG1Ihnrh2iuBziq4a1Gl8HU,11358
-memoryos-1.0.1.dist-info/METADATA,sha256=PvCovFOrn2QX_LoH22DuSjJQZwgl9rxgjVJvCfQMfxo,16011
-memoryos-1.0.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-memoryos-1.0.1.dist-info/entry_points.txt,sha256=p54si8po81Yb-NK2lx5Z0lNr8QV7-5IECtkC6c8MJmI,40
-memoryos-1.0.1.dist-info/RECORD,,
+memoryos-1.1.2.dist-info/METADATA,sha256=g3VA6ZCN6R7jJx55hiQR1gqps_SHSOMBDvJE2lUb7vo,16217
+memoryos-1.1.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+memoryos-1.1.2.dist-info/entry_points.txt,sha256=p54si8po81Yb-NK2lx5Z0lNr8QV7-5IECtkC6c8MJmI,40
+memoryos-1.1.2.dist-info/licenses/LICENSE,sha256=FU-b6N8tVc7dzUZGyNjUIG1Ihnrh2iuBziq4a1Gl8HU,11358
+memoryos-1.1.2.dist-info/RECORD,,

{memoryos-1.0.1.dist-info → memoryos-1.1.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any

memos/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.0.1"
+__version__ = "1.1.2"
 
 from memos.configs.mem_cube import GeneralMemCubeConfig
 from memos.configs.mem_os import MOSConfig

memos/api/client.py

@@ -0,0 +1,109 @@
+import json
+import os
+
+from typing import Any
+
+import requests
+
+from memos.api.product_models import MemOSAddResponse, MemOSGetMessagesResponse, MemOSSearchResponse
+from memos.log import get_logger
+
+
+logger = get_logger(__name__)
+
+MAX_RETRY_COUNT = 3
+
+
+class MemOSClient:
+    """MemOS API client"""
+
+    def __init__(self, api_key: str | None = None, base_url: str | None = None):
+        self.base_url = (
+            base_url or os.getenv("MEMOS_BASE_URL") or "https://memos.memtensor.cn/api/openmem/v1"
+        )
+        api_key = api_key or os.getenv("MEMOS_API_KEY")
+
+        if not api_key:
+            raise ValueError("MemOS API key is required")
+
+        self.headers = {"Content-Type": "application/json", "Authorization": f"Token {api_key}"}
+
+    def _validate_required_params(self, **params):
+        """Validate required parameters - if passed, they must not be empty"""
+        for param_name, param_value in params.items():
+            if not param_value:
+                raise ValueError(f"{param_name} is required")
+
+    def get_message(
+        self, user_id: str, conversation_id: str | None = None
+    ) -> MemOSGetMessagesResponse:
+        """Get messages"""
+        # Validate required parameters
+        self._validate_required_params(user_id=user_id)
+
+        url = f"{self.base_url}/get/message"
+        payload = {"user_id": user_id, "conversation_id": conversation_id}
+        for retry in range(MAX_RETRY_COUNT):
+            try:
+                response = requests.post(
+                    url, data=json.dumps(payload), headers=self.headers, timeout=30
+                )
+                response.raise_for_status()
+                response_data = response.json()
+                return MemOSGetMessagesResponse(**response_data)
+            except Exception as e:
+                logger.error(f"Failed to get messages (retry {retry + 1}/3): {e}")
+                if retry == MAX_RETRY_COUNT - 1:
+                    raise
+
+    def add_message(
+        self, messages: list[dict[str, Any]], user_id: str, conversation_id: str
+    ) -> MemOSAddResponse:
+        """Add memories"""
+        # Validate required parameters
+        self._validate_required_params(
+            messages=messages, user_id=user_id, conversation_id=conversation_id
+        )
+
+        url = f"{self.base_url}/add/message"
+        payload = {"messages": messages, "user_id": user_id, "conversation_id": conversation_id}
+        for retry in range(MAX_RETRY_COUNT):
+            try:
+                response = requests.post(
+                    url, data=json.dumps(payload), headers=self.headers, timeout=30
+                )
+                response.raise_for_status()
+                response_data = response.json()
+                return MemOSAddResponse(**response_data)
+            except Exception as e:
+                logger.error(f"Failed to add memory (retry {retry + 1}/3): {e}")
+                if retry == MAX_RETRY_COUNT - 1:
+                    raise
+
+    def search_memory(
+        self, query: str, user_id: str, conversation_id: str, memory_limit_number: int = 6
+    ) -> MemOSSearchResponse:
+        """Search memories"""
+        # Validate required parameters
+        self._validate_required_params(query=query, user_id=user_id)
+
+        url = f"{self.base_url}/search/memory"
+        payload = {
+            "query": query,
+            "user_id": user_id,
+            "conversation_id": conversation_id,
+            "memory_limit_number": memory_limit_number,
+        }
+
+        for retry in range(MAX_RETRY_COUNT):
+            try:
+                response = requests.post(
+                    url, data=json.dumps(payload), headers=self.headers, timeout=30
+                )
+                response.raise_for_status()
+                response_data = response.json()
+                return MemOSSearchResponse(**response_data)
+            except Exception as e:
+                logger.error(f"Failed to search memory (retry {retry + 1}/3): {e}")
+                if retry == MAX_RETRY_COUNT - 1:
+                    raise

memos/api/config.py

@@ -21,7 +21,7 @@ class APIConfig:
     def get_openai_config() -> dict[str, Any]:
         """Get OpenAI configuration."""
         return {
-            "model_name_or_path": os.getenv("MOS_OPENAI_MODEL", "gpt-4o-mini"),
+            "model_name_or_path": os.getenv("MOS_CHAT_MODEL", "gpt-4o-mini"),
             "temperature": float(os.getenv("MOS_CHAT_TEMPERATURE", "0.8")),
             "max_tokens": int(os.getenv("MOS_MAX_TOKENS", "1024")),
             "top_p": float(os.getenv("MOS_TOP_P", "0.9")),
@@ -100,8 +100,10 @@ class APIConfig:
                 "backend": "http_bge",
                 "config": {
                     "url": os.getenv("MOS_RERANKER_URL"),
-                    "model": "bge-reranker-v2-m3",
+                    "model": os.getenv("MOS_RERANKER_MODEL", "bge-reranker-v2-m3"),
                     "timeout": 10,
+                    "headers_extra": os.getenv("MOS_RERANKER_HEADERS_EXTRA"),
+                    "rerank_source": os.getenv("MOS_RERANK_SOURCE"),
                 },
             }
         else:
@@ -186,22 +188,22 @@ class APIConfig:
         return {
             "uri": os.getenv("NEO4J_URI", "bolt://localhost:7687"),
             "user": os.getenv("NEO4J_USER", "neo4j"),
-            "db_name": os.getenv("NEO4J_DB_NAME", "shared-tree-textual-memory"),
+            "db_name": os.getenv("NEO4J_DB_NAME", "neo4j"),
             "password": os.getenv("NEO4J_PASSWORD", "12345678"),
             "user_name": f"memos{user_id.replace('-', '')}",
-            "auto_create": True,
+            "auto_create": False,
             "use_multi_db": False,
-            "embedding_dimension": int(os.getenv("EMBEDDING_DIMENSION", 3072)),
+            "embedding_dimension": int(os.getenv("EMBEDDING_DIMENSION", 1024)),
             "vec_config": {
                 # Pass nested config to initialize external vector DB
                 # If you use qdrant, please use Server instead of local mode.
                 "backend": "qdrant",
                 "config": {
                     "collection_name": "neo4j_vec_db",
-                    "vector_dimension": int(os.getenv("EMBEDDING_DIMENSION", 3072)),
+                    "vector_dimension": int(os.getenv("EMBEDDING_DIMENSION", 1024)),
                     "distance_metric": "cosine",
-                    "host": "localhost",
-                    "port": 6333,
+                    "host": os.getenv("QDRANT_HOST", "localhost"),
+                    "port": int(os.getenv("QDRANT_PORT", "6333")),
                 },
             },
         }
@@ -271,7 +273,7 @@ class APIConfig:
     def get_scheduler_config() -> dict[str, Any]:
         """Get scheduler configuration."""
         return {
-            "backend": "general_scheduler",
+            "backend": "optimized_scheduler",
             "config": {
                 "top_k": int(os.getenv("MOS_SCHEDULER_TOP_K", "10")),
                 "act_mem_update_interval": int(

memos/api/context/dependencies.py

@@ -1,8 +1,6 @@
 import logging
 
-from fastapi import Depends, Header, Request
-
-from memos.api.context.context import RequestContext, set_request_context
+from memos.context.context import RequestContext, get_current_context
 
 
 logger = logging.getLogger(__name__)
@@ -11,56 +9,17 @@ logger = logging.getLogger(__name__)
 G = RequestContext
 
 
-def get_trace_id_from_header(
-    trace_id: str | None = Header(None, alias="trace-id"),
-    x_trace_id: str | None = Header(None, alias="x-trace-id"),
-    g_trace_id: str | None = Header(None, alias="g-trace-id"),
-) -> str | None:
-    """
-    Extract trace_id from various possible headers.
-
-    Priority: g-trace-id > x-trace-id > trace-id
-    """
-    return g_trace_id or x_trace_id or trace_id
-
-
-def get_request_context(
-    request: Request, trace_id: str | None = Depends(get_trace_id_from_header)
-) -> RequestContext:
-    """
-    Get request context object with trace_id and request metadata.
-
-    This function creates a RequestContext and automatically sets it
-    in the global context for use throughout the request lifecycle.
-    """
-    # Create context object
-    ctx = RequestContext(trace_id=trace_id)
-
-    # Set the context globally for this request
-    set_request_context(ctx)
-
-    # Log request start
-    logger.info(f"Request started with trace_id: {ctx.trace_id}")
-
-    # Add request metadata to context
-    ctx.set("method", request.method)
-    ctx.set("path", request.url.path)
-    ctx.set("client_ip", request.client.host if request.client else None)
-
-    return ctx
-
-
-def get_g_object(trace_id: str | None = Depends(get_trace_id_from_header)) -> G:
+def get_g_object() -> G:
     """
     Get Flask g-like object for the current request.
-
-    This creates a RequestContext and sets it globally for access
-    throughout the request lifecycle.
+    Returns the context created by middleware.
     """
-    g = RequestContext(trace_id=trace_id)
-    set_request_context(g)
-    logger.info(f"Request g object created with trace_id: {g.trace_id}")
-    return g
+    ctx = get_current_context()
+    if ctx is None:
+        raise RuntimeError(
+            "No request context available. Make sure RequestContextMiddleware is properly configured."
+        )
+    return ctx
 
 
 def get_current_g() -> G | None:
@@ -70,8 +29,6 @@ def get_current_g() -> G | None:
     Returns:
         The current request's g object if available, None otherwise.
     """
-    from memos.context import get_current_context
-
     return get_current_context()
 
 
@@ -85,6 +42,9 @@ def require_g() -> G:
     Raises:
         RuntimeError: If called outside of a request context.
     """
-    from memos.context import require_context
-
-    return require_context()
+    ctx = get_current_context()
+    if ctx is None:
+        raise RuntimeError(
+            "No request context available. This function must be called within a request handler."
+        )
+    return ctx