wip

Author: moonmeister

package.json

@@ -60,7 +60,8 @@
 		"shiki": "^3.6.0",
 		"strip-markdown": "^6.0.0",
 		"unified": "^11.0.5",
-		"vfile-matter": "^5.0.1"
+		"vfile-matter": "^5.0.1",
+		"zod": "^3.25.76"
 	},
 	"devDependencies": {
 		"@tailwindcss/postcss": "^4.1.9",

pnpm-lock.yaml

@@ -0,0 +1,61 @@
+import { tool } from "ai";
+import { z } from "zod";
+import { getContext } from "./smart-search.mjs";
+
+export const smartSearchTool = tool({
+	description:
+		"Search for information about Faust using WP Engine Smart Search. Use this to answer questions about Faust, its features, capabilities, and more when the information is not already known.",
+	parameters: z.object({
+		query: z
+			.string()
+			.describe(
+				"The search query to find relevant Faust information based on the user's question.",
+			),
+	}),
+	execute: async ({ query }) => {
+		console.log(`[Tool Execution] Searching with query: "${query}"`);
+		try {
+			const context = await getContext(query);
+
+			if (context.errors && context.errors.length > 0) {
+				console.error(
+					"[Tool Execution] Error fetching context:",
+					context.errors,
+				);
+				// Return a structured error message that the LLM can understand
+				return {
+					error: `Error fetching context: ${context.errors[0].message}`,
+				};
+			}
+
+			if (context?.data?.similarity?.docs?.length === 0) {
+				console.log("[Tool Execution] No documents found for query:", query);
+				return {
+					searchResults: "No relevant information found for your query.",
+				};
+			}
+
+			const formattedResults = context.data.similarity.docs.map((doc) => {
+				if (!doc) {
+					return {};
+				}
+
+				return {
+					id: doc.id,
+					title: doc.data.post_title,
+					content: doc.data.post_content,
+					url: doc.data.post_url,
+					categories: doc.data.categories.map((category) => category.name),
+					searchScore: doc.score,
+				};
+			});
+
+			// console.log("[Tool Execution] Search results:", formattedResults);
+
+			return { searchResults: formattedResults }; // Return the formatted string
+		} catch (error) {
+			console.error("[Tool Execution] Exception:", error);
+			return { error: `An error occurred while searching: ${error.message}` };
+		}
+	},
+});

src/lib/rag.mjs

@@ -5,7 +5,7 @@ const token = env.NEXT_SEARCH_ACCESS_TOKEN ?? "";
 
 // Field might need to be adjusted to include the correct field for your search index, e.g. "content", "post_title", etc.
 export async function getContext(message) {
-	const query = `query GetRagContext($message: String!) {
+	const query = `query GetContext($message: String!) {
     similarity(
       input: {
         nearest: {