Skip to content

Commit 45fa380

Browse files
authored
Update memory provider to use Gemini 2.5 Flash (#10)
Gemini 2.0 models are deprecated. See https://ai.google.dev/gemini-api/docs/models#previous_models
1 parent 84c3f96 commit 45fa380

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/memory_bench/memory/mem0.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414

1515
class Mem0MemoryProvider(MemoryProvider):
1616
name = "mem0"
17-
description = "Agentic memory with Gemini 2.0 Flash for reflective extraction + local Qdrant store."
17+
description = "Agentic memory with Gemini 2.5 Flash for reflective extraction + local Qdrant store."
1818
kind = "local"
1919
provider = "mem0"
2020
variant = "local"
@@ -36,7 +36,7 @@ def _build_memory(self, qdrant_path: str) -> Memory:
3636
{
3737
"llm": {
3838
"provider": "gemini",
39-
"config": {"model": "gemini-2.0-flash", "temperature": 0.0},
39+
"config": {"model": "gemini-2.5-flash", "temperature": 0.0},
4040
},
4141
"embedder": {
4242
"provider": "huggingface",

0 commit comments

Comments
 (0)