Skip to content

Commit 6632fe2

Browse files
chore(docs): Update vector store notebook (#96)
* Update vector_store.ipynb 1. Added Installation of additional dependencies "langchain-google-vertexai" it is needed for text embedding generation. 2. Set "GOOGLE_CLOUD_PROJECT" env variable for PROJECT_ID, it is needed for table creation / SQL execution in vector_store module. 3. Specified "id_column" value as "row_id", to match it with table creation in "init_vector_store_table" function, if "id_column" is not set then default value is "langchain_id" * Update docs/vector_store.ipynb Co-authored-by: Averi Kitsch <akitsch@google.com> --------- Co-authored-by: Averi Kitsch <akitsch@google.com>
1 parent 3a8367c commit 6632fe2

1 file changed

Lines changed: 17 additions & 12 deletions

File tree

docs/vector_store.ipynb

Lines changed: 17 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@
5252
}
5353
],
5454
"source": [
55-
"%pip install --upgrade --quiet langchain-google-spanner"
55+
"%pip install --upgrade --quiet langchain-google-spanner langchain-google-vertexai"
5656
]
5757
},
5858
{
@@ -124,7 +124,8 @@
124124
"PROJECT_ID = \"my-project-id\" # @param {type:\"string\"}\n",
125125
"\n",
126126
"# Set the project id\n",
127-
"!gcloud config set project {PROJECT_ID}"
127+
"!gcloud config set project {PROJECT_ID}\n",
128+
"%env GOOGLE_CLOUD_PROJECT={PROJECT_ID}"
128129
]
129130
},
130131
{
@@ -194,14 +195,16 @@
194195
" instance_id=INSTANCE,\n",
195196
" database_id=DATABASE,\n",
196197
" table_name=TABLE_NAME,\n",
197-
" id_column=\"row_id\",\n",
198-
" metadata_columns=[\n",
199-
" TableColumn(name=\"metadata\", type=\"JSON\", is_null=True),\n",
200-
" TableColumn(name=\"title\", type=\"STRING(MAX)\", is_null=False),\n",
201-
" ],\n",
202-
" secondary_indexes=[\n",
203-
" SecondaryIndex(index_name=\"row_id_and_title\", columns=[\"row_id\", \"title\"])\n",
204-
" ],\n",
198+
" # Customize the table creation\n",
199+
" # id_column=\"row_id\",\n",
200+
" # content_column=\"content_column\",\n",
201+
" # metadata_columns=[\n",
202+
" # TableColumn(name=\"metadata\", type=\"JSON\", is_null=True),\n",
203+
" # TableColumn(name=\"title\", type=\"STRING(MAX)\", is_null=False),\n",
204+
" # ],\n",
205+
" # secondary_indexes=[\n",
206+
" # SecondaryIndex(index_name=\"row_id_and_title\", columns=[\"row_id\", \"title\"])\n",
207+
" # ],\n",
205208
")"
206209
]
207210
},
@@ -262,9 +265,11 @@
262265
" instance_id=INSTANCE,\n",
263266
" database_id=DATABASE,\n",
264267
" table_name=TABLE_NAME,\n",
265-
" ignore_metadata_columns=[],\n",
266268
" embedding_service=embeddings,\n",
267-
" metadata_json_column=\"metadata\",\n",
269+
" # Connect to a custom vector store table\n",
270+
" # id_column=\"row_id\",\n",
271+
" # content_column=\"content\",\n",
272+
" # metadata_columns=[\"metadata\", \"title\"],\n",
268273
")"
269274
]
270275
},

0 commit comments

Comments
 (0)