From c156d0281acc63a3d21af62a109e771c0272522a Mon Sep 17 00:00:00 2001
From: Enric Soler Rastrollo <enric.soler@sage.com>
Date: Mon, 27 Nov 2023 21:51:08 +0100
Subject: [PATCH] BUGFIX: Use embedding key in azure_cosmos_db index creation
 (#13919)

Description: Implement embedding key parametrisation
Issue: https://github.com/langchain-ai/langchain/issues/13918
Dependencies: None
Tag maintainer: @hwchase17 @izzymsft
Twitter handle:@MaddogoS
---
 libs/langchain/langchain/vectorstores/azure_cosmos_db.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/libs/langchain/langchain/vectorstores/azure_cosmos_db.py b/libs/langchain/langchain/vectorstores/azure_cosmos_db.py
index 156b729fe2d..c3997471d43 100644
--- a/libs/langchain/langchain/vectorstores/azure_cosmos_db.py
+++ b/libs/langchain/langchain/vectorstores/azure_cosmos_db.py
@@ -221,7 +221,7 @@ class AzureCosmosDBVectorSearch(VectorStore):
             "indexes": [
                 {
                     "name": self._index_name,
-                    "key": {"vectorContent": "cosmosSearch"},
+                    "key": {self._embedding_key: "cosmosSearch"},
                     "cosmosSearchOptions": {
                         "kind": "vector-ivf",
                         "numLists": num_lists,