@deepagents/text2sql 0.17.1 → 0.18.0

package/dist/index.js

@@ -801,19 +801,21 @@ async function toSql(options) {
       const sqlOutput = structuredOutput2({
         model,
         context,
-        schema: z4.union([
-          z4.object({
-            sql: z4.string().describe("The SQL query that answers the question"),
-            reasoning: z4.string().optional().describe("The reasoning steps taken to generate the SQL")
-          }),
-          z4.object({
-            error: z4.string().describe(
-              "Error message explaining why the question cannot be answered with the given schema"
-            )
-          })
-        ])
+        schema: z4.object({
+          result: z4.union([
+            z4.object({
+              sql: z4.string().describe("The SQL query that answers the question"),
+              reasoning: z4.string().describe("The reasoning steps taken to generate the SQL")
+            }),
+            z4.object({
+              error: z4.string().describe(
+                "Error message explaining why the question cannot be answered with the given schema"
+              )
+            })
+          ])
+        })
       });
-      const output = await sqlOutput.generate();
+      const { result: output } = await sqlOutput.generate();
       if ("error" in output) {
         throw new UnanswerableSQLError(output.error);
       }
@@ -843,6 +845,16 @@ function formatErrorMessage(error) {
   }
   return error.message;
 }
+function isModelUnavailableError(error) {
+  if (!APICallError.isInstance(error)) {
+    return false;
+  }
+  const message2 = error.message.toLowerCase();
+  const responseBody = (error.responseBody ?? "").toLowerCase();
+  const is404ModelError = error.statusCode === 404 && (message2.includes("model") || responseBody.includes("model_not_found"));
+  const errorCode = typeof error.data === "object" && error.data !== null && "error" in error.data && typeof error.data.error === "object" && error.data.error !== null && "code" in error.data.error && typeof error.data.error.code === "string" ? error.data.error.code.toLowerCase() : void 0;
+  return is404ModelError || errorCode === "model_not_found" || responseBody.includes('"code":"model_not_found"') || message2.includes("model") && message2.includes("does not exist or you do not have access to it");
+}
 async function withRetry(computation, options = { retries: 3 }) {
   const errors = [];
   let attempts = 0;
@@ -856,6 +868,9 @@ async function withRetry(computation, options = { retries: 3 }) {
         if (UnanswerableSQLError.isInstance(context.error)) {
           return false;
         }
+        if (isModelUnavailableError(context.error)) {
+          return false;
+        }
         if (SQLValidationError.isInstance(context.error)) {
           return true;
         }