chore: change type for structured to match with sources

src/__tests__/linkup-client.test.ts

@@ -142,7 +142,16 @@ describe('LinkupClient', () => {
 
   it('should handle structured output type using JSON schema', async () => {
     maxios.post.mockResolvedValueOnce({
-      data: 'foo',
+      data: {
+        data: 'foo',
+        sources: [
+          {
+            name: 'foo',
+            snippet: 'foo bar baz',
+            url: 'http://foo.bar/baz',
+          },
+        ],
+      },
     } as AxiosResponse);
 
     const result = await underTest.search({
@@ -152,7 +161,10 @@ describe('LinkupClient', () => {
       structuredOutputSchema: { type: 'string' },
     });
 
-    expect(result).toEqual('foo');
+    expect(result).toEqual({
+      data: 'foo',
+      sources: [{ name: 'foo', snippet: 'foo bar baz', url: 'http://foo.bar/baz' }],
+    });
   });
 
   it('should handle structured output type using Zod schema', async () => {

src/linkup-client.ts

@@ -18,7 +18,7 @@ import {
   SearchParams,
   SearchResults,
   SourcedAnswer,
-  StructuredOutputSchema,
+  StructuredResult,
 } from './types';
 import { concatErrorAndDetails, isZodObject } from './utils';
 
@@ -42,7 +42,7 @@ export class LinkupClient {
       : T extends 'searchResults'
         ? SearchResults
         : T extends 'structured'
-          ? StructuredOutputSchema
+          ? StructuredResult
           : never
   > {
     const headers: Record<string, string> = {
@@ -108,8 +108,11 @@ export class LinkupClient {
         return {
           results: (searchResponse as SearchResults).results,
         } as LinkupSearchResponse<T>;
-      // biome-ignore lint/complexity/noUselessSwitchCase: left for exhaustiveness
       case 'structured':
+        return {
+          data: (searchResponse as StructuredResult).data,
+          sources: (searchResponse as StructuredResult).sources,
+        } as LinkupSearchResponse<T>;
       default:
         return searchResponse as LinkupSearchResponse<T>;
     }

src/types.ts

@@ -29,7 +29,7 @@ export type LinkupSearchResponse<T> = T extends 'sourcedAnswer'
   : T extends 'searchResults'
     ? SearchResults
     : T extends 'structured'
-      ? StructuredOutputSchema
+      ? StructuredResult
       : never;
 
 export interface SearchResults {
@@ -49,6 +49,11 @@ export interface ImageSearchResult {
   url: string;
 }
 
+export interface StructuredResult {
+  data: string;
+  sources: (Source | TextSearchResult | ImageSearchResult)[];
+}
+
 export interface SourcedAnswer {
   answer: string;
   sources: (Source | TextSearchResult | ImageSearchResult)[];