15 lines
478 B
TypeScript
15 lines
478 B
TypeScript
import { tool } from "ai";
|
|
import { z } from "zod";
|
|
import { searchPinecone } from "@/lib/pinecone";
|
|
|
|
export const vectorDatabaseSearch = tool({
|
|
description: 'Search the vector database for information',
|
|
inputSchema: z.object({
|
|
query: z.string().describe('The query to search the vector database for. Optimally is a hypothetical answer for similarity search.'),
|
|
}),
|
|
execute: async ({ query }) => {
|
|
return await searchPinecone(query);
|
|
},
|
|
});
|
|
|