langgraph-checkpointer-couchbase 1.0.7__tar.gz → 1.0.8__tar.gz

{langgraph_checkpointer_couchbase-1.0.7 → langgraph_checkpointer_couchbase-1.0.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-checkpointer-couchbase
-Version: 1.0.7
+Version: 1.0.8
 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
@@ -18,11 +18,13 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: Implementation :: CPython
 Classifier: Programming Language :: Python :: Implementation :: PyPy
 Requires-Python: >=3.8
-Requires-Dist: couchbase>=4.3.5
-Requires-Dist: langchain-openai>=0.3.11
-Requires-Dist: langgraph>=0.3.22
-Requires-Dist: pydantic>=2.11.1
-Requires-Dist: typing-extensions>=4.13.0
+Requires-Dist: couchbase>=4.5.0
+Requires-Dist: langchain-openai>=1.1.3
+Requires-Dist: langchain>=1.1.3
+Requires-Dist: langgraph>=1.0.5
+Requires-Dist: pydantic>=2.12.5
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: typing-extensions>=4.15.0
 Description-Content-Type: text/markdown
 
 # LangGraph Checkpoint Couchbase

{langgraph_checkpointer_couchbase-1.0.7 → langgraph_checkpointer_couchbase-1.0.8}/langgraph_checkpointer_couchbase/async_cb_saver.py

@@ -22,6 +22,11 @@ from .utils import _encode_binary, _decode_binary
 
 logger = logging.getLogger(__name__)
 
+# Default timeout for database operations
+DEFAULT_TIMEOUT = timedelta(seconds=5)
+# Default serialization type for metadata (for backward compatibility)
+DEFAULT_METADATA_TYPE = "json"
+
 class AsyncCouchbaseSaver(BaseCheckpointSaver):
     """A checkpoint saver that stores checkpoints in a Couchbase database."""
 
@@ -94,10 +99,6 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
             auth = PasswordAuthenticator(cb_username, cb_password)
             options = ClusterOptions(auth)
             cluster = await ACluster.connect(cb_conn_str, options)
-            
-            cls.cluster = cluster
-            cls.bucket_name = bucket_name
-            cls.scope_name = scope_name
 
             bucket = cluster.bucket(bucket_name)
             await bucket.on_connect()
@@ -211,7 +212,7 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
             return CheckpointTuple(
                 {"configurable": config_values},
                 checkpoint,
-                self.serde.loads(_decode_binary(doc["metadata"])),
+                self.serde.loads_typed((doc.get("metadata_type", DEFAULT_METADATA_TYPE), _decode_binary(doc["metadata"]))) if doc.get("metadata") else None,
                 (
                     {
                         "configurable": {
@@ -226,6 +227,8 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
                 pending_writes,
             )
 
+        return None
+
     async def alist(
             self,
             config: Optional[RunnableConfig],
@@ -285,7 +288,7 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
                     }
                 },
                 checkpoint,
-                self.serde.loads(_decode_binary(doc["metadata"])),
+                self.serde.loads_typed((doc.get("metadata_type", DEFAULT_METADATA_TYPE), _decode_binary(doc["metadata"]))) if doc.get("metadata") else None,
                 (
                     {
                         "configurable": {
@@ -327,15 +330,16 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
         if serialized_checkpoint:
             serialized_checkpoint = _encode_binary(serialized_checkpoint)
             
-        metadata = self.serde.dumps(metadata)
-        if metadata:
-            metadata = _encode_binary(metadata)
+        # Serialize and encode metadata
+        metadata_type, metadata_bytes = self.serde.dumps_typed(metadata)
+        serialized_metadata = _encode_binary(metadata_bytes) if metadata_bytes else None
         
         doc = {
             "parent_checkpoint_id": config["configurable"].get("checkpoint_id"),
             "type": type_,
             "checkpoint": serialized_checkpoint,
-            "metadata": metadata,
+            "metadata": serialized_metadata,
+            "metadata_type": metadata_type,
             "thread_id" : thread_id,
             "checkpoint_ns": checkpoint_ns,
             "checkpoint_id": checkpoint_id,
@@ -346,7 +350,7 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
         # ensure bucket connected (idempotent)
         await self.bucket.on_connect()
         collection = self.bucket.scope(self.scope_name).collection(self.checkpoints_collection_name)
-        await collection.upsert(upsert_key, (doc), UpsertOptions(timeout=timedelta(seconds=5)))
+        await collection.upsert(upsert_key, (doc), UpsertOptions(timeout=DEFAULT_TIMEOUT))
 
         return {
             "configurable": {
@@ -394,4 +398,4 @@ class AsyncCouchbaseSaver(BaseCheckpointSaver):
                 "type": type_,
                 "value": serialized_value,
             }
-            await collection.upsert(upsert_key, (doc), UpsertOptions(timeout=timedelta(seconds=5)))
+            await collection.upsert(upsert_key, (doc), UpsertOptions(timeout=DEFAULT_TIMEOUT))

{langgraph_checkpointer_couchbase-1.0.7 → langgraph_checkpointer_couchbase-1.0.8}/langgraph_checkpointer_couchbase/couchbase_saver.py

@@ -22,6 +22,11 @@ from langgraph.checkpoint.base import (
 from .utils import _encode_binary, _decode_binary
 
 logger = logging.getLogger(__name__)
+
+# Default timeout for database operations
+DEFAULT_TIMEOUT = timedelta(seconds=5)
+# Default serialization type for metadata (for backward compatibility)
+DEFAULT_METADATA_TYPE = "json"
 class CouchbaseSaver(BaseCheckpointSaver):
     """A checkpoint saver that stores checkpoints in a Couchbase database.
     
@@ -77,11 +82,7 @@ class CouchbaseSaver(BaseCheckpointSaver):
             auth = PasswordAuthenticator(cb_username, cb_password)
             options = ClusterOptions(auth)
             cluster = Cluster(cb_conn_str, options)
-            cluster.wait_until_ready(timedelta(seconds=5))
-            
-            cls.cluster = cluster
-            cls.bucket_name = bucket_name
-            cls.scope_name = scope_name
+            cluster.wait_until_ready(DEFAULT_TIMEOUT)
 
             saver = CouchbaseSaver(cluster, bucket_name, scope_name, checkpoints_collection_name, checkpoint_writes_collection_name)
 
@@ -197,8 +198,8 @@ class CouchbaseSaver(BaseCheckpointSaver):
                     )
 
             # Decode and deserialize metadata
-            metadata = _decode_binary(doc["metadata"])
-            metadata = self.serde.loads(metadata)
+            metadata_data = _decode_binary(doc["metadata"])
+            metadata = self.serde.loads_typed((doc.get("metadata_type", DEFAULT_METADATA_TYPE), metadata_data))
 
             return CheckpointTuple(
                 {"configurable": config_values},
@@ -218,6 +219,8 @@ class CouchbaseSaver(BaseCheckpointSaver):
                 pending_writes,
             )
 
+        return None
+
     def list(
             self,
             config: Optional[RunnableConfig],
@@ -277,7 +280,7 @@ class CouchbaseSaver(BaseCheckpointSaver):
                     }
                 },
                 checkpoint,
-                self.serde.loads(_decode_binary(doc["metadata"])),
+                self.serde.loads_typed((doc.get("metadata_type", DEFAULT_METADATA_TYPE), _decode_binary(doc["metadata"]))),
                 (
                     {
                         "configurable": {
@@ -322,15 +325,15 @@ class CouchbaseSaver(BaseCheckpointSaver):
             serialized_checkpoint = _encode_binary(serialized_checkpoint)
             
         # Serialize and encode metadata
-        metadata_bytes = self.serde.dumps(metadata)
-        if metadata_bytes:
-            metadata = _encode_binary(metadata_bytes)
+        metadata_type, metadata_bytes = self.serde.dumps_typed(metadata)
+        serialized_metadata = _encode_binary(metadata_bytes) if metadata_bytes else None
         
         doc = {
             "parent_checkpoint_id": config["configurable"].get("checkpoint_id"),
             "type": type_,
             "checkpoint": serialized_checkpoint,
-            "metadata": metadata,
+            "metadata": serialized_metadata,
+            "metadata_type": metadata_type,
             "thread_id" : thread_id,
             "checkpoint_ns": checkpoint_ns,
             "checkpoint_id": checkpoint_id,
@@ -338,7 +341,7 @@ class CouchbaseSaver(BaseCheckpointSaver):
         upsert_key = f"{thread_id}::{checkpoint_ns}::{checkpoint_id}"
         
         collection = self.checkpoints_collection
-        collection.upsert(upsert_key, (doc), UpsertOptions(timeout=timedelta(seconds=5)))
+        collection.upsert(upsert_key, (doc), UpsertOptions(timeout=DEFAULT_TIMEOUT))
 
         return {
             "configurable": {
@@ -387,4 +390,4 @@ class CouchbaseSaver(BaseCheckpointSaver):
                 "type": type_,
                 "value": serialized_value,
             }
-            collection.upsert(upsert_key, (doc), UpsertOptions(timeout=timedelta(seconds=5)))
+            collection.upsert(upsert_key, doc, UpsertOptions(timeout=DEFAULT_TIMEOUT))

{langgraph_checkpointer_couchbase-1.0.7 → langgraph_checkpointer_couchbase-1.0.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-checkpointer-couchbase"
-version = "1.0.7"
+version = "1.0.8"
 description = ''
 readme = "README.md"
 requires-python = ">=3.8"
@@ -25,11 +25,13 @@ classifiers = [
   "Programming Language :: Python :: Implementation :: PyPy",
 ]
 dependencies = [
-  "couchbase>=4.3.5",
-  "langgraph>=0.3.22",
-  "langchain-openai>=0.3.11",
-  "pydantic>=2.11.1",
-  "typing_extensions>=4.13.0"
+  "couchbase>=4.5.0",
+  "langgraph>=1.0.5",
+  "langchain-openai>=1.1.3",
+  "pydantic>=2.12.5",
+  "typing_extensions>=4.15.0",
+  "langchain>=1.1.3",
+  "python-dotenv>=1.0.0",
 ]
 
 [project.urls]

{langgraph_checkpointer_couchbase-1.0.7 → langgraph_checkpointer_couchbase-1.0.8}/tests/agent_e2e_test.py

@@ -1,12 +1,12 @@
 import asyncio
 from typing import Literal
-from langchain_core.runnables import ConfigurableField
 from langchain_core.tools import tool
 from langchain_openai import ChatOpenAI
-from langgraph.prebuilt import create_react_agent
+from langchain.agents import create_agent
 from langgraph_checkpointer_couchbase import CouchbaseSaver, AsyncCouchbaseSaver
 from dotenv import load_dotenv
 import os
+
 load_dotenv()
 
 @tool
@@ -21,17 +21,22 @@ def get_weather(city: Literal["nyc", "sf"]):
 
 
 tools = [get_weather]
-model = ChatOpenAI(model_name="gpt-4o-mini", temperature=0)
+model = ChatOpenAI(model="gpt-5-mini", temperature=0)
+
 
 def syncTest():
     with CouchbaseSaver.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",
+        cb_conn_str=os.getenv("CB_CLUSTER"),
+        cb_username=os.getenv("CB_USERNAME"),
+        cb_password=os.getenv("CB_PASSWORD"),
+        bucket_name=os.getenv("CB_BUCKET"),
+        scope_name=os.getenv("CB_SCOPE"),
     ) as checkpointer:
-        graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)
+        graph = create_agent(
+            model,
+            tools=tools,
+            checkpointer=checkpointer,
+        )
         config = {"configurable": {"thread_id": "1"}}
         res = graph.invoke({"messages": [("human", "what's the weather in sf")]}, config)
         
@@ -39,19 +44,26 @@ def syncTest():
         latest_checkpoint_tuple = checkpointer.get_tuple(config)
         checkpoint_tuples = list(checkpointer.list(config))
 
-        print(latest_checkpoint)
-        print(latest_checkpoint_tuple)
-        print(checkpoint_tuples)
+        print("=== Sync Test Results ===")
+        print(f"Response: {res}")
+        print(f"Latest checkpoint: {latest_checkpoint}")
+        print(f"Latest checkpoint tuple: {latest_checkpoint_tuple}")
+        print(f"All checkpoint tuples: {checkpoint_tuples}")
+
 
 async def asyncTest():
     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",
+        cb_conn_str=os.getenv("CB_CLUSTER"),
+        cb_username=os.getenv("CB_USERNAME"),
+        cb_password=os.getenv("CB_PASSWORD"),
+        bucket_name=os.getenv("CB_BUCKET"),
+        scope_name=os.getenv("CB_SCOPE"),
     ) as checkpointer:
-        graph = create_react_agent(model, tools=tools, checkpointer=checkpointer)
+        graph = create_agent(
+            model,
+            tools=tools,
+            checkpointer=checkpointer,
+        )
         config = {"configurable": {"thread_id": "2"}}
         res = await graph.ainvoke(
             {"messages": [("human", "what's the weather in nyc")]}, config
@@ -61,10 +73,16 @@ async def asyncTest():
         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)
+        print("=== Async Test Results ===")
+        print(f"Response: {res}")
+        print(f"Latest checkpoint: {latest_checkpoint}")
+        print(f"Latest checkpoint tuple: {latest_checkpoint_tuple}")
+        print(f"All checkpoint tuples: {checkpoint_tuples}")
+
 
 if __name__ == "__main__":
+    print("Running sync test...")
     syncTest()
-    asyncio.run(asyncTest())
+    print("\nRunning async test...")
+    asyncio.run(asyncTest())
+    print("\nAll tests completed!")