search

mem0ai / mem0

The Memory layer for your AI apps
https://mem0.ai
Apache License 2.0
21.77k stars 1.98k forks source link

memory/memory_graph.py is always set to use OpenAI #1759

Closed SwastikGorai closed 2 weeks ago

SwastikGorai commented 3 weeks ago

🐛 Describe the bug

In mem0/memory/main_graph.py line 11:

client = OpenAI()

It is always set to use OpenAI models(also, hardcoded : self.model_name = "gpt-4o-2024-08-06", line 48).

It would be nice if mem0 also added other models from mem0.llms like groq.

prateekchhikara commented 2 weeks ago

Hi @SwastikGorai thanks for highlighting this issue. We will pick this task and add custom LLM support for graph memory.

prateekchhikara commented 2 weeks ago

Hi @SwastikGorai we have integrated the support for other LLMs as well. Checkout our updated documentation: https://docs.mem0.ai/open-source/graph_memory/overview

prateekchhikara commented 2 weeks ago

@SwastikGorai, I am closing this issue as it is fixed.