addressed comments

Author: azaddhirajkumar

README.md

@@ -68,58 +68,72 @@ with CouchbaseSaver.from_conn_info(
         bucket_name=os.getenv("CB_BUCKET") or "test",
         scope_name=os.getenv("CB_SCOPE") or "langgraph",
     ) as checkpointer:
-        # Create the agent with checkpointing
-        graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)
-        
-        # Configure with a unique thread ID
-        config = {"configurable": {"thread_id": "1"}}
-        
-        # Run the agent
-        res = graph.invoke({"messages": [("human", "what's the weather in sf")]}, config)
-        
-        # Retrieve checkpoints
-        latest_checkpoint = checkpointer.get(config)
-        latest_checkpoint_tuple = checkpointer.get_tuple(config)
-        checkpoint_tuples = list(checkpointer.list(config))
-
-        print(latest_checkpoint)
-        print(latest_checkpoint_tuple)
-        print(checkpoint_tuples)
+    # Create the agent with checkpointing
+    graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)
+    
+    # Configure with a unique thread ID
+    config = {"configurable": {"thread_id": "1"}}
+    
+    # Run the agent
+    res = graph.invoke({"messages": [("human", "what's the weather in sf")]}, config)
+    
+    # Retrieve checkpoints
+    latest_checkpoint = checkpointer.get(config)
+    latest_checkpoint_tuple = checkpointer.get_tuple(config)
+    checkpoint_tuples = list(checkpointer.list(config))
+
+    print(latest_checkpoint)
+    print(latest_checkpoint_tuple)
+    print(checkpoint_tuples)
 ```
 
 ### Asynchronous Usage
 
 ```python
 import os
+from acouchbase.cluster import Cluster as ACluster
+from couchbase.auth import PasswordAuthenticator
+from couchbase.options import ClusterOptions
 from langgraph_checkpointer_couchbase import AsyncCouchbaseSaver
 from langgraph.graph import create_react_agent
 
-async with AsyncCouchbaseSaver.from_conn_info(
-        cb_conn_str=os.getenv("CB_CLUSTER") or "couchbase://localhost",
-        cb_username=os.getenv("CB_USERNAME") or "Administrator",
-        cb_password=os.getenv("CB_PASSWORD") or "password",
-        bucket_name=os.getenv("CB_BUCKET") or "test",
-        scope_name=os.getenv("CB_SCOPE") or "langgraph",
+auth = PasswordAuthenticator(
+    os.getenv("CB_USERNAME") or "Administrator",
+    os.getenv("CB_PASSWORD") or "password",
+)
+options = ClusterOptions(auth)
+cluster = await ACluster.connect(os.getenv("CB_CLUSTER") or "couchbase://localhost", options)
+
+bucket_name = os.getenv("CB_BUCKET") or "test"
+scope_name = os.getenv("CB_SCOPE") or "langgraph"
+
+async with AsyncCouchbaseSaver.from_cluster(
+        cluster=cluster,
+        bucket_name=bucket_name,
+        scope_name=scope_name,
     ) as checkpointer:
-        # Create the agent with checkpointing
-        graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)
-        
-        # Configure with a unique thread ID
-        config = {"configurable": {"thread_id": "2"}}
-        
-        # Run the agent asynchronously
-        res = await graph.ainvoke(
-            {"messages": [("human", "what's the weather in nyc")]}, config
-        )
-
-        # Retrieve checkpoints asynchronously
-        latest_checkpoint = await checkpointer.aget(config)
-        latest_checkpoint_tuple = await checkpointer.aget_tuple(config)
-        checkpoint_tuples = [c async for c in checkpointer.alist(config)]
-
-        print(latest_checkpoint)
-        print(latest_checkpoint_tuple)
-        print(checkpoint_tuples)
+    # Create the agent with checkpointing
+    graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)
+    
+    # Configure with a unique thread ID
+    config = {"configurable": {"thread_id": "2"}}
+    
+    # Run the agent asynchronously
+    res = await graph.ainvoke(
+        {"messages": [("human", "what's the weather in nyc")]}, config
+    )
+
+    # Retrieve checkpoints asynchronously
+    latest_checkpoint = await checkpointer.aget(config)
+    latest_checkpoint_tuple = await checkpointer.aget_tuple(config)
+    checkpoint_tuples = [c async for c in checkpointer.alist(config)]
+
+    print(latest_checkpoint)
+    print(latest_checkpoint_tuple)
+    print(checkpoint_tuples)
+
+# Close the cluster when done
+await cluster.close()
 ```
 
 ## Configuration Options

langgraph_checkpointer_couchbase/async_cb_saver.py

@@ -1,6 +1,7 @@
 from contextlib import asynccontextmanager
 from datetime import timedelta
 from typing import Any, AsyncIterator, Dict, Optional, Sequence, Tuple
+import logging
 
 from langchain_core.runnables import RunnableConfig
 from acouchbase.cluster import Cluster as ACluster
@@ -19,6 +20,8 @@
 )
 from .utils import _encode_binary, _decode_binary
 
+logger = logging.getLogger(__name__)
+
 class AsyncCouchbaseSaver(BaseCheckpointSaver):
     """A checkpoint saver that stores checkpoints in a Couchbase database."""
 
@@ -50,7 +53,7 @@ async def create_collections(self):
         except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
-            print(f"Error creating collections: {e}")
+            logger.exception("Error creating collections")
             raise e
         finally:
             self.checkpoints_collection =  self.bucket.scope(self.scope_name).collection(self.checkpoints_collection_name)
@@ -60,7 +63,7 @@ async def create_collections(self):
         except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
-            print(f"Error creating collections: {e}")
+            logger.exception("Error creating collections")
             raise e
         finally:
             self.checkpoint_writes_collection = self.bucket.scope(self.scope_name).collection(self.checkpoint_writes_collection_name)
@@ -195,7 +198,7 @@ async def aget_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
             async for write_doc in serialized_writes_result:
                 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")
+                    logger.warning("'task_id' is not present in checkpoint_writes")
                 else:
                     pending_writes.append(
                         (

langgraph_checkpointer_couchbase/couchbase_saver.py

@@ -1,6 +1,7 @@
 from contextlib import contextmanager
 from datetime import timedelta
 from typing import Any, Dict, Iterator, Optional, Sequence, Tuple
+import logging
 
 from langchain_core.runnables import RunnableConfig
 from couchbase.cluster import Cluster
@@ -19,6 +20,8 @@
 )
 
 from .utils import _encode_binary, _decode_binary
+
+logger = logging.getLogger(__name__)
 class CouchbaseSaver(BaseCheckpointSaver):
     """A checkpoint saver that stores checkpoints in a Couchbase database.
     
@@ -118,7 +121,7 @@ def create_collections(self):
         except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
-            print(f"Error creating collections: {e}")
+            logger.exception("Error creating collections")
             raise e
         finally:
             self.checkpoints_collection = self.bucket.scope(self.scope_name).collection(self.checkpoints_collection_name)
@@ -128,7 +131,7 @@ def create_collections(self):
         except CollectionAlreadyExistsException as _:
             pass
         except Exception as e:
-            print(f"Error creating collections: {e}")
+            logger.exception("Error creating collections")
             raise e
         finally:
             self.checkpoint_writes_collection = self.bucket.scope(self.scope_name).collection(self.checkpoint_writes_collection_name)
@@ -181,7 +184,7 @@ def get_tuple(self, config: RunnableConfig) -> Optional[CheckpointTuple]:
             for write_doc in serialized_writes_result:
                 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")
+                    logger.warning("'task_id' is not present in checkpoint_writes")
                 else:
                     # Decode and deserialize value data
                     value_data = _decode_binary(checkpoint_writes["value"])