@youdotcom-oss/mcp 1.4.0 → 1.5.0

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

@@ -68,7 +68,6 @@ describe('fetchSearchResults', () => {
       expect(Array.isArray(webResult?.snippets)).toBe(true);
 
       // Test that news results have required properties
-      // biome-ignore lint/style/noNonNullAssertion: Test
       // const newsResult = result.results.news![0];
       // expect(newsResult).toHaveProperty('url');
       // expect(newsResult).toHaveProperty('title');
@@ -169,19 +168,21 @@ describe('formatSearchResults', () => {
     expect(result.content[0]).toHaveProperty('text');
     expect(result.content[0]?.text).toContain('WEB RESULTS:');
     expect(result.content[0]?.text).toContain('Test Title');
-    // URLs should NOT be in text content
-    expect(result.content[0]?.text).not.toContain('https://example.com');
+    // URL and page_age should be in text content
+    expect(result.content[0]?.text).toContain('URL: https://example.com');
+    expect(result.content[0]?.text).toContain('Published: 2023-01-01T00:00:00');
     expect(result.structuredContent).toHaveProperty('resultCounts');
     expect(result.structuredContent.resultCounts).toHaveProperty('web', 1);
     expect(result.structuredContent.resultCounts).toHaveProperty('news', 0);
     expect(result.structuredContent.resultCounts).toHaveProperty('total', 1);
-    // URLs should be in structuredContent.results
+    // All fields should be in structuredContent.results
     expect(result.structuredContent).toHaveProperty('results');
     expect(result.structuredContent.results?.web).toBeDefined();
     expect(result.structuredContent.results?.web?.length).toBe(1);
-    expect(result.structuredContent.results?.web?.[0]).toEqual({
+    expect(result.structuredContent.results?.web?.[0]).toMatchObject({
       url: 'https://example.com',
       title: 'Test Title',
+      page_age: '2023-01-01T00:00:00',
     });
     expect(result.fullResponse).toBe(mockResponse);
   });
@@ -211,19 +212,20 @@ describe('formatSearchResults', () => {
     expect(result.content[0]?.text).toContain('NEWS RESULTS:');
     expect(result.content[0]?.text).toContain('News Title');
     expect(result.content[0]?.text).toContain('Published: 2023-01-01T00:00:00');
-    // URLs should NOT be in text content
-    expect(result.content[0]?.text).not.toContain('https://news.com/article');
+    // URL should be in text content
+    expect(result.content[0]?.text).toContain('URL: https://news.com/article');
     expect(result.structuredContent).toHaveProperty('resultCounts');
     expect(result.structuredContent.resultCounts).toHaveProperty('web', 0);
     expect(result.structuredContent.resultCounts).toHaveProperty('news', 1);
     expect(result.structuredContent.resultCounts).toHaveProperty('total', 1);
-    // URLs should be in structuredContent.results
+    // All fields should be in structuredContent.results
     expect(result.structuredContent).toHaveProperty('results');
     expect(result.structuredContent.results?.news).toBeDefined();
     expect(result.structuredContent.results?.news?.length).toBe(1);
-    expect(result.structuredContent.results?.news?.[0]).toEqual({
+    expect(result.structuredContent.results?.news?.[0]).toMatchObject({
       url: 'https://news.com/article',
       title: 'News Title',
+      page_age: '2023-01-01T00:00:00',
     });
   });
 
@@ -261,25 +263,27 @@ describe('formatSearchResults', () => {
     expect(result.content[0]?.text).toContain('WEB RESULTS:');
     expect(result.content[0]?.text).toContain('NEWS RESULTS:');
     expect(result.content[0]?.text).toContain(`=${'='.repeat(49)}`);
-    // URLs should NOT be in text content
-    expect(result.content[0]?.text).not.toContain('https://web.com');
-    expect(result.content[0]?.text).not.toContain('https://news.com/article');
+    // URLs should be in text content
+    expect(result.content[0]?.text).toContain('URL: https://web.com');
+    expect(result.content[0]?.text).toContain('URL: https://news.com/article');
     expect(result.structuredContent.resultCounts).toHaveProperty('web', 1);
     expect(result.structuredContent.resultCounts).toHaveProperty('news', 1);
     expect(result.structuredContent.resultCounts).toHaveProperty('total', 2);
-    // URLs should be in structuredContent.results
+    // All fields should be in structuredContent.results
     expect(result.structuredContent).toHaveProperty('results');
     expect(result.structuredContent.results?.web).toBeDefined();
     expect(result.structuredContent.results?.news).toBeDefined();
     expect(result.structuredContent.results?.web?.length).toBe(1);
     expect(result.structuredContent.results?.news?.length).toBe(1);
-    expect(result.structuredContent.results?.web?.[0]).toEqual({
+    expect(result.structuredContent.results?.web?.[0]).toMatchObject({
       url: 'https://web.com',
       title: 'Web Title',
+      page_age: '2023-01-01T00:00:00',
     });
-    expect(result.structuredContent.results?.news?.[0]).toEqual({
+    expect(result.structuredContent.results?.news?.[0]).toMatchObject({
       url: 'https://news.com/article',
       title: 'News Title',
+      page_age: '2023-01-01T00:00:00',
     });
   });
 });

package/src/shared/format-search-results-text.ts

@@ -8,19 +8,27 @@ type GenericSearchResult = {
   description?: string;
   snippet?: string;
   snippets?: string[];
+  page_age?: string;
 };
 
 /**
  * Format array of search results into display text
  * Used by both search and express agent formatting
  * @param results - Array of search results to format
- * @param includeUrls - Whether to include URLs in the text (default: true)
  */
 export const formatSearchResultsText = (results: GenericSearchResult[]): string => {
   return results
     .map((result) => {
       const parts: string[] = [`Title: ${result.title}`];
 
+      // Add URL
+      parts.push(`URL: ${result.url}`);
+
+      // Add page age if present
+      if (result.page_age) {
+        parts.push(`Published: ${result.page_age}`);
+      }
+
       // Add description if present (from Search API)
       if (result.description) {
         parts.push(`Description: ${result.description}`);

package/src/tests/tool.spec.ts

@@ -85,8 +85,8 @@ describe('registerSearchTool', () => {
       const text = content[0]?.text;
       expect(text).toContain('WEB RESULTS:');
       expect(text).toContain('Title:');
-      // URL should NOT be in text content anymore
-      expect(text).not.toContain('URL:');
+      // URL should be in text content
+      expect(text).toContain('URL:');
       expect(text).toContain('Description:');
       expect(text).toContain('Snippets:');
 
@@ -500,11 +500,12 @@ describe('registerContentsTool', () => {
       const text = content[0]?.text;
       expect(text).toContain('Successfully extracted content');
       expect(text).toContain('https://documentation.you.com/developer-resources/mcp-server');
-      expect(text).toContain('Format: markdown');
+      expect(text).toContain('Formats: markdown');
 
       const structuredContent = result.structuredContent as ContentsStructuredContent;
       expect(structuredContent).toHaveProperty('count', 1);
-      expect(structuredContent).toHaveProperty('format', 'markdown');
+      expect(structuredContent).toHaveProperty('formats');
+      expect(structuredContent.formats).toEqual(['markdown']);
       expect(structuredContent).toHaveProperty('items');
       expect(structuredContent.items).toHaveLength(1);
 
@@ -512,10 +513,9 @@ describe('registerContentsTool', () => {
       expect(item).toBeDefined();
 
       expect(item).toHaveProperty('url', 'https://documentation.you.com/developer-resources/mcp-server');
-      expect(item).toHaveProperty('content');
-      expect(item).toHaveProperty('contentLength');
-      expect(typeof item?.content).toBe('string');
-      expect(item?.content.length).toBeGreaterThan(0);
+      expect(item).toHaveProperty('markdown');
+      expect(typeof item?.markdown).toBe('string');
+      expect(item?.markdown?.length).toBeGreaterThan(0);
     },
     { retry: 2 },
   );
@@ -557,11 +557,11 @@ describe('registerContentsTool', () => {
       });
 
       const structuredContent = result.structuredContent as ContentsStructuredContent;
-      expect(structuredContent.format).toBe('html');
+      expect(structuredContent.formats).toEqual(['html']);
 
       const content = result.content as { type: string; text: string }[];
       const text = content[0]?.text;
-      expect(text).toContain('Format: html');
+      expect(text).toContain('Formats: html');
     },
     { retry: 2 },
   );
@@ -577,7 +577,7 @@ describe('registerContentsTool', () => {
       });
 
       const structuredContent = result.structuredContent as ContentsStructuredContent;
-      expect(structuredContent.format).toBe('markdown');
+      expect(structuredContent.formats).toEqual(['markdown']);
     },
     { retry: 2 },
   );