fix: fix types and user agent version

Author: Shedna

src/__tests__/linkup-client.test.ts

@@ -92,7 +92,7 @@ describe('LinkupClient', () => {
       baseURL: 'http://foo.bar/baz',
       headers: {
         Authorization: 'Bearer 1234',
-        'User-Agent': 'Linkup-JS-SDK/2.0.0',
+        'User-Agent': 'Linkup-JS-SDK/2.1.1',
       },
     });
   });

src/linkup-client.ts

@@ -10,13 +10,12 @@ import {
   SearchParams,
   SearchResults,
   SourcedAnswer,
-  StructuredOutputSchema,
 } from './types';
 import { refineError } from './utils/refine-error.utils';
 import { isZodObject } from './utils/schema.utils';
 
 export class LinkupClient {
-  private readonly USER_AGENT = 'Linkup-JS-SDK/2.0.0';
+  private readonly USER_AGENT = 'Linkup-JS-SDK/2.1.1';
   private readonly client: AxiosInstance;
 
   constructor(config: ApiConfig) {
@@ -41,24 +40,16 @@ export class LinkupClient {
     );
   }
 
-  async search<T extends 'sourcedAnswer' | 'searchResults' | 'structured'>(
+  async search<T extends SearchOutputType>(
     params: SearchParams<T>,
-  ): Promise<
-    T extends 'sourcedAnswer'
-      ? SourcedAnswer
-      : T extends 'searchResults'
-        ? SearchResults
-        : T extends 'structured'
-          ? StructuredOutputSchema
-          : never
-  > {
+  ): Promise<LinkupSearchResponse<T>> {
     return this.client
       .post('/search', this.sanitizeParams(params))
-      .then(response => this.formatResponse<T>(response.data, params.outputType));
+      .then(response => this.formatResponse(response.data, params.outputType));
   }
 
-  async fetch(params: FetchParams): Promise<LinkupFetchResponse> {
-    return this.client.post<LinkupFetchResponse>('/fetch', params).then(response => response.data);
+  async fetch<T extends FetchParams>(params: T): Promise<LinkupFetchResponse<T>> {
+    return this.client.post('/fetch', params).then(response => response.data);
   }
 
   private sanitizeParams<T extends SearchOutputType>({
@@ -91,9 +82,9 @@ export class LinkupClient {
     };
   }
 
-  private formatResponse<T>(
+  private formatResponse<T extends SearchOutputType>(
     searchResponse: unknown,
-    outputType: SearchOutputType,
+    outputType: T,
   ): LinkupSearchResponse<T> {
     switch (outputType) {
       case 'sourcedAnswer':

src/types.ts

@@ -9,14 +9,16 @@ export interface ApiConfig {
   baseUrl?: string;
 }
 
-export type StructuredOutputSchema = Record<string, unknown> | ZodObject<ZodRawShape>;
+export type StructuredOutputSchema = Record<string, unknown>;
+
+export type StructuredOutputInputSchema = StructuredOutputSchema | ZodObject<ZodRawShape>;
 
 export interface SearchParams<T extends SearchOutputType> {
   query: string;
   depth: SearchDepth;
   outputType: T;
   includeImages?: boolean;
-  structuredOutputSchema?: StructuredOutputSchema;
+  structuredOutputSchema?: StructuredOutputInputSchema;
   includeDomains?: string[];
   excludeDomains?: string[];
   fromDate?: Date;
@@ -77,7 +79,6 @@ export interface FetchParams {
   includeRawHtml?: boolean;
 }
 
-export interface LinkupFetchResponse {
+export type LinkupFetchResponse<T extends FetchParams = FetchParams> = {
   markdown: string;
-  rawHtml?: string;
-}
+} & (T['includeRawHtml'] extends true ? { rawHtml: string } : Record<string, never>);