Installation

Install this library in a virtualenv using pip. virtualenv is a tool to create isolated Python environments. The basic problem it addresses is one of dependencies and versions, and indirectly permissions.

With virtualenv, it’s possible to install this library without needing system install permissions, and without clashing with the installed system dependencies.

pip install virtualenv
virtualenv <your-env>
source <your-env>/bin/activate
<your-env>/bin/pip install llama-index-alloydb-pg

pip install virtualenv
virtualenv <your-env>
<your-env>\Scripts\activate
<your-env>\Scripts\pip.exe install llama-index-alloydb-pg

Vector Store Usage

Use a vector store to store embedded data and perform vector search.

import google.auth
from llama_index.core import Settings
from llama_index.embeddings.vertex import VertexTextEmbedding
from llama_index_alloydb_pg import AlloyDBEngine, AlloyDBVectorStore


credentials, project_id = google.auth.default()
engine = await AlloyDBEngine.afrom_instance(
   "project-id", "region", "my-cluster", "my-instance", "my-database"
)
Settings.embed_model = VertexTextEmbedding(
   model_name="textembedding-gecko@003",
   project="project-id",
   credentials=credentials,
)

vector_store = await AlloyDBVectorStore.create(
   engine=engine, table_name="vector_store"
)

Embedding Usage

Use an embedding model to generate embeddings inside AlloyDB with a registered model.

from llama_index_alloydb_pg import AlloyDBEmbedding, AlloyDBEngine


engine = await AlloyDBEngine.afrom_instance(
   "project-id", "region", "my-cluster", "my-instance", "my-database"
)
embed_model = AlloyDBEmbedding(engine=engine, model_id="text-embedding-005")

Chat Store Usage

A chat store serves as a centralized interface to store your chat history.

from llama_index.core.memory import ChatMemoryBuffer
from llama_index_cloud_sql_pg import AlloyDBChatStore, AlloyDBEngine


engine = await AlloyDBEngine.afrom_instance(
   "project-id", "region", "my-cluster", "my-instance", "my-database"
)
chat_store = await AlloyDBChatStore.create(
   engine=engine, table_name="chat_store"
)
memory = ChatMemoryBuffer.from_defaults(
   token_limit=3000,
   chat_store=chat_store,
   chat_store_key="user1",
)

Document Reader Usage

A Reader ingest data from different data sources and data formats into a simple Document representation.

from llama_index.core.memory import ChatMemoryBuffer
from llama_index_cloud_sql_pg import AlloyDBReader, AlloyDBEngine


engine = await AlloyDBEngine.afrom_instance(
   "project-id", "region", "my-cluster", "my-instance", "my-database"
)
reader = await AlloyDBReader.create(
   engine=engine, table_name="my-db-table"
)
documents = reader.load_data()

Document Store Usage

Use a document store to make storage and maintenance of data easier.

from llama_index_alloydb_pg import AlloyDBEngine, AlloyDBDocumentStore


engine = await AlloyDBEngine.afrom_instance(
   "project-id", "region", "my-cluster", "my-instance", "my-database"
)
doc_store = await AlloyDBDocumentStore.create(
   engine=engine, table_name="doc_store"
)

Index Store Usage

Use an index store to keep track of indexes built on documents.

from llama_index_alloydb_pg import AlloyDBIndexStore, AlloyDBEngine


engine = await AlloyDBEngine.from_instance(
   "project-id", "region", "my-cluster", "my-instance", "my-database"
)
index_store = await AlloyDBIndexStore.create(
   engine=engine, table_name="index_store"
)

ToolSpec Usage

Use a tool spec to give an agent AlloyDB AI functions. See the AI tools sample.

from llama_index_alloydb_pg import AlloyDBEngine, AlloyDBToolSpec
engine = await AlloyDBEngine.afrom_instance(
   "project-id", "region", "my-cluster", "my-instance", "my-database"
)
tools = AlloyDBToolSpec(engine).to_tool_list()

Contributions

Contributions to this library are always welcome and highly encouraged.

See CONTRIBUTING for more information how to get started.

Please note that this project is released with a Contributor Code of Conduct. By participating in this project you agree to abide by its terms. See Code of Conduct for more information.