langgraph-checkpointer-couchbase 1.0.4__tar.gz → 1.0.6__tar.gz

{langgraph_checkpointer_couchbase-1.0.4 → langgraph_checkpointer_couchbase-1.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-checkpointer-couchbase
-Version: 1.0.4
+Version: 1.0.6
 Project-URL: Documentation, https://github.com/couchbase-ecosystem/langgraph-checkpointer-couchbase#readme
 Project-URL: Issues, https://github.com/couchbase-ecosystem/langgraph-checkpointer-couchbase/issues
 Project-URL: Source, https://github.com/couchbase-ecosystem/langgraph-checkpointer-couchbase

{langgraph_checkpointer_couchbase-1.0.4 → langgraph_checkpointer_couchbase-1.0.6}/langgraph_checkpointer_couchbase/async_cb_saver.py

@@ -123,10 +123,10 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
         checkpoint_id = get_checkpoint_id(config)
 
         if checkpoint_id:
-            query = f'SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3 ORDER BY checkpoint_id DESC LIMIT 1'
+            query = f'SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3 ORDER BY checkpoint_id DESC LIMIT 1'
             query_params = [thread_id, checkpoint_ns, checkpoint_id]
         else:
-            query = f'SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 ORDER BY checkpoint_id DESC LIMIT 1'
+            query = f'SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 ORDER BY checkpoint_id DESC LIMIT 1'
             query_params = [thread_id, checkpoint_ns]
 
         result = self.cluster.query(query, QueryOptions(positional_parameters=query_params))
@@ -140,7 +140,7 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
             }
             checkpoint = self.serde.loads_typed((doc["type"], _decode_binary(doc["checkpoint"])))
 
-            serialized_writes_query = f'SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoint_writes_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3'
+            serialized_writes_query = f'SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoint_writes_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3'
             serialized_writes_params = [thread_id, checkpoint_ns, doc["checkpoint_id"] or ""]
             
             serialized_writes_result = self.cluster.query(serialized_writes_query, QueryOptions(positional_parameters=serialized_writes_params))
@@ -200,7 +200,7 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
             AsyncIterator[CheckpointTuple]: An asynchronous iterator of checkpoint tuples.
         """
 
-        query = f"SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoints_collection_name}` WHERE 1=1"
+        query = f"SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoints_collection_name}` WHERE 1=1"
         query_params = []
 
         if config is not None:

{langgraph_checkpointer_couchbase-1.0.4 → langgraph_checkpointer_couchbase-1.0.6}/langgraph_checkpointer_couchbase/couchbase_saver.py

@@ -114,7 +114,7 @@ class CouchbaseSaver(BaseCheckpointSaver):
 
         collection_manager = self.bucket.collections()
         try:
-            collection_manager.create_collection(self.checkpoints_collection_name, self.scope_name)
+            collection_manager.create_collection(self.scope_name, self.checkpoints_collection_name)
         except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
@@ -124,7 +124,7 @@ class CouchbaseSaver(BaseCheckpointSaver):
             self.checkpoints_collection = self.bucket.scope(self.scope_name).collection(self.checkpoints_collection_name)
         
         try:
-            collection_manager.create_collection(self.checkpoint_writes_collection_name, self.scope_name)
+            collection_manager.create_collection(self.scope_name, self.checkpoint_writes_collection_name)
         except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
@@ -153,10 +153,10 @@ class CouchbaseSaver(BaseCheckpointSaver):
         checkpoint_id = get_checkpoint_id(config)
 
         if checkpoint_id:
-            query = f'SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3 ORDER BY checkpoint_id DESC LIMIT 1'
+            query = f'SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3 ORDER BY checkpoint_id DESC LIMIT 1'
             query_params = [thread_id, checkpoint_ns, checkpoint_id]
         else:
-            query = f'SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 ORDER BY checkpoint_id DESC LIMIT 1'
+            query = f'SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoints_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 ORDER BY checkpoint_id DESC LIMIT 1'
             query_params = [thread_id, checkpoint_ns]
 
         result = self.cluster.query(query, QueryOptions(positional_parameters=query_params))
@@ -173,7 +173,7 @@ class CouchbaseSaver(BaseCheckpointSaver):
             checkpoint_data = _decode_binary(doc["checkpoint"])
             checkpoint = self.serde.loads_typed((doc["type"], checkpoint_data))
 
-            serialized_writes_query = f'SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoint_writes_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3'
+            serialized_writes_query = f'SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoint_writes_collection_name}` WHERE thread_id = $1 AND checkpoint_ns = $2 AND checkpoint_id = $3'
             serialized_writes_params = [thread_id, checkpoint_ns, doc["checkpoint_id"] or ""]
             serialized_writes_result = self.cluster.query(serialized_writes_query, QueryOptions(positional_parameters=serialized_writes_params))
 
@@ -238,7 +238,7 @@ class CouchbaseSaver(BaseCheckpointSaver):
             Iterator[CheckpointTuple]: An iterator of checkpoint tuples.
         """
 
-        query = f"SELECT * FROM {self.bucket_name}.{self.scope_name}.`{self.checkpoints_collection_name}` WHERE 1=1"
+        query = f"SELECT * FROM `{self.bucket_name}`.`{self.scope_name}`.`{self.checkpoints_collection_name}` WHERE 1=1"
         query_params = []
 
         if config is not None:

{langgraph_checkpointer_couchbase-1.0.4 → langgraph_checkpointer_couchbase-1.0.6}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-checkpointer-couchbase"
-version = "1.0.4"
+version = "1.0.6"
 description = ''
 readme = "README.md"
 requires-python = ">=3.8"