notte-sdk 0.0.1 → 0.0.2

package/dist/index.mjs

@@ -1585,6 +1585,16 @@ var Session = class {
 
 // src/agent.ts
 import WebSocket from "ws";
+import { z as z2 } from "zod";
+function prepareAgentRequest(data) {
+  const apiData = { ...data };
+  const rf = data.response_format;
+  if (rf && typeof rf.parse === "function" && typeof rf.safeParse === "function") {
+    apiData.response_format = z2.toJSONSchema(rf);
+    return { apiData, zodSchema: rf };
+  }
+  return { apiData, zodSchema: null };
+}
 function validatePersonaOption(persona, persona_id) {
   if (persona) {
     try {
@@ -1707,9 +1717,6 @@ var Agent = class {
       throw new Error("Either session (for running a new agent) or agent_id (for accessing an existing agent) have to be provided");
     }
   }
-  /**
-   * Start an agent task (non-blocking) - mirrors Python start method
-   */
   async start(data) {
     if (this.existingAgent) {
       throw new Error("You cannot call start() on an agent instantiated from agent id");
@@ -1718,9 +1725,10 @@ var Agent = class {
       throw new Error("Agent not properly initialized with session");
     }
     try {
+      const { apiData } = prepareAgentRequest(data);
       const requestBody = {
         ...this.request,
-        ...data
+        ...apiData
       };
       const response = await agentStart({
         client: this.client.getClient(),
@@ -1735,17 +1743,28 @@ var Agent = class {
       throw new Error(`Failed to start agent: ${error instanceof Error ? error.message : String(error)}`);
     }
   }
-  /**
-   * Run agent task (blocking) - mirrors Python run method behavior
-   * This is the main method that mirrors the Python SDK's run method
-   */
   async run(data) {
     if (this.existingAgent) {
       throw new Error("You cannot call run() on an agent instantiated from agent id");
     }
-    this.response = await this.start(data);
+    const { apiData, zodSchema } = prepareAgentRequest(data);
+    this.response = await this.start(apiData);
     console.log(`[Agent] ${this.agentId} started`);
-    return await this.watchLogsAndWait(true, data.updateHandler || null);
+    const result = await this.watchLogsAndWait(true, data.updateHandler || null);
+    if (zodSchema && result && result.answer !== void 0 && result.answer !== null) {
+      let candidate;
+      try {
+        candidate = typeof result.answer === "string" ? JSON.parse(result.answer) : result.answer;
+      } catch {
+        throw new Error(`Agent answer is not valid JSON and cannot be validated against the schema`);
+      }
+      try {
+        result.answer = zodSchema.parse(candidate);
+      } catch (error) {
+        throw new Error(`Agent answer schema validation failed: ${error instanceof Error ? error.message : String(error)}`);
+      }
+    }
+    return result;
   }
   /**
    * Watch logs and wait for completion - mirrors Python watch_logs_and_wait
@@ -2463,7 +2482,7 @@ var NottePersona = class {
 };
 
 // src/version.ts
-var SDK_VERSION = true ? "0.0.1" : "dev";
+var SDK_VERSION = true ? "0.0.2" : "dev";
 
 // src/functions.ts
 var NotteFunction = class {
@@ -2540,7 +2559,7 @@ var NotteFunction = class {
 };
 
 // src/client.ts
-import { z as z2 } from "zod";
+import { z as z3 } from "zod";
 var NotteClient = class {
   constructor(config = {}) {
     const apiKey = config.apiKey || process.env.NOTTE_API_KEY;
@@ -2622,7 +2641,7 @@ var NotteClient = class {
       if (options.json_schema) {
         apiOptions.response_format = options.json_schema;
       } else {
-        apiOptions.response_format = z2.toJSONSchema(options.response_format);
+        apiOptions.response_format = z3.toJSONSchema(options.response_format);
       }
       if (!apiOptions.instructions) {
         const schemaStr = JSON.stringify(apiOptions.response_format);