Merge branch 'main' into async-reader

.github/blunderbuss.yml

@@ -0,0 +1,4 @@
+assign_issues:
+  - googleapis/llama-index-alloydb
+assign_prs:
+  - googleapis/llama-index-alloydb

.github/workflows/schedule_reporter.yml

@@ -0,0 +1,25 @@
+# Copyright 2024 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+name: Schedule Reporter
+
+on:
+  schedule:
+    - cron: '0 6 * * *'  # Runs at 6 AM every morning
+
+jobs:
+  run_reporter:
+    uses: googleapis/langchain-google-alloydb-pg-python/.github/workflows/cloud_build_failure_reporter.yml@main
+    with:
+      trigger_names: "integration-test-nightly,continuous-test-on-merge"

.kokoro/docker/docs/requirements.txt

@@ -32,7 +32,7 @@ platformdirs==4.2.1 \
     --hash=sha256:031cd18d4ec63ec53e82dceaac0417d218a6863f7745dfcc9efe7793b7039bdf \
     --hash=sha256:17d5a1161b3fd67b390023cb2d3b026bbd40abde6fdb052dfbd3a29c3ba22ee1
     # via virtualenv
-virtualenv==20.26.0 \
-    --hash=sha256:0846377ea76e818daaa3e00a4365c018bc3ac9760cbb3544de542885aad61fb3 \
-    --hash=sha256:ec25a9671a5102c8d2657f62792a27b48f016664c6873f6beed3800008577210
-    # via nox
+virtualenv==20.26.6 \
+    --hash=sha256:280aede09a2a5c317e409a00102e7077c6432c5a38f0ef938e643805a7ad2c48 \
+    --hash=sha256:7345cc5b25405607a624d8418154577459c3e0277f5466dd79c49d5e492995f2
+    # via nox

.kokoro/requirements.txt

@@ -277,9 +277,9 @@ jeepney==0.8.0 \
     # via
     #   keyring
     #   secretstorage
-jinja2==3.1.4 \
-    --hash=sha256:4a3aee7acbbe7303aede8e9648d13b8bf88a429282aa6122a993f0ac800cb369 \
-    --hash=sha256:bc5dd2abb727a5319567b7a813e6a2e7318c39f4f487cfe6c89c6f9c7d25197d
+jinja2==3.1.5 \
+    --hash=sha256:8fefff8dc3034e27bb80d67c671eb8a9bc424c0ef4c0826edbff304cceff43bb \
+    --hash=sha256:aba0f4dc9ed8013c424088f68a5c226f7d6097ed89b246d7749c2ec4175c6adb
     # via gcp-releasetool
 keyring==24.3.1 \
     --hash=sha256:c3327b6ffafc0e8befbdb597cacdb4928ffe5c1212f7645f186e6d9957a898db \
@@ -509,9 +509,9 @@ urllib3==2.2.2 \
     # via
     #   requests
     #   twine
-virtualenv==20.25.1 \
-    --hash=sha256:961c026ac520bac5f69acb8ea063e8a4f071bcc9457b9c1f28f6b085c511583a \
-    --hash=sha256:e08e13ecdca7a0bd53798f356d5831434afa5b07b93f0abdf0797b7a06ffe197
+virtualenv==20.26.6 \
+    --hash=sha256:280aede09a2a5c317e409a00102e7077c6432c5a38f0ef938e643805a7ad2c48 \
+    --hash=sha256:7345cc5b25405607a624d8418154577459c3e0277f5466dd79c49d5e492995f2
     # via nox
 wheel==0.43.0 \
     --hash=sha256:465ef92c69fa5c5da2d1cf8ac40559a8c940886afcef87dcf14b9470862f1d85 \
@@ -525,4 +525,4 @@ zipp==3.19.1 \
 # WARNING: The following packages were not pinned, but pip requires them to be
 # pinned when the requirements file includes hashes and the requirement is not
 # satisfied by a package already installed. Consider using the --allow-unsafe flag.
-# setuptools
+# setuptools

pyproject.toml

@@ -39,8 +39,8 @@ Changelog = "https://github.com/googleapis/llama-index-alloydb-pg-python/blob/ma
 test = [
     "black[jupyter]==24.10.0",
     "isort==5.13.2",
-    "mypy==1.13.0",
-    "pytest-asyncio==0.24.0",
+    "mypy==1.14.1",
+    "pytest-asyncio==0.25.2",
     "pytest==8.3.4",
     "pytest-cov==6.0.0",
     "pytest-depends==1.0.1",

requirements.txt

@@ -1,4 +1,4 @@
-google-cloud-alloydb-connector[asyncpg]==1.6.0
-llama-index-core==0.12.5
+google-cloud-alloydb-connector[asyncpg]==1.7.0
+llama-index-core==0.12.11
 pgvector==0.3.6
-SQLAlchemy[asyncio]==2.0.36
+SQLAlchemy[asyncio]==2.0.37

src/llama_index_alloydb_pg/__init__.py

@@ -12,13 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from .chat_store import AlloyDBChatStore
 from .document_store import AlloyDBDocumentStore
 from .engine import AlloyDBEngine, Column
 from .index_store import AlloyDBIndexStore
 from .vector_store import AlloyDBVectorStore
 from .version import __version__
 
 _all = [
+    "AlloyDBChatStore",
     "AlloyDBDocumentStore",
     "AlloyDBEngine",
     "AlloyDBIndexStore",