langgraph_checkpoint_cosmosdb 0.2.6__tar.gz

langgraph_checkpoint_cosmosdb-0.2.6/.gitignore

@@ -0,0 +1,42 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# Distribution / packaging
+dist/
+build/
+*.egg-info/
+
+# Virtual environments
+venv/
+env/
+.env/
+.venv/
+ENV/
+
+# IDE specific files
+.idea/
+.vscode/
+*.swp
+*.swo
+
+# Unit test / coverage reports
+htmcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# Environment variables
+.env
+
+# Local development settings
+*.log
+.DS_Store

langgraph_checkpoint_cosmosdb-0.2.6/PKG-INFO

@@ -0,0 +1,74 @@
+Metadata-Version: 2.4
+Name: langgraph_checkpoint_cosmosdb
+Version: 0.2.6
+Summary: Azure CosmosDB checkpoint saver implementation for LangGraph
+Author-email: Kamal <skamalj@github.com>
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.10
+Requires-Dist: azure-cosmos
+Requires-Dist: azure-identity
+Requires-Dist: langchain-core
+Requires-Dist: langgraph
+Provides-Extra: dev
+Requires-Dist: black; extra == 'dev'
+Requires-Dist: isort; extra == 'dev'
+Requires-Dist: mypy; extra == 'dev'
+Requires-Dist: pytest-cov; extra == 'dev'
+Requires-Dist: pytest>=7.0; extra == 'dev'
+Provides-Extra: reducer
+Requires-Dist: agentstate-reducer>=0.1.1; extra == 'reducer'
+Description-Content-Type: text/markdown
+
+# LangGraph Checkpoint CosmosDB
+
+This project provides an implementation of a checkpoint saver for LangGraph using Azure CosmosDB. 
+
+## Features
+- Save and retrieve langgraph checkpoints in Azure CosmosDB.
+- Suppports both sync and async saver functionality
+
+## Installation
+
+To install the package, ensure you have Python 3.9 or higher, and run:
+
+```pip install langgraph-checkpoint-cosmosdb```
+
+## Usage
+
+### Setting Up Environment
+
+To use the `CosmosDBSaver`
+- You need to set CosmosDB endpoint and key if you want it to create your specified database and container.
+```
+export COSMOSDB_ENDPOINT='your_cosmosdb_endpoint'
+export COSMOSDB_KEY='your_cosmosdb_key'
+```
+- If database and container already exists then this can work via default RBAC credentials. Ex. az login or by setting TENANT_ID, CLIENT_ID and CLIENT_SECRET in environment.
+    - For user-assigned managed identity, ensure AZURE_CLIENT_ID environment variable is set to the identity's client ID. 
+    - Note that in this case error will be thrown if database and container do not exist.
+  
+
+
+## Import
+
+```
+from langgraph_checkpoint_cosmosdb import CosmosDBSaver
+```
+
+## Initialize the saver
+Database and Container is created if it does not exists
+```
+saver = CosmosDBSaver(database_name='your_database', container_name='your_container')
+```
+
+## Limitations
+List function does not support filters. You can only pass config on thread id to get the list.
+
+```
+print(list(memory.list(config=config)))
+```
+## License
+
+This project is licensed under the MIT License.

langgraph_checkpoint_cosmosdb-0.2.6/README.md

@@ -0,0 +1,51 @@
+# LangGraph Checkpoint CosmosDB
+
+This project provides an implementation of a checkpoint saver for LangGraph using Azure CosmosDB. 
+
+## Features
+- Save and retrieve langgraph checkpoints in Azure CosmosDB.
+- Suppports both sync and async saver functionality
+
+## Installation
+
+To install the package, ensure you have Python 3.9 or higher, and run:
+
+```pip install langgraph-checkpoint-cosmosdb```
+
+## Usage
+
+### Setting Up Environment
+
+To use the `CosmosDBSaver`
+- You need to set CosmosDB endpoint and key if you want it to create your specified database and container.
+```
+export COSMOSDB_ENDPOINT='your_cosmosdb_endpoint'
+export COSMOSDB_KEY='your_cosmosdb_key'
+```
+- If database and container already exists then this can work via default RBAC credentials. Ex. az login or by setting TENANT_ID, CLIENT_ID and CLIENT_SECRET in environment.
+    - For user-assigned managed identity, ensure AZURE_CLIENT_ID environment variable is set to the identity's client ID. 
+    - Note that in this case error will be thrown if database and container do not exist.
+  
+
+
+## Import
+
+```
+from langgraph_checkpoint_cosmosdb import CosmosDBSaver
+```
+
+## Initialize the saver
+Database and Container is created if it does not exists
+```
+saver = CosmosDBSaver(database_name='your_database', container_name='your_container')
+```
+
+## Limitations
+List function does not support filters. You can only pass config on thread id to get the list.
+
+```
+print(list(memory.list(config=config)))
+```
+## License
+
+This project is licensed under the MIT License.

langgraph_checkpoint_cosmosdb-0.2.6/pyproject.toml

@@ -0,0 +1,54 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "langgraph_checkpoint_cosmosdb"
+version = "0.2.6"
+description = "Azure CosmosDB checkpoint saver implementation for LangGraph"
+authors = [{name = "Kamal", email = "skamalj@github.com"}]
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "langchain-core",
+    "langgraph",
+    "azure-cosmos",
+    "azure-identity"
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[project.optional-dependencies]
+reducer = [
+    "agentstate-reducer>=0.1.1",
+]
+dev = [
+    "pytest>=7.0",
+    "pytest-cov",
+    "black",
+    "isort",
+    "mypy"
+]
+
+
+[dependency-groups]
+dev = [
+    "pytest>=7.0",
+    "pytest-cov",
+    "langchain-openai",
+    "langgraph-checkpoint-cosmosdb[reducer]",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/langgraph_checkpoint_cosmosdb"]
+
+[tool.black]
+line-length = 88
+target-version = ['py39']
+
+[tool.isort]
+profile = "black"
+multi_line_output = 3

langgraph_checkpoint_cosmosdb-0.2.6/readme.md

@@ -0,0 +1,51 @@
+# LangGraph Checkpoint CosmosDB
+
+This project provides an implementation of a checkpoint saver for LangGraph using Azure CosmosDB. 
+
+## Features
+- Save and retrieve langgraph checkpoints in Azure CosmosDB.
+- Suppports both sync and async saver functionality
+
+## Installation
+
+To install the package, ensure you have Python 3.9 or higher, and run:
+
+```pip install langgraph-checkpoint-cosmosdb```
+
+## Usage
+
+### Setting Up Environment
+
+To use the `CosmosDBSaver`
+- You need to set CosmosDB endpoint and key if you want it to create your specified database and container.
+```
+export COSMOSDB_ENDPOINT='your_cosmosdb_endpoint'
+export COSMOSDB_KEY='your_cosmosdb_key'
+```
+- If database and container already exists then this can work via default RBAC credentials. Ex. az login or by setting TENANT_ID, CLIENT_ID and CLIENT_SECRET in environment.
+    - For user-assigned managed identity, ensure AZURE_CLIENT_ID environment variable is set to the identity's client ID. 
+    - Note that in this case error will be thrown if database and container do not exist.
+  
+
+
+## Import
+
+```
+from langgraph_checkpoint_cosmosdb import CosmosDBSaver
+```
+
+## Initialize the saver
+Database and Container is created if it does not exists
+```
+saver = CosmosDBSaver(database_name='your_database', container_name='your_container')
+```
+
+## Limitations
+List function does not support filters. You can only pass config on thread id to get the list.
+
+```
+print(list(memory.list(config=config)))
+```
+## License
+
+This project is licensed under the MIT License.

langgraph_checkpoint_cosmosdb-0.2.6/src/langgraph_checkpoint_cosmosdb/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2018 The Python Packaging Authority
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

langgraph_checkpoint_cosmosdb-0.2.6/src/langgraph_checkpoint_cosmosdb/__init__.py

@@ -0,0 +1,7 @@
+
+
+from langgraph_checkpoint_cosmosdb.cosmosdbSaver import CosmosDBSaver
+from .cosmosSerializer import CosmosSerializer
+
+__all__ = ["CosmosDBSaver", "CosmosSerializer"]
+

langgraph_checkpoint_cosmosdb-0.2.6/src/langgraph_checkpoint_cosmosdb/cosmosSerializer.py

@@ -0,0 +1,24 @@
+import base64
+
+class CosmosSerializer:
+    def __init__(self, serde):
+        self.serde = serde
+    
+    def dumps_typed(self, obj):
+        type_, data = self.serde.dumps_typed(obj)
+        data_base64 = base64.b64encode(data).decode('utf-8')
+        return type_, data_base64
+
+    def loads_typed(self, data):
+        type_name, serialized_obj = data
+        serialized_obj = base64.b64decode(serialized_obj.encode('utf-8'))
+        return self.serde.loads_typed((type_name, serialized_obj))
+
+    def dumps(self, obj):
+        data = self.serde.dumps(obj)
+        data_base64 = base64.b64encode(data).decode('utf-8')
+        return data_base64
+
+    def loads(self, serialized_obj):
+        serialized_obj = base64.b64decode(serialized_obj.encode('utf-8'))
+        return self.serde.loads(serialized_obj)