klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Docs update (#1875)

Browse Source
This commit is contained in:
Prateek Chhikara
2024-09-17 10:53:14 -07:00
committed by GitHub
parent 55c54beeab
commit 8c3c9e1520
5 changed files with 33 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
mem0/memory/graph_memory.py
View File

@@ -3,14 +3,19 @@ import logging
from langchain_community.graphs import Neo4jGraph
from rank_bm25 import BM25Okapi
from mem0.graphs.tools import (ADD_MEMORY_STRUCT_TOOL_GRAPH,
ADD_MEMORY_TOOL_GRAPH, ADD_MESSAGE_STRUCT_TOOL,
ADD_MESSAGE_TOOL, NOOP_STRUCT_TOOL, NOOP_TOOL,
SEARCH_STRUCT_TOOL, SEARCH_TOOL,
UPDATE_MEMORY_STRUCT_TOOL_GRAPH,
UPDATE_MEMORY_TOOL_GRAPH)
from mem0.graphs.utils import (EXTRACT_ENTITIES_PROMPT,
get_update_memory_messages)
from mem0.graphs.tools import (
ADD_MEMORY_STRUCT_TOOL_GRAPH,
ADD_MEMORY_TOOL_GRAPH,
ADD_MESSAGE_STRUCT_TOOL,
ADD_MESSAGE_TOOL,
NOOP_STRUCT_TOOL,
NOOP_TOOL,
SEARCH_STRUCT_TOOL,
SEARCH_TOOL,
UPDATE_MEMORY_STRUCT_TOOL_GRAPH,
UPDATE_MEMORY_TOOL_GRAPH,
)
from mem0.graphs.utils import EXTRACT_ENTITIES_PROMPT, get_update_memory_messages
from mem0.utils.factory import EmbedderFactory, LlmFactory
logger = logging.getLogger(__name__)

4
mem0/memory/telemetry.py
View File

@@ -63,7 +63,9 @@ def capture_event(event_name, memory_instance, additional_data=None):
"collection": memory_instance.collection_name,
"vector_size": memory_instance.embedding_model.config.embedding_dims,
"history_store": "sqlite",
"graph_store": f"{memory_instance.graph.__class__.__module__}.{memory_instance.graph.__class__.__name__}" if memory_instance.config.graph_store.config else None,
"graph_store": f"{memory_instance.graph.__class__.__module__}.{memory_instance.graph.__class__.__name__}"
if memory_instance.config.graph_store.config
else None,
"vector_store": f"{memory_instance.vector_store.__class__.__module__}.{memory_instance.vector_store.__class__.__name__}",
"llm": f"{memory_instance.llm.__class__.__module__}.{memory_instance.llm.__class__.__name__}",
"embedding_model": f"{memory_instance.embedding_model.__class__.__module__}.{memory_instance.embedding_model.__class__.__name__}",