Building a Research Agent

import { Tool, createToolWithValidation } from '@waymakerai/aicofounder-core';
import { z } from 'zod';

const webSearchTool = createToolWithValidation({
  name: 'web_search',
  description: 'Search the web for information on a topic. Returns titles, URLs, and snippets. Use this as the first step when researching any topic.',
  schema: z.object({
    query: z.string().min(3).max(200),
    maxResults: z.number().int().min(1).max(10).default(5),
  }),
  execute: async ({ query, maxResults }) => {
    const response = await fetch(`https://api.search.example/v1/search?q=${encodeURIComponent(query)}&num=${maxResults}`, {
      headers: { Authorization: `Bearer ${process.env.SEARCH_API_KEY}` },
    });
    const data = await response.json();
    return JSON.stringify(data.results.map((r: any) => ({
      title: r.title,
      url: r.url,
      snippet: r.snippet,
    })));
  },
});

const fetchPageTool = createToolWithValidation({
  name: 'fetch_page',
  description: 'Fetch and extract the main text content from a web page URL. Use this after web_search to get full article content.',
  schema: z.object({
    url: z.string().url(),
  }),
  execute: async ({ url }) => {
    const response = await fetch(url, { signal: AbortSignal.timeout(15000) });
    if (!response.ok) {
      return JSON.stringify({ error: true, message: `Failed to fetch: HTTP ${response.status}` });
    }
    const html = await response.text();
    // Extract main content (simplified -- use a library like readability in production)
    const text = html.replace(/<[^>]*>/g, ' ').replace(/\s+/g, ' ').slice(0, 5000);
    return JSON.stringify({ url, content: text });
  },
});

import { createAgent } from '@waymakerai/aicofounder-core';

const researchAgent = createAgent({
  name: 'research-assistant',
  model: 'gpt-4o',
  temperature: 0.3, // Lower temperature for factual accuracy
  maxSteps: 15,
  systemPrompt: `You are a thorough research assistant. When given a research question:

1. Search for relevant information using web_search
2. Fetch the 2-3 most promising pages using fetch_page
3. Synthesize the information into a clear, well-organized answer
4. Always cite your sources with [Source Title](URL) format

Guidelines:
- Cross-reference facts across multiple sources
- If sources disagree, note the discrepancy
- Distinguish between established facts and opinions
- If you cannot find reliable information, say so honestly`,
  tools: [webSearchTool, fetchPageTool],
  memory: {
    type: 'sliding-window',
    maxMessages: 30,
  },
});

import { createAgent, AgentHooks } from '@waymakerai/aicofounder-core';

const hooks: AgentHooks = {
  afterRun: async (result) => {
    // Extract URLs from the agent's output
    const urlRegex = /\[([^\]]+)\]\((https?:\/\/[^)]+)\)/g;
    const citations: Array<{ title: string; url: string }> = [];
    let match;
    while ((match = urlRegex.exec(result.output)) !== null) {
      citations.push({ title: match[1], url: match[2] });
    }
    return {
      ...result,
      metadata: { ...result.metadata, citations, citationCount: citations.length },
    };
  },
};

const researchAgent = createAgent({
  name: 'research-assistant',
  model: 'gpt-4o',
  tools: [webSearchTool, fetchPageTool],
  hooks,
  // ... rest of config
});

'use client';
import { useAgent } from '@waymakerai/aicofounder-react';

export function ResearchChat() {
  const { messages, sendMessage, isRunning, currentStep } = useAgent({
    agentId: 'research-assistant',
  });

  return (
    <div>
      <div className="messages">
        {messages.map((msg, i) => (
          <div key={i} className={`message ${msg.role}`}>
            {msg.content}
          </div>
        ))}
      </div>
      {isRunning && (
        <div className="status">
          Researching... (step {currentStep})
        </div>
      )}
      <input
        type="text"
        placeholder="Ask a research question..."
        onKeyDown={(e) => {
          if (e.key === 'Enter') {
            sendMessage(e.currentTarget.value);
            e.currentTarget.value = '';
          }
        }}
      />
    </div>
  );
}