updated documentation (langchain-ai#2785)

added X-Source to header
agilee · Oct 4, 2023 · 347c839 · 347c839
1 parent 2c94d0c
commit 347c839
Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/docs/extras/modules/data_connection/vectorstores/integrations/vectara.mdx b/docs/extras/modules/data_connection/vectorstores/integrations/vectara.mdx
@@ -30,7 +30,7 @@ const store = await VectaraStore.fromTexts(
 
 You'll need to:
 
-- Create a [free Vectara account](https://console.vectara.com/signup).
+- Create a [free Vectara account](https://vectara.com/integrations/langchain).
 - Create a [corpus](https://docs.vectara.com/docs/console-ui/creating-a-corpus) to store your data
 - Create an [API key](https://docs.vectara.com/docs/common-use-cases/app-authn-authz/api-keys) with QueryService and IndexService access so you can access this corpus
 

diff --git a/langchain/src/vectorstores/vectara.ts b/langchain/src/vectorstores/vectara.ts
@@ -13,6 +13,7 @@ export interface VectaraLibArgs {
   corpusId: number | number[];
   apiKey: string;
   verbose?: boolean;
+  source?: string;
 }
 
 /**
@@ -23,6 +24,7 @@ interface VectaraCallHeader {
     "x-api-key": string;
     "Content-Type": string;
     "customer-id": string;
+    "X-Source": string;
   };
 }
 
@@ -94,6 +96,8 @@ export class VectaraStore extends VectorStore {
 
   private verbose: boolean;
 
+  private source: string;
+
   private vectaraApiTimeoutSeconds = 60;
 
   _vectorstoreType(): string {
@@ -110,6 +114,7 @@ export class VectaraStore extends VectorStore {
       throw new Error("Vectara api key is not provided.");
     }
     this.apiKey = apiKey;
+    this.source = args.source ?? "langchainjs";
 
     const corpusId =
       args.corpusId ??
@@ -153,6 +158,7 @@ export class VectaraStore extends VectorStore {
         "x-api-key": this.apiKey,
         "Content-Type": "application/json",
         "customer-id": this.customerId.toString(),
+        "X-Source": this.source,
       },
     };
   }
@@ -275,6 +281,7 @@ export class VectaraStore extends VectorStore {
           method: "POST",
           headers: {
             "x-api-key": this.apiKey,
+            "X-Source": this.source,
           },
           body: data,
         }