Skip to content
Snippets Groups Projects
Unverified Commit 3d503cb8 authored by Aman Rao's avatar Aman Rao Committed by GitHub
Browse files

chore: update azure cosmos db (#1484)


Co-authored-by: default avatarAlex Yang <himself65@outlook.com>
parent daf8522b
Branches
Tags
No related merge requests found
---
"llamaindex": patch
---
Update azure cosmos db
...@@ -3,7 +3,7 @@ import { Container, CosmosClient, Database } from "@azure/cosmos"; ...@@ -3,7 +3,7 @@ import { Container, CosmosClient, Database } from "@azure/cosmos";
import { DefaultAzureCredential, type TokenCredential } from "@azure/identity"; import { DefaultAzureCredential, type TokenCredential } from "@azure/identity";
import { BaseKVStore } from "@llamaindex/core/storage/kv-store"; import { BaseKVStore } from "@llamaindex/core/storage/kv-store";
import { getEnv } from "@llamaindex/env"; 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_DATABASE = "KVStoreDB";
const DEFAULT_CHAT_CONTAINER = "KVStoreContainer"; const DEFAULT_CHAT_CONTAINER = "KVStoreContainer";
const DEFAULT_OFFER_THROUGHPUT = 400; const DEFAULT_OFFER_THROUGHPUT = 400;
...@@ -82,7 +82,7 @@ export class AzureCosmosNoSqlKVStore extends BaseKVStore { ...@@ -82,7 +82,7 @@ export class AzureCosmosNoSqlKVStore extends BaseKVStore {
super(); super();
if (!cosmosClient) { if (!cosmosClient) {
throw new Error( throw new Error(
"CosmosClient is required for AzureCosmosDBNoSQLVectorStore initialization", "CosmosClient is required for AzureCosmosDBNoSQLKVStore initialization",
); );
} }
this.cosmosClient = cosmosClient; this.cosmosClient = cosmosClient;
......
...@@ -56,7 +56,7 @@ export interface AzureCosmosDBNoSQLConfig ...@@ -56,7 +56,7 @@ export interface AzureCosmosDBNoSQLConfig
readonly idKey?: string; readonly idKey?: string;
} }
const USER_AGENT_SUFFIX = "LlamaIndex-CDBNoSQL-VectorStore-JavaScript"; const USER_AGENT_SUFFIX = "llamaindex-cdbnosql-vectorstore-javascript";
const DEFAULT_VECTOR_EMBEDDING_POLICY = { const DEFAULT_VECTOR_EMBEDDING_POLICY = {
vectorEmbeddings: [ vectorEmbeddings: [
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment