Merge branch 'main' into feat/chat-ui-with-rsc-example

apps/docs/CHANGELOG.md

@@ -1,32 +1,5 @@
 # docs
 
-## 0.0.122
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 0.0.121
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 0.0.120
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.0.119
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.0.118
 
 ### Patch Changes

apps/docs/package.json

@@ -1,6 +1,6 @@
 {
   "name": "docs",
-  "version": "0.0.122",
+  "version": "0.0.118",
   "private": true,
   "scripts": {
     "docusaurus": "docusaurus",

apps/next/CHANGELOG.md

@@ -1,40 +1,5 @@
 # @llamaindex/doc
 
-## 0.0.20
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-  - @llamaindex/cloud@2.0.11
-  - llamaindex@0.8.17
-  - @llamaindex/node-parser@0.0.12
-  - @llamaindex/openai@0.1.36
-  - @llamaindex/readers@1.0.13
-
-## 0.0.19
-
-### Patch Changes
-
-- Updated dependencies [e60328b]
-  - @llamaindex/readers@1.0.12
-  - llamaindex@0.8.16
-
-## 0.0.18
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.0.17
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.0.16
 
 ### Patch Changes

apps/next/next.config.mjs

@@ -6,7 +6,6 @@ const withMDX = createMDX();
 const config = {
   reactStrictMode: true,
   transpilePackages: ["monaco-editor"],
-  serverExternalPackages: ["@huggingface/transformers"],
   webpack: (config, { isServer }) => {
     if (Array.isArray(config.target) && config.target.includes("web")) {
       config.target = ["web", "es2020"];
@@ -27,7 +26,6 @@ const config = {
         }),
       );
     }
-    config.resolve.alias["replicate"] = false;
     return config;
   },
 };

apps/next/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/doc",
-  "version": "0.0.20",
+  "version": "0.0.16",
   "private": true,
   "scripts": {
     "build": "pnpm run build:docs && next build",

apps/next/src/app/api/chat/route.ts

@@ -1,9 +1,6 @@
-import { llm } from "@/lib/utils";
-import { LlamaIndexAdapter, type Message } from "ai";
-import { Settings, SimpleChatEngine, type ChatMessage } from "llamaindex";
-import { NextResponse, type NextRequest } from "next/server";
-
-Settings.llm = llm;
+import { Message } from "ai";
+import { simulateReadableStream } from "ai/test";
+import { NextRequest, NextResponse } from "next/server";
 
 export async function POST(request: NextRequest) {
   try {
@@ -15,16 +12,19 @@ export async function POST(request: NextRequest) {
         { status: 400 },
       );
     }
-
-    const chatEngine = new SimpleChatEngine();
-
-    return LlamaIndexAdapter.toDataStreamResponse(
-      await chatEngine.chat({
-        message: userMessage.content,
-        chatHistory: messages as ChatMessage[],
-        stream: true,
-      }),
-      {},
+    const mockResponse = `Hello! This is a mock response to: ${userMessage.content}`;
+    return new Response(
+      simulateReadableStream({
+        chunkDelayInMs: 20,
+        values: mockResponse.split(" ").map((t) => `0:"${t} "\n`),
+      }).pipeThrough(new TextEncoderStream()),
+      {
+        status: 200,
+        headers: {
+          "X-Vercel-AI-Data-Stream": "v1",
+          "Content-Type": "text/plain; charset=utf-8",
+        },
+      },
     );
   } catch (error) {
     const detail = (error as Error).message;

apps/next/src/components/demo/chat/rsc/ai-action.tsx

@@ -1,8 +1,7 @@
-import { llm } from "@/lib/utils";
 import { Markdown } from "@llamaindex/chat-ui/widgets";
-import { generateId, Message } from "ai";
+import { generateId, Message, parseStreamPart } from "ai";
 import { createAI, createStreamableUI, getMutableAIState } from "ai/rsc";
-import { type ChatMessage, Settings, SimpleChatEngine } from "llamaindex";
+import { simulateReadableStream } from "ai/test";
 import { ReactNode } from "react";
 
 type ServerState = Message[];
@@ -11,8 +10,6 @@ type Actions = {
   chat: (message: Message) => Promise<Message & { display: ReactNode }>;
 };
 
-Settings.llm = llm;
-
 export const AI = createAI<ServerState, FrontendState, Actions>({
   initialAIState: [],
   initialUIState: [],
@@ -23,30 +20,31 @@ export const AI = createAI<ServerState, FrontendState, Actions>({
       const aiState = getMutableAIState<typeof AI>();
       aiState.update((prev) => [...prev, message]);
 
+      const mockResponse = `Hello! This is a mock response to: ${message.content}`;
+      const responseStream = simulateReadableStream({
+        chunkDelayInMs: 20,
+        values: mockResponse.split(" ").map((t) => `0:"${t} "\n`),
+      });
+
       const uiStream = createStreamableUI();
-      const chatEngine = new SimpleChatEngine();
       const assistantMessage: Message = {
         id: generateId(),
         role: "assistant",
         content: "",
       };
 
-      // run the async function without blocking
-      (async () => {
-        const chatResponse = await chatEngine.chat({
-          stream: true,
-          message: message.content,
-          chatHistory: aiState.get() as ChatMessage[],
-        });
-
-        for await (const chunk of chatResponse) {
-          assistantMessage.content += chunk.delta;
-          uiStream.update(<Markdown content={assistantMessage.content} />);
-        }
-
-        aiState.done([...aiState.get(), assistantMessage]);
-        uiStream.done();
-      })();
+      responseStream.pipeTo(
+        new WritableStream({
+          write: async (message) => {
+            assistantMessage.content += parseStreamPart(message).value;
+            uiStream.update(<Markdown content={assistantMessage.content} />);
+          },
+          close: () => {
+            aiState.done([...aiState.get(), assistantMessage]);
+            uiStream.done();
+          },
+        }),
+      );
 
       return {
         ...assistantMessage,

apps/next/src/components/website/Footer.tsx

@@ -85,33 +85,6 @@ const Footer = () => {
                   <Text as="span">SharePoint</Text>
                 </a>
               </li>
-              <li>
-                <a
-                  href="https://llamaindex.ai/llamacloud-aws-s3-data-loading-for-generative-ai"
-                  data-tracking-variant="link"
-                  data-tracking-section="footer"
-                >
-                  <Text as="span">AWS S3</Text>
-                </a>
-              </li>
-              <li>
-                <a
-                  href="https://llamaindex.ai/llamacloud-azure-blob-storage-data-loading-for-generative-ai"
-                  data-tracking-variant="link"
-                  data-tracking-section="footer"
-                >
-                  <Text as="span">Azure Blob Storage</Text>
-                </a>
-              </li>
-              <li>
-                <a
-                  href="https://llamaindex.ai/llamacloud-google-drive-data-loading-for-generative-ai"
-                  data-tracking-variant="link"
-                  data-tracking-section="footer"
-                >
-                  <Text as="span">Google Drive</Text>
-                </a>
-              </li>
             </ul>
           </div>
           <div>
@@ -198,6 +171,11 @@ const Footer = () => {
                   <Text as="span">SEC Insights</Text>
                 </a>
               </li>
+              <li>
+                <a href="https://chat.llamaindex.ai/">
+                  <Text as="span">Chat LlamaIndex</Text>
+                </a>
+              </li>
               <li>
                 <a href="https://github.com/run-llama/llamabot">
                   <Text as="span">LlamaBot</Text>

apps/next/src/lib/utils.ts

@@ -1,34 +1,6 @@
-import { clsx, type ClassValue } from "clsx";
-import { LLM, LLMMetadata } from "llamaindex";
-import { twMerge } from "tailwind-merge";
+import { clsx, type ClassValue } from "clsx"
+import { twMerge } from "tailwind-merge"
 
 export function cn(...inputs: ClassValue[]) {
   return twMerge(clsx(inputs))
 }
-
-class MockLLM  {
-  metadata: LLMMetadata = {
-    model: "MockLLM",
-    temperature: 0.5,
-    topP: 0.5,
-    contextWindow: 1024,
-    tokenizer: undefined,
-  };
-
-  chat() {
-    const mockResponse = "Hello! This is a mock response";
-    return Promise.resolve(
-      new ReadableStream({
-        async start(controller) {
-          for (const char of mockResponse) {
-            controller.enqueue({ delta: char });
-            await new Promise((resolve) => setTimeout(resolve, 20));
-          }
-          controller.close();
-        },
-      }),
-    );
-  }
-}
-
-export const llm = new MockLLM() as unknown as LLM;

e2e/examples/cloudflare-worker-agent/CHANGELOG.md

@@ -1,32 +1,5 @@
 # @llamaindex/cloudflare-worker-agent-test
 
-## 0.0.113
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 0.0.112
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 0.0.111
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.0.110
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.0.109
 
 ### Patch Changes

e2e/examples/cloudflare-worker-agent/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/cloudflare-worker-agent-test",
-  "version": "0.0.113",
+  "version": "0.0.109",
   "type": "module",
   "private": true,
   "scripts": {

e2e/examples/llama-parse-browser/CHANGELOG.md

@@ -1,11 +1,5 @@
 # @llamaindex/llama-parse-browser-test
 
-## 0.0.31
-
-### Patch Changes
-
-- @llamaindex/cloud@2.0.11
-
 ## 0.0.30
 
 ### Patch Changes

e2e/examples/llama-parse-browser/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/llama-parse-browser-test",
   "private": true,
-  "version": "0.0.31",
+  "version": "0.0.30",
   "type": "module",
   "scripts": {
     "dev": "vite",

e2e/examples/nextjs-agent/CHANGELOG.md

@@ -1,32 +1,5 @@
 # @llamaindex/next-agent-test
 
-## 0.1.113
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 0.1.112
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 0.1.111
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.1.110
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.1.109
 
 ### Patch Changes

e2e/examples/nextjs-agent/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/next-agent-test",
-  "version": "0.1.113",
+  "version": "0.1.109",
   "private": true,
   "scripts": {
     "dev": "next dev",

e2e/examples/nextjs-edge-runtime/CHANGELOG.md

@@ -1,32 +1,5 @@
 # test-edge-runtime
 
-## 0.1.112
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 0.1.111
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 0.1.110
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.1.109
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.1.108
 
 ### Patch Changes

e2e/examples/nextjs-edge-runtime/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/nextjs-edge-runtime-test",
-  "version": "0.1.112",
+  "version": "0.1.108",
   "private": true,
   "scripts": {
     "dev": "next dev",

e2e/examples/nextjs-node-runtime/CHANGELOG.md

@@ -1,32 +1,5 @@
 # @llamaindex/next-node-runtime
 
-## 0.0.94
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 0.0.93
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 0.0.92
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.0.91
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.0.90
 
 ### Patch Changes

e2e/examples/nextjs-node-runtime/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/next-node-runtime-test",
-  "version": "0.0.94",
+  "version": "0.0.90",
   "private": true,
   "scripts": {
     "dev": "next dev",

e2e/examples/waku-query-engine/CHANGELOG.md

@@ -1,32 +1,5 @@
 # @llamaindex/waku-query-engine-test
 
-## 0.0.113
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 0.0.112
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 0.0.111
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.0.110
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.0.109
 
 ### Patch Changes

e2e/examples/waku-query-engine/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/waku-query-engine-test",
-  "version": "0.0.113",
+  "version": "0.0.109",
   "type": "module",
   "private": true,
   "scripts": {

examples/cosmosdb/queryVectorData.ts

@@ -83,6 +83,14 @@ async function query() {
     });
   }
 
+  // configure the Azure CosmosDB NoSQL Vector Store
+  const dbConfig: AzureCosmosDBNoSQLConfig = {
+    client: cosmosClient,
+    databaseName,
+    containerName,
+    flatMetadata: false,
+  };
+
   // use Azure CosmosDB as a vectorStore, docStore, and indexStore
   const { vectorStore, docStore, indexStore } = await initializeStores();
 

packages/autotool/CHANGELOG.md

@@ -1,32 +1,5 @@
 # @llamaindex/autotool
 
-## 5.0.17
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 5.0.16
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 5.0.15
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 5.0.14
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 5.0.13
 
 ### Patch Changes

packages/autotool/examples/01_node/CHANGELOG.md

@@ -1,36 +1,5 @@
 # @llamaindex/autotool-01-node-example
 
-## 0.0.60
-
-### Patch Changes
-
-- llamaindex@0.8.17
-- @llamaindex/autotool@5.0.17
-
-## 0.0.59
-
-### Patch Changes
-
-- llamaindex@0.8.16
-- @llamaindex/autotool@5.0.16
-
-## 0.0.58
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-  - @llamaindex/autotool@5.0.15
-
-## 0.0.57
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-  - @llamaindex/autotool@5.0.14
-
 ## 0.0.56
 
 ### Patch Changes

packages/autotool/examples/01_node/package.json

@@ -13,5 +13,5 @@
   "scripts": {
     "start": "node --import tsx --import @llamaindex/autotool/node ./src/index.ts"
   },
-  "version": "0.0.60"
+  "version": "0.0.56"
 }

packages/autotool/examples/02_nextjs/CHANGELOG.md

@@ -1,36 +1,5 @@
 # @llamaindex/autotool-02-next-example
 
-## 0.1.104
-
-### Patch Changes
-
-- llamaindex@0.8.17
-- @llamaindex/autotool@5.0.17
-
-## 0.1.103
-
-### Patch Changes
-
-- llamaindex@0.8.16
-- @llamaindex/autotool@5.0.16
-
-## 0.1.102
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-  - @llamaindex/autotool@5.0.15
-
-## 0.1.101
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-  - @llamaindex/autotool@5.0.14
-
 ## 0.1.100
 
 ### Patch Changes

packages/autotool/examples/02_nextjs/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/autotool-02-next-example",
   "private": true,
-  "version": "0.1.104",
+  "version": "0.1.100",
   "scripts": {
     "dev": "next dev",
     "build": "next build",

packages/autotool/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/autotool",
   "type": "module",
-  "version": "5.0.17",
+  "version": "5.0.13",
   "description": "auto transpile your JS function to LLM Agent compatible",
   "files": [
     "dist",

packages/cloud/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/cloud
 
-## 2.0.11
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 2.0.10
 
 ### Patch Changes

packages/cloud/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/cloud",
-  "version": "2.0.11",
+  "version": "2.0.10",
   "type": "module",
   "license": "MIT",
   "scripts": {

packages/community/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/community
 
-## 0.0.69
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 0.0.68
 
 ### Patch Changes

packages/community/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/community",
   "description": "Community package for LlamaIndexTS",
-  "version": "0.0.69",
+  "version": "0.0.68",
   "type": "module",
   "types": "dist/type/index.d.ts",
   "main": "dist/cjs/index.js",

packages/core/CHANGELOG.md

@@ -1,11 +1,5 @@
 # @llamaindex/core
 
-## 0.4.11
-
-### Patch Changes
-
-- 6d22fa2: Get PromptTemplate template variables at run-time
-
 ## 0.4.10
 
 ### Patch Changes

packages/core/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/core",
   "type": "module",
-  "version": "0.4.11",
+  "version": "0.4.10",
   "description": "LlamaIndex Core Module",
   "exports": {
     "./agent": {

packages/core/src/prompts/base.ts

@@ -32,10 +32,6 @@ export abstract class BasePromptTemplate<
   const Vars extends readonly string[] = string[],
 > {
   metadata: Metadata = {};
-  /**
-   * Set of template variables used in the prompt template. Used for type hints only.
-   * To get the list of template variables used in the prompt at run-time, use the `vars` method.
-   */
   templateVars: Set<string> = new Set();
   options: Partial<Record<TemplatesVar[number] | (string & {}), string>> = {};
   outputParser: BaseOutputParser | undefined;
@@ -227,13 +223,4 @@ export class PromptTemplate<
   get template(): Template {
     return this.#template;
   }
-
-  /**
-   * Returns all the template variables used in the prompt template.
-   */
-  vars(): string[] {
-    const template = this.template;
-    const matches = template.match(/\{([^}]+)\}/g) || [];
-    return [...new Set(matches.map((match) => match.slice(1, -1)))];
-  }
 }

packages/core/tests/prompts.test.ts

@@ -158,15 +158,4 @@ describe("PromptTemplate", () => {
     const formatted = prompt.format({ text: "world", foo: "bar" });
     expect(formatted).toBe("hello world bar\noutput_instruction");
   });
-
-  test("should return all unique template vars of a prompt", () => {
-    const prompt = new PromptTemplate({
-      template: "Hello {name}! Your age is {age}. Nice to meet you {name}!",
-    });
-
-    const vars = prompt.vars();
-    expect(vars).toHaveLength(2);
-    expect(vars).toContain("name");
-    expect(vars).toContain("age");
-  });
 });

packages/experimental/CHANGELOG.md

@@ -1,32 +1,5 @@
 # @llamaindex/experimental
 
-## 0.0.129
-
-### Patch Changes
-
-- llamaindex@0.8.17
-
-## 0.0.128
-
-### Patch Changes
-
-- llamaindex@0.8.16
-
-## 0.0.127
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.0.126
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.0.125
 
 ### Patch Changes

packages/experimental/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/experimental",
   "description": "Experimental package for LlamaIndexTS",
-  "version": "0.0.129",
+  "version": "0.0.125",
   "type": "module",
   "types": "dist/type/index.d.ts",
   "main": "dist/cjs/index.js",

packages/llamaindex/CHANGELOG.md

@@ -1,45 +1,5 @@
 # llamaindex
 
-## 0.8.17
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-  - @llamaindex/cloud@2.0.11
-  - @llamaindex/node-parser@0.0.12
-  - @llamaindex/anthropic@0.0.20
-  - @llamaindex/clip@0.0.20
-  - @llamaindex/deepinfra@0.0.20
-  - @llamaindex/huggingface@0.0.20
-  - @llamaindex/ollama@0.0.27
-  - @llamaindex/openai@0.1.36
-  - @llamaindex/portkey-ai@0.0.20
-  - @llamaindex/replicate@0.0.20
-  - @llamaindex/readers@1.0.13
-  - @llamaindex/groq@0.0.35
-  - @llamaindex/vllm@0.0.6
-
-## 0.8.16
-
-### Patch Changes
-
-- Updated dependencies [e60328b]
-  - @llamaindex/readers@1.0.12
-
-## 0.8.15
-
-### Patch Changes
-
-- 3d503cb: Update azure cosmos db
-- 5dae534: fix: propagate queryStr to concrete vectorStore
-
-## 0.8.14
-
-### Patch Changes
-
-- 630b425: feat: add Azure CosmosDB NoSql Chat store
-
 ## 0.8.13
 
 ### Patch Changes

packages/llamaindex/package.json

@@ -1,6 +1,6 @@
 {
   "name": "llamaindex",
-  "version": "0.8.17",
+  "version": "0.8.13",
   "license": "MIT",
   "type": "module",
   "keywords": [

packages/llamaindex/src/indices/vectorStore/index.ts

@@ -18,7 +18,6 @@ import {
   type NodeWithScore,
 } from "@llamaindex/core/schema";
 import type { BaseIndexStore } from "@llamaindex/core/storage/index-store";
-import { extractText } from "@llamaindex/core/utils";
 import type { ServiceContext } from "../../ServiceContext.js";
 import { nodeParserFromSettingsOrContext } from "../../Settings.js";
 import { RetrieverQueryEngine } from "../../engines/query/RetrieverQueryEngine.js";
@@ -450,13 +449,8 @@ export class VectorIndexRetriever extends BaseRetriever {
     filters?: MetadataFilters,
   ): Promise<NodeWithScore[]> {
     // convert string message to multi-modal format
-
-    let queryStr = query;
     if (typeof query === "string") {
-      queryStr = query;
-      query = [{ type: "text", text: queryStr }];
-    } else {
-      queryStr = extractText(query);
+      query = [{ type: "text", text: query }];
     }
     // overwrite embed model if specified, otherwise use the one from the vector store
     const embedModel = this.index.embedModel ?? vectorStore.embedModel;
@@ -466,7 +460,6 @@ export class VectorIndexRetriever extends BaseRetriever {
       const queryEmbedding = await embedModel.getQueryEmbedding(item);
       if (queryEmbedding) {
         const result = await vectorStore.query({
-          queryStr,
           queryEmbedding,
           mode: VectorStoreQueryMode.DEFAULT,
           similarityTopK: this.topK[type]!,

packages/llamaindex/src/storage/chatStore/AzureCosmosNoSqlChatStore.ts

@@ -1,345 +0,0 @@
-import { CosmosClient, type Container, type Database } from "@azure/cosmos";
-import { DefaultAzureCredential, type TokenCredential } from "@azure/identity";
-import type {
-  ChatMessage,
-  MessageContent,
-  MessageType,
-} from "@llamaindex/core/llms";
-import { BaseChatStore } from "@llamaindex/core/storage/chat-store";
-import { getEnv } from "@llamaindex/env";
-
-const USER_AGENT_SUFFIX = "llamaindex-cdbnosql-chatstore-javascript";
-const DEFAULT_CHAT_DATABASE = "ChatMessagesDB";
-const DEFAULT_CHAT_CONTAINER = "ChatMessagesContainer";
-const DEFAULT_OFFER_THROUGHPUT = 400;
-
-function parseConnectionString(connectionString: string): {
-  endpoint: string;
-  key: string;
-} {
-  const parts = connectionString.split(";");
-  let endpoint = "";
-  let accountKey = "";
-
-  parts.forEach((part) => {
-    const [key, value] = part.split("=");
-    if (key && key.trim() === "AccountEndpoint") {
-      endpoint = value?.trim() ?? "";
-    } else if ((key ?? "").trim() === "AccountKey") {
-      accountKey = value?.trim() ?? "";
-    }
-  });
-
-  if (!endpoint || !accountKey) {
-    throw new Error(
-      "Invalid connection string: missing AccountEndpoint or AccountKey.",
-    );
-  }
-  return { endpoint, key: accountKey };
-}
-
-export interface AzureCosmosChatDatabaseProperties {
-  throughput?: number;
-}
-
-export interface AzureCosmosChatContainerProperties {
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  [key: string]: any;
-}
-
-export interface AzureCosmosNoSqlChatStoreConfig {
-  cosmosClient?: CosmosClient;
-  dbName?: string;
-  containerName?: string;
-  userId?: string;
-  sessionId?: string;
-  cosmosContainerProperties?: AzureCosmosChatContainerProperties;
-  cosmosDatabaseProperties?: AzureCosmosChatDatabaseProperties;
-  ttlInSeconds?: number;
-}
-
-export class AzureCosmosNoSqlChatStore<
-  AdditionalMessageOptions extends object = object,
-> extends BaseChatStore<AdditionalMessageOptions> {
-  private userId: string;
-  private ttl: number;
-  private cosmosClient: CosmosClient;
-  private database!: Database;
-  private container!: Container;
-  private initPromise?: Promise<void>;
-
-  private dbName: string;
-  private containerName: string;
-  private cosmosContainerProperties: AzureCosmosChatContainerProperties;
-  private cosmosDatabaseProperties: AzureCosmosChatDatabaseProperties;
-  private initialize: () => Promise<void>;
-
-  constructor({
-    cosmosClient,
-    dbName = DEFAULT_CHAT_DATABASE,
-    containerName = DEFAULT_CHAT_CONTAINER,
-    cosmosContainerProperties = { partitionKey: "/userId" },
-    cosmosDatabaseProperties = {},
-    ttlInSeconds = -1,
-  }: AzureCosmosNoSqlChatStoreConfig) {
-    super();
-    if (!cosmosClient) {
-      throw new Error(
-        "CosmosClient is required for AzureCosmosDBNoSQLChatStore initialization",
-      );
-    }
-    this.ttl = ttlInSeconds;
-    this.userId = cosmosContainerProperties.userId || "anonymous";
-    this.cosmosClient = cosmosClient;
-    this.dbName = dbName;
-    this.containerName = containerName;
-    this.cosmosContainerProperties = cosmosContainerProperties;
-    this.cosmosDatabaseProperties = cosmosDatabaseProperties;
-
-    this.initialize = () => {
-      if (this.initPromise === undefined) {
-        this.initPromise = this.init().catch((error) => {
-          console.error(
-            "Error during AzureCosmosDBNoSQLChatStore initialization",
-            error,
-          );
-        });
-      }
-      return this.initPromise;
-    };
-  }
-
-  client(): CosmosClient {
-    return this.cosmosClient;
-  }
-
-  // Asynchronous initialization method to create database and container
-  private async init(): Promise<void> {
-    // Set default throughput if not provided
-    const throughput =
-      this.cosmosDatabaseProperties?.throughput || DEFAULT_OFFER_THROUGHPUT;
-
-    // Create the database if it doesn't exist
-    const { database } = await this.cosmosClient.databases.createIfNotExists({
-      id: this.dbName,
-      throughput,
-    });
-    this.database = database;
-
-    // Create the container if it doesn't exist
-    const { container } = await this.database.containers.createIfNotExists({
-      id: this.containerName,
-      throughput: this.cosmosContainerProperties?.throughput,
-      partitionKey: "/userId",
-      indexingPolicy: this.cosmosContainerProperties?.indexingPolicy,
-      defaultTtl: this.ttl,
-      uniqueKeyPolicy: this.cosmosContainerProperties?.uniqueKeyPolicy,
-      conflictResolutionPolicy:
-        this.cosmosContainerProperties?.conflictResolutionPolicy,
-      computedProperties: this.cosmosContainerProperties?.computedProperties,
-    });
-    this.container = container;
-  }
-  /**
-   * Static method for creating an instance using a connection string.
-   * If no connection string is provided, it will attempt to use the env variable `AZURE_COSMOSDB_NOSQL_CONNECTION_STRING` as connection string.
-   * @returns Instance of AzureCosmosNoSqlKVStore
-   */
-  static fromConnectionString(
-    config: {
-      connectionString?: string;
-    } & AzureCosmosNoSqlChatStoreConfig = {},
-  ): AzureCosmosNoSqlChatStore {
-    const cosmosConnectionString =
-      config.connectionString ||
-      (getEnv("AZURE_COSMOSDB_NOSQL_CONNECTION_STRING") as string);
-    if (!cosmosConnectionString) {
-      throw new Error("Azure CosmosDB connection string must be provided");
-    }
-    const { endpoint, key } = parseConnectionString(cosmosConnectionString);
-    const cosmosClient = new CosmosClient({
-      endpoint,
-      key,
-      userAgentSuffix: USER_AGENT_SUFFIX,
-    });
-    return new AzureCosmosNoSqlChatStore({
-      ...config,
-      cosmosClient,
-    });
-  }
-
-  /**
-   * Static method for creating an instance using a account endpoint and key.
-   * If no endpoint and key  is provided, it will attempt to use the env variable `AZURE_COSMOSDB_NOSQL_ACCOUNT_ENDPOINT` as enpoint and `AZURE_COSMOSDB_NOSQL_ACCOUNT_KEY` as key.
-   * @returns Instance of AzureCosmosNoSqlKVStore
-   */
-  static fromAccountAndKey(
-    config: {
-      endpoint?: string;
-      key?: string;
-    } & AzureCosmosNoSqlChatStoreConfig = {},
-  ): AzureCosmosNoSqlChatStore {
-    const cosmosEndpoint =
-      config.endpoint ||
-      (getEnv("AZURE_COSMOSDB_NOSQL_ACCOUNT_ENDPOINT") as string);
-    const cosmosKey =
-      config.key || (getEnv("AZURE_COSMOSDB_NOSQL_ACCOUNT_KEY") as string);
-
-    if (!cosmosEndpoint || !cosmosKey) {
-      throw new Error(
-        "Azure CosmosDB account endpoint and key must be provided",
-      );
-    }
-    const cosmosClient = new CosmosClient({
-      endpoint: cosmosEndpoint,
-      key: cosmosKey,
-      userAgentSuffix: USER_AGENT_SUFFIX,
-    });
-    return new AzureCosmosNoSqlChatStore({
-      ...config,
-      cosmosClient,
-    });
-  }
-
-  /**
-   * Static method for creating an instance using AAD token.
-   * If no endpoint and credentials are provided, it will attempt to use the env variable `AZURE_COSMOSDB_NOSQL_ACCOUNT_ENDPOINT` as endpoint and use DefaultAzureCredential() as credentials.
-   * @returns Instance of AzureCosmosNoSqlKVStore
-   */
-  static fromAadToken(
-    config: {
-      endpoint?: string;
-      credential?: TokenCredential;
-    } & AzureCosmosNoSqlChatStoreConfig = {},
-  ): AzureCosmosNoSqlChatStore {
-    const cosmosEndpoint =
-      config.endpoint ||
-      (getEnv("AZURE_COSMOSDB_NOSQL_CONNECTION_STRING") as string);
-
-    if (!cosmosEndpoint) {
-      throw new Error("Azure CosmosDB account endpoint must be provided");
-    }
-    const credentials = config.credential ?? new DefaultAzureCredential();
-    const cosmosClient = new CosmosClient({
-      endpoint: cosmosEndpoint,
-      aadCredentials: credentials,
-      userAgentSuffix: USER_AGENT_SUFFIX,
-    });
-    return new AzureCosmosNoSqlChatStore({
-      ...config,
-      cosmosClient,
-    });
-  }
-
-  private convertToChatMessage(
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    message: any,
-  ): ChatMessage<AdditionalMessageOptions> {
-    return {
-      content: message.content as MessageContent,
-      role: message.role as MessageType,
-      options: message.options as AdditionalMessageOptions,
-    } as ChatMessage<AdditionalMessageOptions>;
-  }
-
-  private convertToCosmosMessage(
-    message: ChatMessage<AdditionalMessageOptions>,
-  ): // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  any {
-    return {
-      content: message.content,
-      role: message.role,
-      options: message.options,
-    };
-  }
-
-  /**
-   * Set messages for a given key.
-   */
-  async setMessages(
-    key: string,
-    messages: ChatMessage<AdditionalMessageOptions>[],
-  ): Promise<void> {
-    await this.initialize();
-    const inputMessages = messages.map(this.convertToCosmosMessage);
-    await this.container.items.upsert({
-      id: key,
-      messages: inputMessages,
-      userId: this.userId,
-    });
-  }
-
-  /**
-   * Get messages for a given key.
-   */
-  async getMessages(
-    key: string,
-  ): Promise<ChatMessage<AdditionalMessageOptions>[]> {
-    await this.initialize();
-    const res = await this.container.item(key, this.userId).read();
-    const messageHistory = res?.resource?.messages ?? [];
-    const result = messageHistory.map(this.convertToChatMessage);
-    return result;
-  }
-
-  /**
-   * Add a message for a given key.
-   */
-  async addMessage(
-    key: string,
-    message: ChatMessage<AdditionalMessageOptions>,
-    idx?: number,
-  ): Promise<void> {
-    await this.initialize();
-    const res = await this.container.item(key, this.userId).read();
-    const messageHistory = res?.resource?.messages ?? [];
-    if (idx === undefined) {
-      messageHistory.push(this.convertToCosmosMessage(message));
-    } else {
-      messageHistory.splice(idx, 0, this.convertToCosmosMessage(message));
-    }
-    await this.setMessages(key, messageHistory);
-  }
-
-  /**
-   * Deletes all messages for a given key.
-   */
-  async deleteMessages(key: string): Promise<void> {
-    await this.initialize();
-    try {
-      await this.container.item(key, this.userId).delete();
-      // eslint-disable-next-line no-empty
-    } catch (e) {}
-  }
-
-  /**
-   * Deletes one message at idx index for a given key.
-   */
-  async deleteMessage(key: string, idx: number): Promise<void> {
-    await this.initialize();
-    const res = await this.container.item(key, this.userId).read();
-    const messageHistory = res?.resource?.messages ?? [];
-    if (idx >= 0 && idx < messageHistory.length) {
-      messageHistory.splice(idx, 1);
-      await this.setMessages(key, messageHistory);
-    }
-  }
-
-  /**
-   * Get all keys.
-   */
-  async getKeys(): Promise<IterableIterator<string>> {
-    await this.initialize();
-    const result = await this.container.items
-      .query("Select c.id from c")
-      .fetchAll();
-    const keys = result.resources.map((res: { id: string }) => res.id);
-
-    function* keyGenerator(): IterableIterator<string> {
-      for (const key of keys) {
-        yield key;
-      }
-    }
-    return keyGenerator();
-  }
-}

packages/llamaindex/src/storage/index.ts

@@ -2,7 +2,6 @@ export * from "@llamaindex/core/storage/chat-store";
 export * from "@llamaindex/core/storage/doc-store";
 export * from "@llamaindex/core/storage/index-store";
 export * from "@llamaindex/core/storage/kv-store";
-export * from "./chatStore/AzureCosmosNoSqlChatStore.js";
 export * from "./docStore/AzureCosmosNoSqlDocumentStore.js";
 export { PostgresDocumentStore } from "./docStore/PostgresDocumentStore.js";
 export { SimpleDocumentStore } from "./docStore/SimpleDocumentStore.js";

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 AzureCosmosDBNoSQLKVStore initialization",
+        "CosmosClient is required for AzureCosmosDBNoSQLVectorStore initialization",
       );
     }
     this.cosmosClient = cosmosClient;

packages/llamaindex/src/vector-store/AzureCosmosDBNoSqlVectorStore.ts

@@ -55,18 +55,8 @@ export interface AzureCosmosDBNoSQLConfig
   readonly flatMetadata?: boolean;
   readonly idKey?: string;
 }
-/**
- * Query options for the `AzureCosmosDBNoSQLVectorStore.query` method.
- * @property includeEmbeddings - Whether to include the embeddings in the result. Default false
- * @property includeVectorDistance - Whether to include the vector distance in the result. Default true
- * @property whereClause - The where clause to use in the query. While writing this clause, use `c` as the alias for the container and do not include the `WHERE` keyword.
- */
-export interface AzureCosmosQueryOptions {
-  includeVectorDistance?: boolean;
-  whereClause?: string;
-}
 
-const USER_AGENT_SUFFIX = "llamaindex-cdbnosql-vectorstore-javascript";
+const USER_AGENT_SUFFIX = "LlamaIndex-CDBNoSQL-VectorStore-JavaScript";
 
 const DEFAULT_VECTOR_EMBEDDING_POLICY = {
   vectorEmbeddings: [
@@ -108,22 +98,6 @@ function parseConnectionString(connectionString: string): {
 
   return { endpoint, key: accountKey };
 }
-/**
- * utility function to build the query string for the CosmosDB query
- */
-function queryBuilder(options: AzureCosmosQueryOptions): string {
-  let initialQuery =
-    "SELECT TOP @k c[@id] as id, c[@text] as text, c[@metadata] as metadata";
-  if (options.includeVectorDistance !== false) {
-    initialQuery += `, VectorDistance(c[@embeddingKey],@embedding) AS SimilarityScore`;
-  }
-  initialQuery += ` FROM c`;
-  if (options.whereClause) {
-    initialQuery += ` WHERE ${options.whereClause}`;
-  }
-  initialQuery += ` ORDER BY VectorDistance(c[@embeddingKey],@embedding)`;
-  return initialQuery;
-}
 
 export class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
   storesText: boolean = true;
@@ -360,25 +334,21 @@ export class AzureCosmosDBNoSqlVectorStore extends BaseVectorStore {
    */
   async query(
     query: VectorStoreQuery,
-    options: AzureCosmosQueryOptions = {},
+    options?: object,
   ): Promise<VectorStoreQueryResult> {
     await this.initialize();
-    if (!query.queryEmbedding || query.queryEmbedding.length === 0) {
-      throw new Error(
-        "queryEmbedding is required for AzureCosmosDBNoSqlVectorStore query",
-      );
-    }
     const params = {
       vector: query.queryEmbedding!,
       k: query.similarityTopK,
     };
-    const builtQuery = queryBuilder(options);
+
     const nodes: BaseNode[] = [];
     const ids: string[] = [];
     const similarities: number[] = [];
     const queryResults = await this.container.items
       .query({
-        query: builtQuery,
+        query:
+          "SELECT TOP @k c[@id] as id, c[@text] as text, c[@metadata] as metadata, VectorDistance(c[@embeddingKey],@embedding) AS SimilarityScore FROM c ORDER BY VectorDistance(c[@embeddingKey],@embedding)",
         parameters: [
           { name: "@k", value: params.k },
           { name: "@id", value: this.idKey },

packages/llamaindex/tests/vector-stores/AzureCosmosDBNoSqlVectorStore.int.test.ts

@@ -14,10 +14,9 @@ import {
   Settings,
   VectorStoreQueryMode,
   type AzureCosmosDBNoSQLConfig,
-  type AzureCosmosQueryOptions,
   type VectorStoreQueryResult,
 } from "llamaindex";
-import { beforeAll, describe, expect, it } from "vitest";
+import { beforeEach, describe, expect, it } from "vitest";
 dotenv.config();
 /*
  * To run this test, you need have an Azure Cosmos DB for NoSQL instance
@@ -65,10 +64,7 @@ Settings.llm = new OpenAI(llmInit);
 Settings.embedModel = new OpenAIEmbedding(embedModelInit);
 // This test is skipped because it requires an Azure Cosmos DB instance and OpenAI API keys
 describe.skip("AzureCosmosDBNoSQLVectorStore", () => {
-  let vectorStore: AzureCosmosDBNoSqlVectorStore;
-  // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  let embeddings: any = [];
-  beforeAll(async () => {
+  beforeEach(async () => {
     if (process.env.AZURE_COSMOSDB_NOSQL_CONNECTION_STRING) {
       client = new CosmosClient(
         process.env.AZURE_COSMOSDB_NOSQL_CONNECTION_STRING,
@@ -83,12 +79,15 @@ describe.skip("AzureCosmosDBNoSQLVectorStore", () => {
         "Please set the environment variable AZURE_COSMOSDB_NOSQL_CONNECTION_STRING or AZURE_COSMOSDB_NOSQL_ENDPOINT",
       );
     }
+
     // Make sure the database does not exists
     try {
       await client.database(DATABASE_NAME).delete();
     } catch {
       // Ignore error if the database does not exist
     }
+  });
+  it("perform query", async () => {
     const config: AzureCosmosDBNoSQLConfig = {
       idKey: "name",
       textKey: "customText",
@@ -135,9 +134,9 @@ describe.skip("AzureCosmosDBNoSQLVectorStore", () => {
       },
     };
 
-    vectorStore = new AzureCosmosDBNoSqlVectorStore(config);
+    const vectorStore = new AzureCosmosDBNoSqlVectorStore(config);
 
-    embeddings = await Settings.embedModel.getTextEmbeddings([
+    const embeddings = await Settings.embedModel.getTextEmbeddings([
       "This book is about politics",
       "Cats sleeps a lot.",
       "Sandwiches taste good.",
@@ -151,29 +150,28 @@ describe.skip("AzureCosmosDBNoSQLVectorStore", () => {
         id_: "1",
         text: "This book is about politics",
         embedding: embeddings[0],
-        metadata: { key: "politics", number: 1 },
+        metadata: { key: "politics" },
       }),
       new Document({
         id_: "2",
         text: "Cats sleeps a lot.",
         embedding: embeddings[1],
-        metadata: { key: "cats", number: 2 },
+        metadata: { key: "cats" },
       }),
       new Document({
         id_: "3",
         text: "Sandwiches taste good.",
         embedding: embeddings[2],
-        metadata: { key: "sandwiches", number: 3 },
+        metadata: { key: "sandwiches" },
       }),
       new Document({
         id_: "4",
         text: "The house is open",
         embedding: embeddings[3],
-        metadata: { key: "house", number: 4 },
+        metadata: { key: "house" },
       }),
     ]);
-  });
-  it("perform query", async () => {
+
     const results: VectorStoreQueryResult = await vectorStore.query({
       queryEmbedding: embeddings[4] || [],
       similarityTopK: 1,
@@ -181,62 +179,5 @@ describe.skip("AzureCosmosDBNoSQLVectorStore", () => {
     });
     expect(results.ids.length).toEqual(1);
     expect(results.ids[0]).toEqual("3");
-    expect(results.similarities).toBeDefined();
-    expect(results.similarities[0]).toBeDefined();
-  }, 1000000);
-
-  it("perform query with where clause", async () => {
-    const options: AzureCosmosQueryOptions = {
-      whereClause: "c.customMetadata.number > 3",
-    };
-    const results: VectorStoreQueryResult = await vectorStore.query(
-      {
-        queryEmbedding: embeddings[4] || [],
-        similarityTopK: 1,
-        mode: VectorStoreQueryMode.DEFAULT,
-      },
-      options,
-    );
-    expect(results.ids.length).toEqual(1);
-    expect(results.ids[0]).toEqual("4");
-    expect(results.similarities).toBeDefined();
-    expect(results.similarities[0]).toBeDefined();
-  }, 1000000);
-
-  it("perform query with includeVectorDistance false", async () => {
-    const options: AzureCosmosQueryOptions = {
-      includeVectorDistance: false,
-    };
-    const results: VectorStoreQueryResult = await vectorStore.query(
-      {
-        queryEmbedding: embeddings[4] || [],
-        similarityTopK: 1,
-        mode: VectorStoreQueryMode.DEFAULT,
-      },
-      options,
-    );
-    expect(results.ids.length).toEqual(1);
-    expect(results.ids[0]).toEqual("3");
-    expect(results.similarities).toBeDefined();
-    expect(results.similarities[0]).toBeUndefined();
-  }, 1000000);
-
-  it("perform query with includeVectorDistance false and whereClause", async () => {
-    const options: AzureCosmosQueryOptions = {
-      includeVectorDistance: false,
-      whereClause: "c.customMetadata.number > 3",
-    };
-    const results: VectorStoreQueryResult = await vectorStore.query(
-      {
-        queryEmbedding: embeddings[4] || [],
-        similarityTopK: 1,
-        mode: VectorStoreQueryMode.DEFAULT,
-      },
-      options,
-    );
-    expect(results.ids.length).toEqual(1);
-    expect(results.ids[0]).toEqual("4");
-    expect(results.similarities).toBeDefined();
-    expect(results.similarities[0]).toBeUndefined();
   }, 1000000);
 });

packages/llamaindex/tests/vector-stores/AzureCosmosDBNoSqlVectorStore.test.ts

@@ -1,7 +1,6 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import type { BaseNode } from "@llamaindex/core/schema";
 import { beforeEach, describe, expect, it, vi } from "vitest";
-import { VectorStoreQueryMode } from "../../src/vector-store.js";
 import { TestableAzureCosmosDBNoSqlVectorStore } from "../mocks/TestableAzureCosmosDBNoSqlVectorStore.js";
 import { createMockClient } from "../utility/mockCosmosClient.js"; // Import the mock client
 
@@ -96,27 +95,4 @@ describe("AzureCosmosDBNoSqlVectorStore Tests", () => {
     expect(client.databases.containers.items.create).toHaveBeenCalledTimes(2);
     expect(result).toEqual(["node-0", "node-1"]);
   });
-
-  it("should throw error if no query embedding is provided", async () => {
-    const client = createMockClient();
-    const store = new TestableAzureCosmosDBNoSqlVectorStore({
-      client: client as any,
-      endpoint: "https://example.com",
-      idKey: "id",
-      textKey: "text",
-      metadataKey: "metadata",
-    });
-
-    expect(store).toBeDefined();
-
-    await expect(
-      store.query({
-        queryEmbedding: [],
-        similarityTopK: 4,
-        mode: VectorStoreQueryMode.DEFAULT,
-      }),
-    ).rejects.toThrowError(
-      "queryEmbedding is required for AzureCosmosDBNoSqlVectorStore query",
-    );
-  });
 });

packages/node-parser/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/node-parser
 
-## 0.0.12
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 0.0.11
 
 ### Patch Changes

packages/node-parser/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@llamaindex/node-parser",
-  "version": "0.0.12",
+  "version": "0.0.11",
   "description": "Node parser for LlamaIndex",
   "type": "module",
   "exports": {

packages/providers/anthropic/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/anthropic
 
-## 0.0.20
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 0.0.19
 
 ### Patch Changes

packages/providers/anthropic/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/anthropic",
   "description": "Anthropic Adapter for LlamaIndex",
-  "version": "0.0.20",
+  "version": "0.0.19",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/providers/clip/CHANGELOG.md

@@ -1,13 +1,5 @@
 # @llamaindex/clip
 
-## 0.0.20
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-  - @llamaindex/openai@0.1.36
-
 ## 0.0.19
 
 ### Patch Changes

packages/providers/clip/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/clip",
   "description": "Clip Embedding Adapter for LlamaIndex",
-  "version": "0.0.20",
+  "version": "0.0.19",
   "type": "module",
   "types": "dist/index.d.ts",
   "main": "dist/index.cjs",

packages/providers/deepinfra/CHANGELOG.md

@@ -1,13 +1,5 @@
 # @llamaindex/deepinfra
 
-## 0.0.20
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-  - @llamaindex/openai@0.1.36
-
 ## 0.0.19
 
 ### Patch Changes

packages/providers/deepinfra/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/deepinfra",
   "description": "Deepinfra Adapter for LlamaIndex",
-  "version": "0.0.20",
+  "version": "0.0.19",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/providers/groq/CHANGELOG.md

@@ -1,11 +1,5 @@
 # @llamaindex/groq
 
-## 0.0.35
-
-### Patch Changes
-
-- @llamaindex/openai@0.1.36
-
 ## 0.0.34
 
 ### Patch Changes

packages/providers/groq/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/groq",
   "description": "Groq Adapter for LlamaIndex",
-  "version": "0.0.35",
+  "version": "0.0.34",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/providers/huggingface/CHANGELOG.md

@@ -1,13 +1,5 @@
 # @llamaindex/huggingface
 
-## 0.0.20
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-  - @llamaindex/openai@0.1.36
-
 ## 0.0.19
 
 ### Patch Changes

packages/providers/huggingface/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/huggingface",
   "description": "Huggingface Adapter for LlamaIndex",
-  "version": "0.0.20",
+  "version": "0.0.19",
   "type": "module",
   "types": "dist/index.d.ts",
   "main": "dist/index.cjs",

packages/providers/ollama/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/ollama
 
-## 0.0.27
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 0.0.26
 
 ### Patch Changes

packages/providers/ollama/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/ollama",
   "description": "Ollama Adapter for LlamaIndex",
-  "version": "0.0.27",
+  "version": "0.0.26",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/providers/openai/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/openai
 
-## 0.1.36
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 0.1.35
 
 ### Patch Changes

packages/providers/openai/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/openai",
   "description": "OpenAI Adapter for LlamaIndex",
-  "version": "0.1.36",
+  "version": "0.1.35",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/providers/portkey-ai/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/portkey-ai
 
-## 0.0.20
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 0.0.19
 
 ### Patch Changes

packages/providers/portkey-ai/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/portkey-ai",
   "description": "Portkey Adapter for LlamaIndex",
-  "version": "0.0.20",
+  "version": "0.0.19",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/providers/replicate/CHANGELOG.md

@@ -1,12 +1,5 @@
 # @llamaindex/replicate
 
-## 0.0.20
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
 ## 0.0.19
 
 ### Patch Changes

packages/providers/replicate/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/replicate",
   "description": "Replicate Adapter for LlamaIndex",
-  "version": "0.0.20",
+  "version": "0.0.19",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/providers/vllm/CHANGELOG.md

@@ -1,11 +1,5 @@
 # @llamaindex/vllm
 
-## 0.0.6
-
-### Patch Changes
-
-- @llamaindex/openai@0.1.36
-
 ## 0.0.5
 
 ### Patch Changes

packages/providers/vllm/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/vllm",
   "description": "vLLM Adapter for LlamaIndex",
-  "version": "0.0.6",
+  "version": "0.0.5",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

packages/readers/CHANGELOG.md

@@ -1,18 +1,5 @@
 # @llamaindex/readers
 
-## 1.0.13
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-
-## 1.0.12
-
-### Patch Changes
-
-- e60328b: fix: import error in cjs
-
 ## 1.0.11
 
 ### Patch Changes

packages/readers/package.json

@@ -1,12 +1,20 @@
 {
   "name": "@llamaindex/readers",
   "description": "LlamaIndex Readers",
-  "version": "1.0.13",
+  "version": "1.0.11",
   "type": "module",
   "exports": {
     "./node/hook": "./node/dist/hook.js",
     "./node": "./node/dist/index.js",
     "./assembly-ai": {
+      "edge-light": {
+        "types": "./assembly-ai/dist/index.edge-light.d.ts",
+        "default": "./assembly-ai/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./assembly-ai/dist/index.workerd.d.ts",
+        "default": "./assembly-ai/dist/index.workerd.js"
+      },
       "require": {
         "types": "./assembly-ai/dist/index.d.cts",
         "default": "./assembly-ai/dist/index.cjs"
@@ -17,6 +25,14 @@
       }
     },
     "./cosmosdb": {
+      "edge-light": {
+        "types": "./cosmosdb/dist/index.edge-light.d.ts",
+        "default": "./cosmosdb/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./cosmosdb/dist/index.workerd.d.ts",
+        "default": "./cosmosdb/dist/index.workerd.js"
+      },
       "require": {
         "types": "./cosmosdb/dist/index.d.cts",
         "default": "./cosmosdb/dist/index.cjs"
@@ -63,6 +79,14 @@
       }
     },
     "./discord": {
+      "edge-light": {
+        "types": "./discord/dist/index.edge-light.d.ts",
+        "default": "./discord/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./discord/dist/index.workerd.d.ts",
+        "default": "./discord/dist/index.workerd.js"
+      },
       "require": {
         "types": "./discord/dist/index.d.cts",
         "default": "./discord/dist/index.cjs"
@@ -73,6 +97,14 @@
       }
     },
     "./docx": {
+      "edge-light": {
+        "types": "./docx/dist/index.edge-light.d.ts",
+        "default": "./docx/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./docx/dist/index.workerd.d.ts",
+        "default": "./docx/dist/index.workerd.js"
+      },
       "require": {
         "types": "./docx/dist/index.d.cts",
         "default": "./docx/dist/index.cjs"
@@ -83,6 +115,14 @@
       }
     },
     "./html": {
+      "edge-light": {
+        "types": "./html/dist/index.edge-light.d.ts",
+        "default": "./html/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./html/dist/index.workerd.d.ts",
+        "default": "./html/dist/index.workerd.js"
+      },
       "require": {
         "types": "./html/dist/index.d.cts",
         "default": "./html/dist/index.cjs"
@@ -93,6 +133,14 @@
       }
     },
     "./image": {
+      "edge-light": {
+        "types": "./image/dist/index.edge-light.d.ts",
+        "default": "./image/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./image/dist/index.workerd.d.ts",
+        "default": "./image/dist/index.workerd.js"
+      },
       "require": {
         "types": "./image/dist/index.d.cts",
         "default": "./image/dist/index.cjs"
@@ -103,6 +151,14 @@
       }
     },
     "./json": {
+      "edge-light": {
+        "types": "./json/dist/index.edge-light.d.ts",
+        "default": "./json/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./json/dist/index.workerd.d.ts",
+        "default": "./json/dist/index.workerd.js"
+      },
       "require": {
         "types": "./json/dist/index.d.cts",
         "default": "./json/dist/index.cjs"
@@ -113,6 +169,14 @@
       }
     },
     "./markdown": {
+      "edge-light": {
+        "types": "./markdown/dist/index.edge-light.d.ts",
+        "default": "./markdown/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./markdown/dist/index.workerd.d.ts",
+        "default": "./markdown/dist/index.workerd.js"
+      },
       "require": {
         "types": "./markdown/dist/index.d.cts",
         "default": "./markdown/dist/index.cjs"
@@ -123,6 +187,14 @@
       }
     },
     "./mongo": {
+      "edge-light": {
+        "types": "./mongo/dist/index.edge-light.d.ts",
+        "default": "./mongo/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./mongo/dist/index.workerd.d.ts",
+        "default": "./mongo/dist/index.workerd.js"
+      },
       "require": {
         "types": "./mongo/dist/index.d.cts",
         "default": "./mongo/dist/index.cjs"
@@ -133,6 +205,14 @@
       }
     },
     "./notion": {
+      "edge-light": {
+        "types": "./notion/dist/index.edge-light.d.ts",
+        "default": "./notion/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./notion/dist/index.workerd.d.ts",
+        "default": "./notion/dist/index.workerd.js"
+      },
       "require": {
         "types": "./notion/dist/index.d.cts",
         "default": "./notion/dist/index.cjs"
@@ -143,6 +223,14 @@
       }
     },
     "./pdf": {
+      "edge-light": {
+        "types": "./pdf/dist/index.edge-light.d.ts",
+        "default": "./pdf/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./pdf/dist/index.workerd.d.ts",
+        "default": "./pdf/dist/index.workerd.js"
+      },
       "require": {
         "types": "./pdf/dist/index.d.cts",
         "default": "./pdf/dist/index.cjs"
@@ -153,6 +241,14 @@
       }
     },
     "./text": {
+      "edge-light": {
+        "types": "./text/dist/index.edge-light.d.ts",
+        "default": "./text/dist/index.edge-light.js"
+      },
+      "workerd": {
+        "types": "./text/dist/index.workerd.d.ts",
+        "default": "./text/dist/index.workerd.js"
+      },
       "require": {
         "types": "./text/dist/index.d.cts",
         "default": "./text/dist/index.cjs"

unit/CHANGELOG.md

@@ -1,40 +1,5 @@
 # @llamaindex/unit-test
 
-## 0.0.27
-
-### Patch Changes
-
-- Updated dependencies [6d22fa2]
-  - @llamaindex/core@0.4.11
-  - @llamaindex/cloud@2.0.11
-  - llamaindex@0.8.17
-  - @llamaindex/node-parser@0.0.12
-  - @llamaindex/openai@0.1.36
-  - @llamaindex/readers@1.0.13
-
-## 0.0.26
-
-### Patch Changes
-
-- Updated dependencies [e60328b]
-  - @llamaindex/readers@1.0.12
-  - llamaindex@0.8.16
-
-## 0.0.25
-
-### Patch Changes
-
-- Updated dependencies [3d503cb]
-- Updated dependencies [5dae534]
-  - llamaindex@0.8.15
-
-## 0.0.24
-
-### Patch Changes
-
-- Updated dependencies [630b425]
-  - llamaindex@0.8.14
-
 ## 0.0.23
 
 ### Patch Changes

unit/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@llamaindex/unit-test",
   "private": true,
-  "version": "0.0.27",
+  "version": "0.0.23",
   "type": "module",
   "scripts": {
     "test": "vitest run"