fix: query tool

Author: marcusschiesser

.changeset/olive-readers-itch.md

@@ -0,0 +1,6 @@
+---
+"llamaindex": patch
+"@llamaindex/examples": patch
+---
+
+fix undefined values in querytool

examples/agents/agent/query-tool.ts

@@ -1,12 +1,15 @@
-import { openai } from "@llamaindex/openai";
+import { OpenAIEmbedding, openai } from "@llamaindex/openai";
 import {
   agent,
   agentStreamEvent,
   agentToolCallResultEvent,
 } from "@llamaindex/workflow";
-import { Document, VectorStoreIndex } from "llamaindex";
+import { Document, Settings, VectorStoreIndex } from "llamaindex";
 
 async function main() {
+  Settings.embedModel = new OpenAIEmbedding();
+  Settings.llm = openai({ model: "gpt-4o" });
+
   const index = await VectorStoreIndex.fromDocuments([
     new Document({
       text: "Cats have a specialized collarbone that allows them to always land on their feet when they fall.",
@@ -20,7 +23,6 @@ async function main() {
   ]);
 
   const myAgent = agent({
-    llm: openai({ model: "gpt-4o" }),
     tools: [
       index.queryTool({
         options: { similarityTopK: 2 },

examples/pnpm-workspace.yaml

@@ -54,9 +54,11 @@ export class QueryEngineTool implements BaseTool<QueryEngineParam> {
       return { content: response.message.content } as unknown as JSONValue;
     }
 
+    // Use JSON.parse(JSON.stringify()) to remove undefined values from sourceNodes
+    // since undefined is not a valid JSONValue
     return {
       content: response.message.content,
-      sourceNodes: response.sourceNodes,
+      sourceNodes: JSON.parse(JSON.stringify(response.sourceNodes ?? [])),
     } as unknown as JSONValue;
   }
 }