import os
from mem0 import AsyncMemory
from mem0.configs.base import MemoryConfig

os.environ['OPENAI_API_KEY'] = 'YOUR_KEY'

MEM_0_CONFIG = {
    "vector_store": {
        "provider": "chroma",
        "config": {
            "collection_name": "memories",
            "path": "test_db_2",
        }
    },
}

async def test(query, user_id):
    memory = AsyncMemory(config=MemoryConfig(**MEM_0_CONFIG))
    result = await memory.add([{"role": "user", "content": " something"}], user_id=user_id)

if __name__ == "__main__":
    import asyncio
    query = " "
    user_id = "alice"
    asyncio.run(test(query, user_id))