@youdotcom-oss/api 0.1.1 → 0.2.1

package/src/express/express.utils.ts

@@ -1,113 +0,0 @@
-import { EXPRESS_API_URL } from '../shared/api.constants.ts'
-import type { GetUserAgent } from '../shared/api.types.ts'
-import { checkResponseForErrors } from '../shared/check-response-for-errors.ts'
-import {
-  type ExpressAgentApiResponse,
-  ExpressAgentApiResponseSchema,
-  type ExpressAgentInput,
-  type ExpressAgentMcpResponse,
-} from './express.schemas.ts'
-
-/**
- * Checks response status and throws appropriate errors for agent API calls
- */
-const agentThrowOnFailedStatus = async (response: Response) => {
-  const errorCode = response.status
-
-  const errorData = (await response.json()) as {
-    errors?: Array<{ detail?: string }>
-  }
-
-  if (errorCode === 400) {
-    throw new Error(`Bad Request:\n${JSON.stringify(errorData)}`)
-  } else if (errorCode === 401) {
-    throw new Error(
-      `Unauthorized: The Agent APIs require a valid You.com API key with agent access. Ensure your YDC_API_KEY has permissions for agent endpoints.`,
-    )
-  } else if (errorCode === 403) {
-    throw new Error(`Forbidden: You are not allowed to use the requested tool for this agent or tenant`)
-  } else if (errorCode === 429) {
-    throw new Error('Rate limited by You.com API. Please try again later.')
-  }
-  throw new Error(`Failed to call agent. Error code: ${errorCode}`)
-}
-
-export const callExpressAgent = async ({
-  YDC_API_KEY = process.env.YDC_API_KEY,
-  agentInput: { input, tools },
-  getUserAgent,
-}: {
-  agentInput: ExpressAgentInput
-  YDC_API_KEY?: string
-  getUserAgent: GetUserAgent
-}) => {
-  const requestBody: {
-    agent: string
-    input: string
-    stream: boolean
-    tools?: Array<{ type: 'web_search' }>
-  } = {
-    agent: 'express',
-    input,
-    stream: false, // Use non-streaming JSON response
-  }
-
-  // Only include tools if provided
-  if (tools) {
-    requestBody.tools = tools
-  }
-
-  const options = {
-    method: 'POST',
-    headers: new Headers({
-      Authorization: `Bearer ${YDC_API_KEY || ''}`,
-      'Content-Type': 'application/json',
-      Accept: 'application/json',
-      'User-Agent': getUserAgent(),
-    }),
-    body: JSON.stringify(requestBody),
-  }
-
-  const response = await fetch(EXPRESS_API_URL, options)
-
-  if (!response.ok) {
-    await agentThrowOnFailedStatus(response)
-  }
-
-  // Parse JSON response directly
-  const jsonResponse = await response.json()
-
-  // Check for error field in response
-  checkResponseForErrors(jsonResponse)
-
-  // Validate API response schema (full response with all fields)
-  const apiResponse: ExpressAgentApiResponse = ExpressAgentApiResponseSchema.parse(jsonResponse)
-
-  // Find the answer (always present as message.answer, validated by Zod)
-  const answerItem = apiResponse.output.find((item) => item.type === 'message.answer')
-  if (!answerItem) {
-    throw new Error('Express API response missing required message.answer item')
-  }
-
-  // Find search results (optional, present when web_search tool is used)
-  const searchItem = apiResponse.output.find((item) => item.type === 'web_search.results')
-
-  // Transform API response to MCP output format (answer + optional search results, token efficient)
-  const mcpResponse: ExpressAgentMcpResponse = {
-    answer: answerItem.text,
-    agent: apiResponse.agent,
-  }
-
-  // Transform search results if present
-  if (searchItem && 'content' in searchItem && Array.isArray(searchItem.content)) {
-    mcpResponse.results = {
-      web: searchItem.content.map((item) => ({
-        url: item.url || item.citation_uri || '',
-        title: item.title || '',
-        snippet: item.snippet || '',
-      })),
-    }
-  }
-
-  return mcpResponse
-}

package/src/express/tests/express.utils.spec.ts

@@ -1,83 +0,0 @@
-import { describe, expect, setDefaultTimeout, test } from 'bun:test'
-import { callExpressAgent } from '../express.utils.ts'
-
-const getUserAgent = () => 'API/test (You.com;TEST)'
-
-setDefaultTimeout(20_000)
-
-describe('callExpressAgent', () => {
-  test(
-    'returns answer only (WITHOUT web_search tools)',
-    async () => {
-      const result = await callExpressAgent({
-        agentInput: { input: 'What is machine learning?' },
-        getUserAgent,
-      })
-
-      // Verify MCP response structure
-      expect(result).toHaveProperty('answer')
-      expect(typeof result.answer).toBe('string')
-      expect(result.answer.length).toBeGreaterThan(0)
-
-      // Should NOT have results when web_search is not used
-      expect(result.results).toBeUndefined()
-
-      expect(result.agent).toBe('express')
-    },
-    { retry: 2 },
-  )
-
-  test(
-    'returns answer and search results (WITH web_search tools)',
-    async () => {
-      const result = await callExpressAgent({
-        agentInput: {
-          input: 'Latest developments in quantum computing',
-          tools: [{ type: 'web_search' }],
-        },
-        getUserAgent,
-      })
-
-      // Verify MCP response has both answer and results
-      expect(result).toHaveProperty('answer')
-      expect(typeof result.answer).toBe('string')
-      expect(result.answer.length).toBeGreaterThan(0)
-
-      expect(result).toHaveProperty('results')
-      expect(result.results).toHaveProperty('web')
-      expect(Array.isArray(result.results?.web)).toBe(true)
-      expect(result.results?.web.length).toBeGreaterThan(0)
-
-      // Verify each search result has required fields
-      const firstResult = result.results?.web[0]
-      expect(firstResult).toHaveProperty('url')
-      expect(firstResult).toHaveProperty('title')
-      expect(firstResult).toHaveProperty('snippet')
-      expect(typeof firstResult?.url).toBe('string')
-      expect(typeof firstResult?.title).toBe('string')
-      expect(typeof firstResult?.snippet).toBe('string')
-      expect(firstResult?.url.length).toBeGreaterThan(0)
-      expect(firstResult?.title.length).toBeGreaterThan(0)
-
-      expect(result.agent).toBe('express')
-    },
-    { timeout: 30_000, retry: 2 },
-  )
-
-  test(
-    'works without optional parameters',
-    async () => {
-      const result = await callExpressAgent({
-        agentInput: { input: 'What is the capital of France?' },
-        getUserAgent,
-        // No progressToken or sendProgress provided
-      })
-
-      // Should work normally without progress tracking
-      expect(result).toHaveProperty('answer')
-      expect(result.answer.length).toBeGreaterThan(0)
-      expect(result.agent).toBe('express')
-    },
-    { retry: 2 },
-  )
-})