notte-sdk 0.0.1 → 0.0.3

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.3" : "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;
@@ -2549,12 +2568,13 @@ var NotteClient = class {
       throw new Error("API key is required. Provide it via config.apiKey or set the NOTTE_API_KEY environment variable.");
     }
     this.config = {
-      baseUrl: config.baseUrl || "https://api.notte.cc",
+      baseUrl: config.baseUrl || process.env.NOTTE_API_URL || "https://api.notte.cc",
       apiKey,
       ...config
     };
     client.setConfig({
-      baseUrl: this.config.baseUrl
+      baseUrl: this.config.baseUrl,
+      redirect: "manual"
     });
     client.interceptors.request.use((request) => {
       request.headers.set("Authorization", `Bearer ${this.config.apiKey}`);
@@ -2562,6 +2582,33 @@ var NotteClient = class {
       request.headers.set("x-notte-sdk-version", SDK_VERSION);
       return request;
     });
+    client.interceptors.response.use(async (response, request, opts) => {
+      if ((response.status === 307 || response.status === 308) && response.headers.get("location")) {
+        const location = response.headers.get("location");
+        const headers = new Headers(request.headers);
+        if (new URL(location).origin !== new URL(request.url).origin) {
+          headers.delete("Authorization");
+        }
+        const headerRecord = {};
+        headers.forEach((v, k) => {
+          headerRecord[k] = v;
+        });
+        const redirected = await fetch(location, {
+          method: request.method,
+          headers: headerRecord,
+          body: opts.serializedBody ?? void 0
+        });
+        return redirected;
+      }
+      const ct = response.headers.get("content-type") ?? "";
+      if (ct.split(";")[0]?.trim() === "text/plain") {
+        const body = await response.arrayBuffer();
+        const headers = new Headers(response.headers);
+        headers.set("content-type", "application/json");
+        return new Response(body, { status: response.status, statusText: response.statusText, headers });
+      }
+      return response;
+    });
   }
   /**
    * Create a new session
@@ -2622,7 +2669,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);