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

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

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: langgraph-checkpointer-couchbase
-Version: 1.0.2
-Project-URL: Documentation, https://github.com/Lokesh Goel/langgraph-checkpointer-couchbase#readme
-Project-URL: Issues, https://github.com/Lokesh Goel/langgraph-checkpointer-couchbase/issues
-Project-URL: Source, https://github.com/Lokesh Goel/langgraph-checkpointer-couchbase
+Version: 1.0.4
+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
 Author-email: Lokesh Goel <lokesh.goel@couchbase.com>
 License-Expression: MIT
 License-File: LICENSE

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

@@ -129,7 +129,6 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
             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]
 
-        print(query)
         result = self.cluster.query(query, QueryOptions(positional_parameters=query_params))
 
         async for row in result:
@@ -148,7 +147,6 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
 
             pending_writes = []
             async for write_doc in serialized_writes_result:
-                print(f"write_doc: {write_doc}")  # Debugging statement to log the contents of write_doc
                 checkpoint_writes = write_doc.get(self.checkpoint_writes_collection_name, {})
                 if "task_id" not in checkpoint_writes:
                     print("Error: 'task_id' is not present in checkpoint_writes")
@@ -293,7 +291,7 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
             "checkpoint_ns": checkpoint_ns,
             "checkpoint_id": checkpoint_id,
         }
-        # print(json.dumps(doc))
+
         upsert_key = f"{thread_id}::{checkpoint_ns}::{checkpoint_id}"
 
         collection = self.bucket.scope(self.scope_name).collection(self.checkpoints_collection_name)

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

@@ -7,7 +7,6 @@ from couchbase.cluster import Cluster
 from couchbase.bucket import Bucket
 from couchbase.auth import PasswordAuthenticator
 from couchbase.options import ClusterOptions, QueryOptions, UpsertOptions
-from couchbase.management.collections import CollectionSpec
 from couchbase.exceptions import CollectionAlreadyExistsException
 
 from langgraph.checkpoint.base import (
@@ -115,9 +114,8 @@ class CouchbaseSaver(BaseCheckpointSaver):
 
         collection_manager = self.bucket.collections()
         try:
-            collection_spec = CollectionSpec(collection_name=self.checkpoints_collection_name, scope_name=self.scope_name)
-            collection_manager.create_collection(collection_spec)
-        except CollectionAlreadyExistsException as caee:
+            collection_manager.create_collection(self.checkpoints_collection_name, self.scope_name)
+        except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
             print(f"Error creating collections: {e}")
@@ -126,9 +124,8 @@ class CouchbaseSaver(BaseCheckpointSaver):
             self.checkpoints_collection = self.bucket.scope(self.scope_name).collection(self.checkpoints_collection_name)
         
         try:
-            collection_spec = CollectionSpec(collection_name=self.checkpoint_writes_collection_name, scope_name=self.scope_name)
-            collection_manager.create_collection(collection_spec)
-        except CollectionAlreadyExistsException as caee:
+            collection_manager.create_collection(self.checkpoint_writes_collection_name, self.scope_name)
+        except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
             print(f"Error creating collections: {e}")

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

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-checkpointer-couchbase"
-version = "1.0.2"
+version = "1.0.4"
 description = ''
 readme = "README.md"
 requires-python = ">=3.8"
@@ -33,9 +33,9 @@ dependencies = [
 ]
 
 [project.urls]
-Documentation = "https://github.com/Lokesh Goel/langgraph-checkpointer-couchbase#readme"
-Issues = "https://github.com/Lokesh Goel/langgraph-checkpointer-couchbase/issues"
-Source = "https://github.com/Lokesh Goel/langgraph-checkpointer-couchbase"
+Documentation = "https://github.com/couchbase-ecosystem/langgraph-checkpointer-couchbase#readme"
+Issues = "https://github.com/couchbase-ecosystem/langgraph-checkpointer-couchbase/issues"
+Source = "https://github.com/couchbase-ecosystem/langgraph-checkpointer-couchbase"
 
 [tool.hatch.version]
 path = "src/langgraph_checkpointer_couchbase/__about__.py"