@deepagents/text2sql 0.22.0 → 0.24.0

package/dist/index.js

@@ -98,7 +98,8 @@ function createGroundingContext() {
     tables: [],
     views: [],
     relationships: [],
-    info: void 0
+    info: void 0,
+    cache: /* @__PURE__ */ new Map()
   };
 }
 
@@ -516,8 +517,8 @@ var sqlValidationMarker = Symbol("SQLValidationError");
 var unanswerableSqlMarker = Symbol("UnanswerableSQLError");
 var SQLValidationError = class _SQLValidationError extends Error {
   [sqlValidationMarker];
-  constructor(message2) {
-    super(message2);
+  constructor(message) {
+    super(message);
     this.name = "SQLValidationError";
     this[sqlValidationMarker] = true;
   }
@@ -527,8 +528,8 @@ var SQLValidationError = class _SQLValidationError extends Error {
 };
 var UnanswerableSQLError = class _UnanswerableSQLError extends Error {
   [unanswerableSqlMarker];
-  constructor(message2) {
-    super(message2);
+  constructor(message) {
+    super(message);
     this.name = "UnanswerableSQLError";
     this[unanswerableSqlMarker] = true;
   }
@@ -1449,7 +1450,7 @@ import {
   user as user2,
   workflow
 } from "@deepagents/context";
-var RETRY_TEMPERATURES = [0, 0.4, 0.8];
+var RETRY_TEMPERATURES = [0, 0.2, 0.3];
 var SQL_AGENT_ROLE = "Expert SQL query generator.";
 var SQL_AGENT_OBJECTIVE = "Generate precise SQL grounded in provided schema.";
 var SQL_AGENT_POLICIES = [
@@ -1785,11 +1786,11 @@ function isModelUnavailableError(error) {
   if (!APICallError.isInstance(error)) {
     return false;
   }
-  const message2 = error.message.toLowerCase();
+  const message = error.message.toLowerCase();
   const responseBody = (error.responseBody ?? "").toLowerCase();
-  const is404ModelError = error.statusCode === 404 && (message2.includes("model") || responseBody.includes("model_not_found"));
+  const is404ModelError = error.statusCode === 404 && (message.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");
+  return is404ModelError || errorCode === "model_not_found" || responseBody.includes('"code":"model_not_found"') || message.includes("model") && message.includes("does not exist or you do not have access to it");
 }
 async function withRetry(computation, options = { retries: 3 }) {
   const errors = [];
@@ -5226,8 +5227,9 @@ import "@deepagents/agent";
 import {
   agent as agent2,
   assistant,
+  chatMessageToUIMessage,
   errorRecoveryGuardrail,
-  message
+  toMessageFragment
 } from "@deepagents/context";
 var Text2Sql = class {
   #config;
@@ -5291,7 +5293,7 @@ var Text2Sql = class {
     const producer = factory(this.#config.adapter);
     return producer.toPairs();
   }
-  async chat(messages) {
+  async chat(messages, options) {
     if (messages.length === 0) {
       throw new Error("messages must not be empty");
     }
@@ -5300,17 +5302,20 @@ var Text2Sql = class {
       ...guidelines(this.#config.teachingsOptions),
       ...await this.index()
     );
-    const lastMessage = messages[messages.length - 1];
+    const lastItem = messages[messages.length - 1];
+    const lastFragment = toMessageFragment(lastItem);
+    const lastUIMessage = chatMessageToUIMessage(lastItem);
     let assistantMsgId;
-    if (lastMessage.role === "assistant") {
-      context.set(message(lastMessage));
+    if (lastUIMessage.role === "assistant") {
+      context.set(lastFragment);
       await context.save({ branch: false });
-      assistantMsgId = lastMessage.id;
+      assistantMsgId = lastUIMessage.id;
     } else {
-      context.set(message(lastMessage));
+      context.set(lastFragment);
       await context.save();
       assistantMsgId = generateId();
     }
+    const uiMessages = messages.map(chatMessageToUIMessage);
     const { mounts: skillMounts } = context.getSkillMounts();
     const { tools: tools2 } = await createResultTools({
       adapter: this.#config.adapter,
@@ -5330,7 +5335,7 @@ var Text2Sql = class {
     });
     const result = await chatAgent.stream(
       {},
-      { transform: this.#config.transform }
+      { abortSignal: options?.abortSignal, transform: this.#config.transform }
     );
     const uiStream = result.toUIMessageStream({
       onError: (error) => this.#formatError(error),
@@ -5338,7 +5343,7 @@ var Text2Sql = class {
       sendFinish: true,
       sendReasoning: true,
       sendSources: true,
-      originalMessages: messages,
+      originalMessages: uiMessages,
       generateMessageId: () => assistantMsgId,
       messageMetadata: ({ part }) => {
         if (part.type === "finish-step") {
@@ -5357,7 +5362,7 @@ var Text2Sql = class {
       }
     });
     return createUIMessageStream({
-      originalMessages: messages,
+      originalMessages: uiMessages,
       generateId: () => assistantMsgId,
       onStepFinish: async ({ responseMessage }) => {
         context.set(assistant({ ...responseMessage, id: assistantMsgId }));