From 6ce9a2860bb0f31139ed645ad3297a018ccfd896 Mon Sep 17 00:00:00 2001 From: Andrew Nguonly Date: Tue, 14 May 2024 12:45:16 -0700 Subject: [PATCH] Rename metadata_query to metadata_filter. --- langgraph/checkpoint/aiosqlite.py | 8 ++++---- langgraph/checkpoint/base.py | 2 +- langgraph/checkpoint/memory.py | 12 ++++++------ langgraph/checkpoint/sqlite.py | 8 ++++---- 4 files changed, 15 insertions(+), 15 deletions(-) diff --git a/langgraph/checkpoint/aiosqlite.py b/langgraph/checkpoint/aiosqlite.py index c0ae80e82..acf62b355 100644 --- a/langgraph/checkpoint/aiosqlite.py +++ b/langgraph/checkpoint/aiosqlite.py @@ -257,7 +257,7 @@ class AsyncSqliteSaver(BaseCheckpointSaver, AbstractAsyncContextManager): async def asearch( self, - metadata_query: CheckpointMetadata, + metadata_filter: CheckpointMetadata, *, before: Optional[RunnableConfig] = None, limit: Optional[int] = None, @@ -265,12 +265,12 @@ class AsyncSqliteSaver(BaseCheckpointSaver, AbstractAsyncContextManager): """Search for checkpoints by metadata asynchronously. This method retrieves a list of checkpoint tuples from the SQLite - database based on the provided metadata query. The metadata query does + database based on the provided metadata filter. The metadata filter does not need to contain all keys defined in the CheckpointMetadata class. The checkpoints are ordered by timestamp in descending order. Args: - metadata_query (CheckpointMetadata): The metadata query to use for searching the checkpoints. + metadata_filter (CheckpointMetadata): The metadata filter to use for searching the checkpoints. before (Optional[RunnableConfig]): If provided, only checkpoints before the specified timestamp are returned. Defaults to None. limit (Optional[int]): The maximum number of checkpoints to return. Defaults to None. @@ -282,7 +282,7 @@ class AsyncSqliteSaver(BaseCheckpointSaver, AbstractAsyncContextManager): # construct query SELECT = "SELECT thread_id, thread_ts, parent_ts, checkpoint, metadata FROM checkpoints " WHERE = search_where( - metadata_query, [] if before is None else ["thread_ts < ?"] + metadata_filter, [] if before is None else ["thread_ts < ?"] ) ORDER_BY = "ORDER BY thread_ts DESC " LIMIT = f"LIMIT {limit}" if limit else "" diff --git a/langgraph/checkpoint/base.py b/langgraph/checkpoint/base.py index 8306511bc..19ced7cfa 100644 --- a/langgraph/checkpoint/base.py +++ b/langgraph/checkpoint/base.py @@ -155,7 +155,7 @@ class BaseCheckpointSaver(ABC): def search( self, - metadata: CheckpointMetadata, + metadata_filter: CheckpointMetadata, *, before: Optional[RunnableConfig] = None, limit: Optional[int] = None, diff --git a/langgraph/checkpoint/memory.py b/langgraph/checkpoint/memory.py index 79b516446..57217f6ad 100644 --- a/langgraph/checkpoint/memory.py +++ b/langgraph/checkpoint/memory.py @@ -122,7 +122,7 @@ class MemorySaver(BaseCheckpointSaver): def search( self, - metadata_query: CheckpointMetadata, + metadata_filter: CheckpointMetadata, *, before: Optional[RunnableConfig] = None, limit: Optional[int] = None, @@ -130,12 +130,12 @@ class MemorySaver(BaseCheckpointSaver): """Search for checkpoints by metadata. This method retrieves a list of checkpoint tuples from the in-memory - storage based on the provided metadata query. The metadata query does + storage based on the provided metadata filter. The metadata filter does not need to contain all keys defined in the CheckpointMetadata class. The checkpoints are ordered by timestamp in descending order. Args: - metadata_query (CheckpointMetadata): The metadata query to use for searching the checkpoints. + metadata_filter (CheckpointMetadata): The metadata filter to use for searching the checkpoints. before (Optional[RunnableConfig]): If provided, only checkpoints before the specified timestamp are returned. Defaults to None. limit (Optional[int]): The maximum number of checkpoints to return. Defaults to None. @@ -152,7 +152,7 @@ class MemorySaver(BaseCheckpointSaver): metadata = self.serde.loads(metadata_bytes) all_keys_match = all( query_value == metadata[query_key] - for query_key, query_value in metadata_query.items() + for query_key, query_value in metadata_filter.items() ) # if all query key/value pairs match, yield the checkpoint @@ -242,7 +242,7 @@ class MemorySaver(BaseCheckpointSaver): async def asearch( self, - metadata_query: CheckpointMetadata, + metadata_filter: CheckpointMetadata, *, before: Optional[RunnableConfig] = None, limit: Optional[int] = None, @@ -254,7 +254,7 @@ class MemorySaver(BaseCheckpointSaver): """ loop = asyncio.get_running_loop() iter = await loop.run_in_executor( - None, partial(self.search, before=before, limit=limit), metadata_query + None, partial(self.search, before=before, limit=limit), metadata_filter ) while True: diff --git a/langgraph/checkpoint/sqlite.py b/langgraph/checkpoint/sqlite.py index b4db7584f..a1210f8e1 100644 --- a/langgraph/checkpoint/sqlite.py +++ b/langgraph/checkpoint/sqlite.py @@ -340,7 +340,7 @@ class SqliteSaver(BaseCheckpointSaver, AbstractContextManager): def search( self, - metadata_query: CheckpointMetadata, + metadata_filter: CheckpointMetadata, *, before: Optional[RunnableConfig] = None, limit: Optional[int] = None, @@ -348,12 +348,12 @@ class SqliteSaver(BaseCheckpointSaver, AbstractContextManager): """Search for checkpoints by metadata. This method retrieves a list of checkpoint tuples from the SQLite - database based on the provided metadata query. The metadata query does + database based on the provided metadata filter. The metadata filter does not need to contain all keys defined in the CheckpointMetadata class. The checkpoints are ordered by timestamp in descending order. Args: - metadata_query (CheckpointMetadata): The metadata query to use for searching the checkpoints. + metadata_filter (CheckpointMetadata): The metadata filter to use for searching the checkpoints. before (Optional[RunnableConfig]): If provided, only checkpoints before the specified timestamp are returned. Defaults to None. limit (Optional[int]): The maximum number of checkpoints to return. Defaults to None. @@ -363,7 +363,7 @@ class SqliteSaver(BaseCheckpointSaver, AbstractContextManager): # construct query SELECT = "SELECT thread_id, thread_ts, parent_ts, checkpoint, metadata FROM checkpoints " WHERE = search_where( - metadata_query, [] if before is None else ["thread_ts < ?"] + metadata_filter, [] if before is None else ["thread_ts < ?"] ) ORDER_BY = "ORDER BY thread_ts DESC " LIMIT = f"LIMIT {limit}" if limit else ""