exa-js 1.8.17 → 1.8.19

package/dist/index.mjs

@@ -953,6 +953,18 @@ var WebsetsClient = class extends WebsetsBaseClient {
   }
 };
 
+// src/zod-utils.ts
+import { ZodType } from "zod";
+import { zodToJsonSchema as convertZodToJsonSchema } from "zod-to-json-schema";
+function isZodSchema(obj) {
+  return obj instanceof ZodType;
+}
+function zodToJsonSchema(schema) {
+  return convertZodToJsonSchema(schema, {
+    $refStrategy: "none"
+  });
+}
+
 // src/research/base.ts
 var ResearchBaseClient = class {
   /**
@@ -1013,23 +1025,17 @@ var ResearchClient = class extends ResearchBaseClient {
   constructor(client) {
     super(client);
   }
-  /**
-   * Create a new research task.
-   *
-   * @param params Object containing:
-   *   - model: The research model to use (e.g., ResearchModel.ExaResearch).
-   *   - instructions: High-level guidance for the research agent.
-   *   - output: An object with a `schema` property (JSONSchema) that defines the expected output structure.
-   *
-   * @returns An object containing the unique ID of the created research task.
-   */
   async createTask(params) {
     const { instructions, model, output } = params;
+    let schema = output?.schema;
+    if (schema && isZodSchema(schema)) {
+      schema = zodToJsonSchema(schema);
+    }
     const payload = {
       instructions,
       model: model ?? "exa-research",
       output: output ? {
-        schema: output.schema,
+        schema,
         inferSchema: output.inferSchema ?? true
       } : { inferSchema: true }
     };
@@ -1162,7 +1168,16 @@ var Exa2 = class {
       contentsOptions.text = true;
     }
     if (text !== void 0) contentsOptions.text = text;
-    if (summary !== void 0) contentsOptions.summary = summary;
+    if (summary !== void 0) {
+      if (typeof summary === "object" && summary !== null && "schema" in summary && summary.schema && isZodSchema(summary.schema)) {
+        contentsOptions.summary = {
+          ...summary,
+          schema: zodToJsonSchema(summary.schema)
+        };
+      } else {
+        contentsOptions.summary = summary;
+      }
+    }
     if (highlights !== void 0) contentsOptions.highlights = highlights;
     if (subpages !== void 0) contentsOptions.subpages = subpages;
     if (subpageTarget !== void 0)
@@ -1355,28 +1370,6 @@ var Exa2 = class {
     };
     return await this.request("/contents", "POST", payload);
   }
-  /**
-   * Generate an answer to a query.
-   * @param {string} query - The question or query to answer.
-   * @param {AnswerOptions} [options] - Additional options for answer generation.
-   * @returns {Promise<AnswerResponse>} The generated answer and source references.
-   *
-   * Example with systemPrompt:
-   * ```ts
-   * const answer = await exa.answer("What is quantum computing?", {
-   *   text: true,
-   *   model: "exa",
-   *   systemPrompt: "Answer in a technical manner suitable for experts."
-   * });
-   * ```
-   *
-   * Note: For streaming responses, use the `streamAnswer` method:
-   * ```ts
-   * for await (const chunk of exa.streamAnswer(query)) {
-   *   // Handle chunks
-   * }
-   * ```
-   */
   async answer(query, options) {
     if (options?.stream) {
       throw new ExaError(
@@ -1384,43 +1377,32 @@ var Exa2 = class {
         400 /* BadRequest */
       );
     }
+    let outputSchema = options?.outputSchema;
+    if (outputSchema && isZodSchema(outputSchema)) {
+      outputSchema = zodToJsonSchema(outputSchema);
+    }
     const requestBody = {
       query,
       stream: false,
       text: options?.text ?? false,
       model: options?.model ?? "exa",
       systemPrompt: options?.systemPrompt,
-      outputSchema: options?.outputSchema
+      outputSchema
     };
     return await this.request("/answer", "POST", requestBody);
   }
-  /**
-   * Stream an answer as an async generator
-   *
-   * Each iteration yields a chunk with partial text (`content`) or new citations.
-   * Use this if you'd like to read the answer incrementally, e.g. in a chat UI.
-   *
-   * Example usage:
-   * ```ts
-   * for await (const chunk of exa.streamAnswer("What is quantum computing?", {
-   *   text: false,
-   *   systemPrompt: "Answer in a concise manner suitable for beginners."
-   * })) {
-   *   if (chunk.content) process.stdout.write(chunk.content);
-   *   if (chunk.citations) {
-   *     console.log("\nCitations: ", chunk.citations);
-   *   }
-   * }
-   * ```
-   */
   async *streamAnswer(query, options) {
+    let outputSchema = options?.outputSchema;
+    if (outputSchema && isZodSchema(outputSchema)) {
+      outputSchema = zodToJsonSchema(outputSchema);
+    }
     const body = {
       query,
       text: options?.text ?? false,
       stream: true,
       model: options?.model ?? "exa",
       systemPrompt: options?.systemPrompt,
-      outputSchema: options?.outputSchema
+      outputSchema
     };
     const response = await fetchImpl(this.baseURL + "/answer", {
       method: "POST",