JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: Missing app_id on Neptune Analytics client (#3278)

Browse Source 
Signed-off-by: Andy Kwok <andy.kwok@improving.com>
This commit is contained in:
Andy Kwok
2025-08-23 21:06:19 +05:30
committed by GitHub
parent 76411e2591
commit 9337a873ec
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/graphs/configs.py
+1
View File
@@ -42,6 +42,7 @@ class MemgraphConfig(BaseModel):
class NeptuneConfig(BaseModel): class NeptuneConfig(BaseModel):
app_id: Optional[str] = Field("Mem0", description="APP_ID for the connection")
endpoint: Optional[str] = ( endpoint: Optional[str] = (
Field( Field(
None, None,
mem0/graphs/neptune/main.py
+4 -1
View File
@@ -4,6 +4,7 @@ from .base import NeptuneBase
try: try:
from langchain_aws import NeptuneAnalyticsGraph from langchain_aws import NeptuneAnalyticsGraph
from botocore.config import Config
except ImportError: except ImportError:
raise ImportError("langchain_aws is not installed. Please install it using 'make install_all'.") raise ImportError("langchain_aws is not installed. Please install it using 'make install_all'.")
@@ -16,9 +17,11 @@ class MemoryGraph(NeptuneBase):
self.graph = None self.graph = None
endpoint = self.config.graph_store.config.endpoint endpoint = self.config.graph_store.config.endpoint
app_id = self.config.graph_store.config.app_id
if endpoint and endpoint.startswith("neptune-graph://"): if endpoint and endpoint.startswith("neptune-graph://"):
graph_identifier = endpoint.replace("neptune-graph://", "") graph_identifier = endpoint.replace("neptune-graph://", "")
self.graph = NeptuneAnalyticsGraph(graph_identifier) self.graph = NeptuneAnalyticsGraph(graph_identifier = graph_identifier,
config = Config(user_agent_appid=app_id))
if not self.graph: if not self.graph:
raise ValueError("Unable to create a Neptune client: missing 'endpoint' in config") raise ValueError("Unable to create a Neptune client: missing 'endpoint' in config")