fix(vector-stores): add agent_id and run_id to Elasticsearch/OpenSearch default mappings (#4906)

Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>

mem0/vector_stores/elasticsearch.py

@@ -66,7 +66,14 @@ class ElasticsearchDB(VectorStoreBase):
                         "index": True,
                         "similarity": "cosine",
                     },
-                    "metadata": {"type": "object", "properties": {"user_id": {"type": "keyword"}}},
+                    "metadata": {
+                        "type": "object",
+                        "properties": {
+                            "user_id": {"type": "keyword"},
+                            "agent_id": {"type": "keyword"},
+                            "run_id": {"type": "keyword"},
+                        },
+                    },
                 }
             },
         }

mem0/vector_stores/opensearch.py

@@ -55,7 +55,14 @@ class OpenSearchDB(VectorStoreBase):
                         "dimension": self.embedding_model_dims,
                         "method": {"engine": "nmslib", "name": "hnsw", "space_type": "cosinesimil"},
                     },
-                    "metadata": {"type": "object", "properties": {"user_id": {"type": "keyword"}}},
+                    "metadata": {
+                        "type": "object",
+                        "properties": {
+                            "user_id": {"type": "keyword"},
+                            "agent_id": {"type": "keyword"},
+                            "run_id": {"type": "keyword"},
+                        },
+                    },
                 }
             },
         }

tests/vector_stores/test_elasticsearch.py

@@ -96,6 +96,10 @@ class TestElasticsearchDB(unittest.TestCase):
         self.assertEqual(mappings["vector"]["index"], True)
         self.assertEqual(mappings["vector"]["similarity"], "cosine")
         self.assertEqual(mappings["metadata"]["type"], "object")
+        metadata_props = mappings["metadata"]["properties"]
+        self.assertEqual(metadata_props["user_id"]["type"], "keyword")
+        self.assertEqual(metadata_props["agent_id"]["type"], "keyword")
+        self.assertEqual(metadata_props["run_id"]["type"], "keyword")
 
         # Reset mocks for next test
         self.client_mock.reset_mock()

tests/vector_stores/test_opensearch.py

@@ -142,6 +142,10 @@ class TestOpenSearchDB(unittest.TestCase):
         mappings = create_args["body"]["mappings"]["properties"]
         self.assertEqual(mappings["vector_field"]["type"], "knn_vector")
         self.assertEqual(mappings["vector_field"]["dimension"], 1536)
+        metadata_props = mappings["metadata"]["properties"]
+        self.assertEqual(metadata_props["user_id"]["type"], "keyword")
+        self.assertEqual(metadata_props["agent_id"]["type"], "keyword")
+        self.assertEqual(metadata_props["run_id"]["type"], "keyword")
         self.client_mock.reset_mock()
         self.client_mock.indices.exists.return_value = True
         self.os_db.create_index()