@youdotcom-oss/ai-sdk-plugin 0.1.0

package/package.json

@@ -0,0 +1,80 @@
+{
+  "name": "@youdotcom-oss/ai-sdk-plugin",
+  "version": "0.1.0",
+  "description": "Vercel AI SDK plugin for You.com web search, AI agents, and content extraction via MCP",
+  "license": "MIT",
+  "engines": {
+    "node": ">=18",
+    "bun": ">= 1.2.21"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/youdotcom-oss/dx-toolkit.git",
+    "directory": "packages/ai-sdk-plugin"
+  },
+  "bugs": {
+    "url": "https://github.com/youdotcom-oss/dx-toolkit/issues"
+  },
+  "homepage": "https://github.com/youdotcom-oss/dx-toolkit/tree/main/packages/ai-sdk-plugin#readme",
+  "author": "You.com (https://you.com)",
+  "keywords": [
+    "ai",
+    "sdk",
+    "plugin",
+    "you",
+    "search",
+    "mcp",
+    "vercel",
+    "model-context-protocol",
+    "ai-agents",
+    "web-search"
+  ],
+  "type": "module",
+  "main": "./dist/main.js",
+  "exports": {
+    ".": {
+      "types": "./dist/main.d.ts",
+      "default": "./dist/main.js"
+    },
+    "./templates/generate-text": "./templates/generate-text.ts",
+    "./templates/streaming-text": "./templates/streaming-text.ts"
+  },
+  "files": [
+    "dist",
+    "templates"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "bun run build:bundle && bun run build:types",
+    "build:bundle": "bun build src/main.ts --outdir dist --target node --external ai",
+    "build:types": "tsc --project tsconfig.build.json  --declaration --emitDeclarationOnly --noEmit false",
+    "check": "bun run check:biome && bun run check:types && bun run check:package",
+    "check:biome": "biome check",
+    "check:package": "format-package --check",
+    "check:types": "tsc --noEmit",
+    "check:write": "bun run format && bun run lint:fix && bun run format:package",
+    "example": "bun scripts/run-example.ts",
+    "format": "biome format --write",
+    "format:check": "biome format",
+    "format:package": "format-package --write",
+    "lint": "biome lint",
+    "lint:fix": "biome lint --write",
+    "prepublishOnly": "bun run build",
+    "test": "bun test",
+    "test:coverage": "bun test --coverage",
+    "test:coverage:watch": "bun test --coverage --watch",
+    "test:watch": "bun test --watch"
+  },
+  "types": "./dist/main.d.ts",
+  "dependencies": {
+    "@youdotcom-oss/mcp": "1.3.9"
+  },
+  "peerDependencies": {
+    "ai": "^5.0.0"
+  },
+  "devDependencies": {
+    "@ai-sdk/anthropic": "^2.0.56"
+  }
+}

package/templates/generate-text.ts

@@ -0,0 +1,78 @@
+/**
+ * AI SDK Integration Template - generateText()
+ *
+ * Shows how to integrate You.com tools with Vercel AI SDK's generateText().
+ *
+ * Key Integration Points:
+ * 1. Import tools: youSearch(), youExpress(), youContents()
+ * 2. Add to tools object
+ * 3. API key: Use env variable or pass directly to each tool
+ * 4. Model handles everything - you just get result.text
+ */
+
+import { createAnthropic } from '@ai-sdk/anthropic';
+import { youContents, youExpress, youSearch } from '@youdotcom-oss/ai-sdk-plugin';
+import { generateText } from 'ai';
+
+// ============================================================================
+// INTEGRATION STEP 1: Environment Variables
+// ============================================================================
+// Option A: Use environment variables (recommended)
+// Set YDC_API_KEY in your .env file - tools read it automatically
+
+if (!process.env.YDC_API_KEY) {
+  throw new Error('YDC_API_KEY environment variable is required');
+}
+
+// Your AI provider key (Anthropic, OpenAI, etc.)
+if (!process.env.ANTHROPIC_API_KEY) {
+  throw new Error('ANTHROPIC_API_KEY environment variable is required');
+}
+
+// ============================================================================
+// INTEGRATION STEP 2: Add Tools to generateText
+// ============================================================================
+
+const anthropic = createAnthropic();
+
+// Single tool example
+const result = await generateText({
+  model: anthropic('claude-sonnet-4-5-20250929'),
+  tools: {
+    search: youSearch(), // Reads YDC_API_KEY from environment
+  },
+  prompt: 'Your dynamic prompt here', // Replace with your actual prompt
+});
+
+console.log(result.text); // Model-formatted response with search results
+
+// ============================================================================
+// Multiple Tools (model chooses which to use)
+// ============================================================================
+
+const multiToolResult = await generateText({
+  model: anthropic('claude-sonnet-4-5-20250929'),
+  tools: {
+    search: youSearch(),
+    agent: youExpress(),
+    extract: youContents(),
+  },
+  prompt: 'Your prompt here',
+});
+
+console.log(multiToolResult.text);
+
+// ============================================================================
+// INTEGRATION OPTION B: Pass API Key Directly
+// ============================================================================
+// Override environment variable per tool if needed
+
+const customKeyResult = await generateText({
+  model: anthropic('claude-sonnet-4-5-20250929'),
+  tools: {
+    search: youSearch({ apiKey: 'your-key-here' }),
+  },
+  prompt: 'Your prompt here',
+});
+
+console.log(customKeyResult.text);

package/templates/streaming-text.ts

@@ -0,0 +1,123 @@
+/**
+ * AI SDK Integration Template - streamText()
+ *
+ * Shows how to integrate You.com tools with Vercel AI SDK's streamText().
+ *
+ * Key Integration Points:
+ * 1. Import tools: youSearch(), youExpress(), youContents()
+ * 2. Add to tools object
+ * 3. IMPORTANT: Use stopWhen with stepCountIs(2 + number_of_tools) minimum
+ * 4. Consume textStream for real-time UI updates
+ * 5. Model handles formatting - you just render the stream
+ */
+
+import { createAnthropic } from '@ai-sdk/anthropic';
+import { youContents, youExpress, youSearch } from '@youdotcom-oss/ai-sdk-plugin';
+import { stepCountIs, streamText } from 'ai';
+
+// ============================================================================
+// INTEGRATION STEP 1: Environment Variables
+// ============================================================================
+
+if (!process.env.YDC_API_KEY) {
+  throw new Error('YDC_API_KEY environment variable is required');
+}
+
+if (!process.env.ANTHROPIC_API_KEY) {
+  throw new Error('ANTHROPIC_API_KEY environment variable is required');
+}
+
+// ============================================================================
+// INTEGRATION STEP 2: Add Tools + Configure stopWhen
+// ============================================================================
+// Use stopWhen: stepCountIs(3) - works for any number of tools with Anthropic
+
+const anthropic = createAnthropic();
+
+const result = streamText({
+  model: anthropic('claude-sonnet-4-5-20250929'),
+  tools: {
+    search: youSearch(),
+  },
+  stopWhen: stepCountIs(3), // 3 steps works regardless of tool count (Anthropic)
+  prompt: 'Your dynamic prompt here', // Replace with actual prompt
+});
+
+// ============================================================================
+// INTEGRATION STEP 3: Consume the Stream
+// ============================================================================
+// Stream text to your UI in real-time
+
+for await (const chunk of result.textStream) {
+  process.stdout.write(chunk); // Or update your UI component
+}
+
+// ============================================================================
+// Multiple Tools Example
+// ============================================================================
+// stepCountIs(3) works with any number of tools when using Anthropic
+
+const multiToolStream = streamText({
+  model: anthropic('claude-sonnet-4-5-20250929'),
+  tools: {
+    search: youSearch(),
+    agent: youExpress(),
+    extract: youContents(),
+  },
+  stopWhen: stepCountIs(3), // 3 steps works for 1, 2, or 3+ tools (Anthropic)
+  prompt: 'Your prompt here',
+});
+
+for await (const chunk of multiToolStream.textStream) {
+  process.stdout.write(chunk);
+}
+
+// ============================================================================
+// Web Framework Integration Examples
+// ============================================================================
+
+// Next.js App Router (Route Handler)
+// export async function POST(req: Request) {
+//   const { messages } = await req.json();
+//
+//   const result = streamText({
+//     model: anthropic('claude-sonnet-4-5-20250929'),
+//     tools: { search: youSearch() },
+//     stopWhen: stepCountIs(3),
+//     messages,
+//   });
+//
+//   return result.toDataStreamResponse();
+// }
+
+// Express.js
+// app.post('/api/chat', async (req, res) => {
+//   const { prompt } = req.body;
+//
+//   const result = streamText({
+//     model: anthropic('claude-sonnet-4-5-20250929'),
+//     tools: { search: youSearch() },
+//     stopWhen: stepCountIs(3),
+//     prompt,
+//   });
+//
+//   result.pipeDataStreamToResponse(res);
+// });
+
+// React Component (using useChat hook)
+// import { useChat } from 'ai/react';
+//
+// function ChatComponent() {
+//   const { messages, input, handleInputChange, handleSubmit } = useChat({
+//     api: '/api/chat', // Your endpoint with You.com tools
+//   });
+//
+//   return (
+//     <div>
+//       {messages.map(m => <div key={m.id}>{m.content}</div>)}
+//       <form onSubmit={handleSubmit}>
+//         <input value={input} onChange={handleInputChange} />
+//       </form>
+//     </div>
+//   );
+// }