@deepagents/text2sql 0.17.0 → 0.17.1

package/dist/index.js

@@ -510,7 +510,6 @@ var fragments = [
   hint("Support both natural language questions AND raw SQL input"),
   hint("When validating user SQL, explain any errors clearly")
 ];
-var developer_agent_default = { tools, fragments };
 
 // packages/text2sql/src/lib/agents/result-tools.ts
 import { tool as tool2 } from "ai";
@@ -4346,15 +4345,24 @@ var Text2Sql = class {
     return producer.toPairs();
   }
   async chat(messages) {
+    if (messages.length === 0) {
+      throw new Error("messages must not be empty");
+    }
     const trackedFs = new TrackedFs(this.#config.filesystem);
     const context = this.#config.context(
       ...guidelines(this.#config.teachingsOptions),
       ...await this.index()
     );
-    const userMsg = messages.at(-1);
-    if (userMsg) {
-      context.set(message(userMsg));
+    const lastMessage = messages[messages.length - 1];
+    let assistantMsgId;
+    if (lastMessage.role === "assistant") {
+      context.set(message(lastMessage));
+      await context.save({ branch: false });
+      assistantMsgId = lastMessage.id;
+    } else {
+      context.set(message(lastMessage));
       await context.save();
+      assistantMsgId = generateId();
     }
     const { mounts: skillMounts } = context.getSkillMounts();
     const { tools: tools2 } = await createResultTools({
@@ -4362,7 +4370,6 @@ var Text2Sql = class {
       skillMounts,
       filesystem: trackedFs
     });
-    const assistantMsgId = generateId();
     const chatAgent = agent2({
       name: "text2sql",
       model: this.#config.model,
@@ -4429,40 +4436,6 @@ var Text2Sql = class {
       }
     });
   }
-  async developer(messages) {
-    const context = this.#config.context(
-      ...guidelines(this.#config.teachingsOptions),
-      ...developer_agent_default.fragments,
-      ...await this.index()
-    );
-    const userMsg = messages.at(-1);
-    if (userMsg) {
-      context.set(message(userMsg));
-      await context.save();
-    }
-    const developerAgent = agent2({
-      name: "developer",
-      model: this.#config.model,
-      context,
-      tools: developer_agent_default.tools
-    });
-    const result = await developerAgent.stream({
-      adapter: this.#config.adapter
-    });
-    return result.toUIMessageStream({
-      onError: (error) => this.#formatError(error),
-      sendStart: true,
-      sendFinish: true,
-      sendReasoning: true,
-      sendSources: true,
-      generateMessageId: generateId,
-      onFinish: async ({ responseMessage }) => {
-        context.set(assistant(responseMessage));
-        await context.save();
-        await context.trackUsage(await result.totalUsage);
-      }
-    });
-  }
   #formatError(error) {
     if (NoSuchToolError.isInstance(error)) {
       return "The model tried to call an unknown tool.";