Merge pull request #237 from kuizuo/main

feat: enhance tools to use 'tool' from 'ai' package

lib/agents/tools/retrieve.tsx

@@ -1,13 +1,14 @@
+import { tool } from 'ai'
 import { retrieveSchema } from '@/lib/schema/retrieve'
 import { ToolProps } from '.'
 import { SearchSkeleton } from '@/components/search-skeleton'
 import { SearchResults as SearchResultsType } from '@/lib/types'
 import RetrieveSection from '@/components/retrieve-section'
 
-export const retrieveTool = ({ uiStream, fullResponse }: ToolProps) => ({
+export const retrieveTool = ({ uiStream, fullResponse }: ToolProps) => tool({
   description: 'Retrieve content from the web',
   parameters: retrieveSchema,
-  execute: async ({ url }: { url: string }) => {
+  execute: async ({ url }) => {
     let hasError = false
     // Append the search section
     uiStream.append(<SearchSkeleton />)

lib/agents/tools/search.tsx

@@ -1,10 +1,11 @@
+import { tool } from 'ai'
 import { createStreamableValue } from 'ai/rsc'
 import Exa from 'exa-js'
 import { searchSchema } from '@/lib/schema/search'
 import { SearchSection } from '@/components/search-section'
 import { ToolProps } from '.'
 
-export const searchTool = ({ uiStream, fullResponse }: ToolProps) => ({
+export const searchTool = ({ uiStream, fullResponse }: ToolProps) => tool({
   description: 'Search the web for information',
   parameters: searchSchema,
   execute: async ({
@@ -13,12 +14,6 @@ export const searchTool = ({ uiStream, fullResponse }: ToolProps) => ({
     search_depth,
     include_domains,
     exclude_domains
-  }: {
-    query: string
-    max_results: number
-    search_depth: 'basic' | 'advanced'
-    include_domains: string[]
-    exclude_domains: string[]
   }) => {
     let hasError = false
     // Append the search section

lib/agents/tools/video-search.tsx

@@ -1,13 +1,14 @@
+import { tool } from 'ai'
 import { createStreamableValue } from 'ai/rsc'
 import { searchSchema } from '@/lib/schema/search'
 import { ToolProps } from '.'
 import { VideoSearchSection } from '@/components/video-search-section'
 
 // Start Generation Here
-export const videoSearchTool = ({ uiStream, fullResponse }: ToolProps) => ({
+export const videoSearchTool = ({ uiStream, fullResponse }: ToolProps) => tool({
   description: 'Search for videos from YouTube',
   parameters: searchSchema,
-  execute: async ({ query }: { query: string }) => {
+  execute: async ({ query }) => {
     let hasError = false
     // Append the search section
     const streamResults = createStreamableValue<string>()

package.json

@@ -46,7 +46,8 @@
     "remark-math": "^6.0.0",
     "sonner": "^1.4.41",
     "tailwind-merge": "^2.2.2",
-    "tailwindcss-animate": "^1.0.7"
+    "tailwindcss-animate": "^1.0.7",
+    "zod": "^3.23.8"
   },
   "devDependencies": {
     "typescript": "^5",