From 3d503cb810bfd6aba0e58f1ae7fde3bd88267d24 Mon Sep 17 00:00:00 2001
From: Aman Rao <54672230+amanrao23@users.noreply.github.com>
Date: Fri, 15 Nov 2024 14:25:32 +0530
Subject: [PATCH] chore: update azure cosmos db (#1484)
Co-authored-by: Alex Yang <himself65@outlook.com>
---
.changeset/olive-bobcats-check.md | 5 +++++
.../src/storage/kvStore/AzureCosmosNoSqlKVStore.ts | 4 ++--
.../src/vector-store/AzureCosmosDBNoSqlVectorStore.ts | 2 +-
3 files changed, 8 insertions(+), 3 deletions(-)
create mode 100644 .changeset/olive-bobcats-check.md
diff --git a/.changeset/olive-bobcats-check.md b/.changeset/olive-bobcats-check.md
new file mode 100644
index 000000000..91f7158b0
--- /dev/null
+++ b/.changeset/olive-bobcats-check.md
@@ -0,0 +1,5 @@
+---
+"llamaindex": patch
+---
+
+Update azure cosmos db
diff --git a/packages/llamaindex/src/storage/kvStore/AzureCosmosNoSqlKVStore.ts b/packages/llamaindex/src/storage/kvStore/AzureCosmosNoSqlKVStore.ts
index 3662c563e..f51c1a76e 100644
--- a/packages/llamaindex/src/storage/kvStore/AzureCosmosNoSqlKVStore.ts
+++ b/packages/llamaindex/src/storage/kvStore/AzureCosmosNoSqlKVStore.ts
@@ -3,7 +3,7 @@ import { Container, CosmosClient, Database } from "@azure/cosmos";
import { DefaultAzureCredential, type TokenCredential } from "@azure/identity";
import { BaseKVStore } from "@llamaindex/core/storage/kv-store";
import { getEnv } from "@llamaindex/env";
-const USER_AGENT_SUFFIX = "LlamaIndex-CDBNoSQL-KVStore-JavaScript";
+const USER_AGENT_SUFFIX = "llamaindex-cdbnosql-kvstore-javascript";
const DEFAULT_CHAT_DATABASE = "KVStoreDB";
const DEFAULT_CHAT_CONTAINER = "KVStoreContainer";
const DEFAULT_OFFER_THROUGHPUT = 400;
@@ -82,7 +82,7 @@ export class AzureCosmosNoSqlKVStore extends BaseKVStore {
super();
if (!cosmosClient) {
throw new Error(
- "CosmosClient is required for AzureCosmosDBNoSQLVectorStore initialization",
+ "CosmosClient is required for AzureCosmosDBNoSQLKVStore initialization",
);
}
this.cosmosClient = cosmosClient;
diff --git a/packages/llamaindex/src/vector-store/AzureCosmosDBNoSqlVectorStore.ts b/packages/llamaindex/src/vector-store/AzureCosmosDBNoSqlVectorStore.ts
index 2c170a219..8aef9b459 100644
--- a/packages/llamaindex/src/vector-store/AzureCosmosDBNoSqlVectorStore.ts
+++ b/packages/llamaindex/src/vector-store/AzureCosmosDBNoSqlVectorStore.ts
@@ -56,7 +56,7 @@ export interface AzureCosmosDBNoSQLConfig
readonly idKey?: string;
}
-const USER_AGENT_SUFFIX = "LlamaIndex-CDBNoSQL-VectorStore-JavaScript";
+const USER_AGENT_SUFFIX = "llamaindex-cdbnosql-vectorstore-javascript";
const DEFAULT_VECTOR_EMBEDDING_POLICY = {
vectorEmbeddings: [
--
GitLab