@fluidframework/tree-agent-langchain 2.90.0 → 2.91.0

package/CHANGELOG.md

@@ -1,5 +1,9 @@
 # @fluidframework/tree-agent-langchain
 
+## 2.91.0
+
+Dependency updates only.
+
 ## 2.90.0
 
 Dependency updates only.

package/dist/alpha.d.ts

@@ -16,6 +16,7 @@
 
 export {
 	// #region @alpha APIs
-	createLangchainChatModel
+	createLangchainChatModel, 
+	createLegacyLangchainChatModel
 	// #endregion
 } from "./index.js";

package/dist/chatModel.d.ts

@@ -5,9 +5,21 @@
 import type { SharedTreeChatModel } from "@fluidframework/tree-agent/alpha";
 import type { BaseChatModel } from "@langchain/core/language_models/chat_models";
 /**
- * Creates a `SharedTreeChatModel` that uses the LangChain library to connect to the underlying LLM.
+ * Creates a stateless {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks Use with {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent}.
  * @param langchainModel - The LangChain chat model to use.
  * @alpha
  */
 export declare function createLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel;
+/**
+ * Creates a legacy stateful {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks This implementation maintains internal message history and manages the edit loop via the
+ * {@link @fluidframework/tree-agent#SharedTreeChatQuery.edit | edit} callback pattern.
+ * Use with {@link @fluidframework/tree-agent#SharedTreeSemanticAgent}.
+ * @param langchainModel - The LangChain chat model to use.
+ * @deprecated Use {@link createLangchainChatModel} with
+ * {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent} instead.
+ * @alpha
+ */
+export declare function createLegacyLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel;
 //# sourceMappingURL=chatModel.d.ts.map

package/dist/chatModel.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chatModel.d.ts","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAKH,OAAO,KAAK,EAEX,mBAAmB,EAEnB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6CAA6C,CAAC;AAKjF;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,cAAc,EAAE,aAAa,GAAG,mBAAmB,CAE3F"}
+{"version":3,"file":"chatModel.d.ts","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,KAAK,EAEX,mBAAmB,EAInB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6CAA6C,CAAC;AAOjF;;;;;GAKG;AACH,wBAAgB,wBAAwB,CAAC,cAAc,EAAE,aAAa,GAAG,mBAAmB,CAE3F;AAuHD;;;;;;;;;GASG;AACH,wBAAgB,8BAA8B,CAC7C,cAAc,EAAE,aAAa,GAC3B,mBAAmB,CAErB"}

package/dist/chatModel.js

@@ -4,13 +4,14 @@
  * Licensed under the MIT License.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createLangchainChatModel = void 0;
-/* eslint-disable import-x/no-internal-modules */
+exports.createLegacyLangchainChatModel = exports.createLangchainChatModel = void 0;
 const internal_1 = require("@fluidframework/telemetry-utils/internal");
-const messages_1 = require("@langchain/core/messages");
-const tools_1 = require("@langchain/core/tools");
+const messages_1 = require("@langchain/core/messages"); // eslint-disable-line import-x/no-internal-modules
+const tools_1 = require("@langchain/core/tools"); // eslint-disable-line import-x/no-internal-modules
+// #region New stateless implementation
 /**
- * Creates a `SharedTreeChatModel` that uses the LangChain library to connect to the underlying LLM.
+ * Creates a stateless {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks Use with {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent}.
  * @param langchainModel - The LangChain chat model to use.
  * @alpha
  */
@@ -18,7 +19,120 @@ function createLangchainChatModel(langchainModel) {
     return new LangchainChatModel(langchainModel);
 }
 exports.createLangchainChatModel = createLangchainChatModel;
+/**
+ * Stateless LangChain adapter for {@link @fluidframework/tree-agent#SharedTreeChatModel}.
+ * @remarks This class does not maintain internal message history. All context is provided
+ * via the `history` parameter to {@link LangchainChatModel.invoke}.
+ */
 class LangchainChatModel {
+    constructor(model) {
+        this.model = model;
+        this.editToolName = "GenerateTreeEditingCode";
+    }
+    get name() {
+        const name = this.model.metadata?.modelName;
+        return typeof name === "string" ? name : undefined;
+    }
+    async invoke(history) {
+        // Convert TreeAgentChatMessage[] to LangChain BaseMessage[]
+        const messages = convertToLangchainMessages(history);
+        // Create a placeholder tool definition so the LLM knows the tool exists.
+        // The actual execution is handled by the agent's edit loop — this tool
+        // is never directly invoked, it only tells the LLM the tool signature.
+        const editingTool = (0, tools_1.tool)(async (js) => js, {
+            name: this.editToolName,
+            description: "Invokes a JavaScript code snippet to edit a tree of application data.",
+        });
+        const runnable = this.model.bindTools?.([editingTool], {
+            tool_choice: "auto",
+        });
+        if (runnable === undefined) {
+            throw new internal_1.UsageError("LLM client must support function calling or tool use.");
+        }
+        const responseMessage = await runnable.invoke(messages);
+        // Parse the response into TreeAgentChatResponse.
+        // Return the first tool call as a tool_call message, preserving the raw args.
+        // Arg parsing (extracting code) is the agent's responsibility.
+        const firstToolCall = responseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0
+            ? responseMessage.tool_calls[0]
+            : undefined;
+        if (firstToolCall !== undefined) {
+            return {
+                role: "tool_call",
+                toolCallId: firstToolCall.id,
+                toolName: firstToolCall.name,
+                toolArgs: firstToolCall.args,
+            };
+        }
+        const content = typeof responseMessage.text === "string"
+            ? responseMessage.text
+            : typeof responseMessage.content === "string"
+                ? responseMessage.content
+                : JSON.stringify(responseMessage.content);
+        return { role: "assistant", content };
+    }
+}
+/**
+ * Converts an array of {@link TreeAgentChatMessage} to LangChain {@link BaseMessage} format.
+ */
+function convertToLangchainMessages(history) {
+    const messages = [];
+    for (const msg of history) {
+        switch (msg.role) {
+            case "system": {
+                messages.push(new messages_1.SystemMessage(msg.content));
+                break;
+            }
+            case "user": {
+                messages.push(new messages_1.HumanMessage(msg.content));
+                break;
+            }
+            case "assistant": {
+                messages.push(new messages_1.AIMessage(msg.content));
+                break;
+            }
+            case "tool_call": {
+                messages.push(new messages_1.AIMessage({
+                    content: "",
+                    tool_calls: [
+                        {
+                            id: msg.toolCallId,
+                            name: msg.toolName,
+                            args: msg.toolArgs,
+                        },
+                    ],
+                }));
+                break;
+            }
+            case "tool_result": {
+                messages.push(new messages_1.ToolMessage({
+                    content: msg.content,
+                    tool_call_id: msg.toolCallId ?? "",
+                }));
+                break;
+            }
+            // No default
+        }
+    }
+    return messages;
+}
+// #endregion
+// #region Legacy stateful implementation
+/**
+ * Creates a legacy stateful {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks This implementation maintains internal message history and manages the edit loop via the
+ * {@link @fluidframework/tree-agent#SharedTreeChatQuery.edit | edit} callback pattern.
+ * Use with {@link @fluidframework/tree-agent#SharedTreeSemanticAgent}.
+ * @param langchainModel - The LangChain chat model to use.
+ * @deprecated Use {@link createLangchainChatModel} with
+ * {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent} instead.
+ * @alpha
+ */
+function createLegacyLangchainChatModel(langchainModel) {
+    return new LegacyLangchainChatModel(langchainModel);
+}
+exports.createLegacyLangchainChatModel = createLegacyLangchainChatModel;
+class LegacyLangchainChatModel {
     constructor(model) {
         this.model = model;
         this.messages = [];
@@ -31,6 +145,7 @@ class LangchainChatModel {
     appendContext(text) {
         this.messages.push(new messages_1.SystemMessage(text));
     }
+    // eslint-disable-next-line import-x/no-deprecated
     async query(query) {
         this.messages.push(new messages_1.HumanMessage(query.text));
         return this.queryEdit(async (js) => query.edit(js));
@@ -40,7 +155,9 @@ class LangchainChatModel {
             name: this.editToolName,
             description: "Invokes a JavaScript code snippet to edit a tree of application data.",
         });
-        const runnable = this.model.bindTools?.([editingTool], { tool_choice: "auto" });
+        const runnable = this.model.bindTools?.([editingTool], {
+            tool_choice: "auto",
+        });
         if (runnable === undefined) {
             throw new internal_1.UsageError("LLM client must support function calling or tool use.");
         }
@@ -77,4 +194,5 @@ function isEditResult(value) {
     return (typeof value.type === "string" &&
         typeof value.message === "string");
 }
+// #endregion
 //# sourceMappingURL=chatModel.js.map

package/dist/chatModel.js.map

@@ -1 +1 @@
-{"version":3,"file":"chatModel.js","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,iDAAiD;AAEjD,uEAAsE;AAQtE,uDAAuE;AACvE,iDAA6C;AAE7C;;;;GAIG;AACH,SAAgB,wBAAwB,CAAC,cAA6B;IACrE,OAAO,IAAI,kBAAkB,CAAC,cAAc,CAAC,CAAC;AAC/C,CAAC;AAFD,4DAEC;AAED,MAAM,kBAAkB;IAGvB,YAAoC,KAAoB;QAApB,UAAK,GAAL,KAAK,CAAe;QAFvC,aAAQ,GAAkB,EAAE,CAAC;QAI9B,iBAAY,GAAG,yBAAyB,CAAC;IAFE,CAAC;IAI5D,IAAW,IAAI;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC;QAC5C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAEM,aAAa,CAAC,IAAY;QAChC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,wBAAa,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7C,CAAC;IAEM,KAAK,CAAC,KAAK,CAAC,KAA0B;QAC5C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,uBAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,EAAU,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,IAAiC;QACxD,MAAM,WAAW,GAAG,IAAA,YAAI,EAAC,IAAI,EAAE;YAC9B,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,WAAW,EAAE,uEAAuE;SACpF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;QAChF,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,qBAAU,CAAC,uDAAuD,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAEpC,IAAI,eAAe,CAAC,UAAU,KAAK,SAAS,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvF,KAAK,MAAM,QAAQ,IAAI,eAAe,CAAC,UAAU,EAAE,CAAC;gBACnD,QAAQ,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACvB,KAAK,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;wBACtD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;wBAC/B,MAAM,UAAU,GAAY,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;wBACxD,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;4BACzE,OAAO,UAAU,CAAC,OAAO,CAAC;wBAC3B,CAAC;wBACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC7B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,uBAAY,CAAC,2BAA2B,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,eAAe,CAAC,IAAI,CAAC;IAC7B,CAAC;CACD;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,KAAc;IACnC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACjD,OAAO,KAAK,CAAC;IACd,CAAC;IACD,OAAO,CACN,OAAQ,KAAoB,CAAC,IAAI,KAAK,QAAQ;QAC9C,OAAQ,KAAoB,CAAC,OAAO,KAAK,QAAQ,CACjD,CAAC;AACH,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\n/* eslint-disable import-x/no-internal-modules */\n\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport type {\n\tEditResult,\n\tSharedTreeChatModel,\n\tSharedTreeChatQuery,\n} from \"@fluidframework/tree-agent/alpha\";\nimport type { BaseChatModel } from \"@langchain/core/language_models/chat_models\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport { HumanMessage, SystemMessage } from \"@langchain/core/messages\";\nimport { tool } from \"@langchain/core/tools\";\n\n/**\n * Creates a `SharedTreeChatModel` that uses the LangChain library to connect to the underlying LLM.\n * @param langchainModel - The LangChain chat model to use.\n * @alpha\n */\nexport function createLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel {\n\treturn new LangchainChatModel(langchainModel);\n}\n\nclass LangchainChatModel implements SharedTreeChatModel {\n\tprivate readonly messages: BaseMessage[] = [];\n\n\tpublic constructor(private readonly model: BaseChatModel) {}\n\n\tpublic readonly editToolName = \"GenerateTreeEditingCode\";\n\n\tpublic get name(): string | undefined {\n\t\tconst name = this.model.metadata?.modelName;\n\t\treturn typeof name === \"string\" ? name : undefined;\n\t}\n\n\tpublic appendContext(text: string): void {\n\t\tthis.messages.push(new SystemMessage(text));\n\t}\n\n\tpublic async query(query: SharedTreeChatQuery): Promise<string> {\n\t\tthis.messages.push(new HumanMessage(query.text));\n\t\treturn this.queryEdit(async (js: string) => query.edit(js));\n\t}\n\n\tprivate async queryEdit(edit: SharedTreeChatQuery[\"edit\"]): Promise<string> {\n\t\tconst editingTool = tool(edit, {\n\t\t\tname: this.editToolName,\n\t\t\tdescription: \"Invokes a JavaScript code snippet to edit a tree of application data.\",\n\t\t});\n\t\tconst runnable = this.model.bindTools?.([editingTool], { tool_choice: \"auto\" });\n\t\tif (runnable === undefined) {\n\t\t\tthrow new UsageError(\"LLM client must support function calling or tool use.\");\n\t\t}\n\n\t\tconst responseMessage = await runnable.invoke(this.messages);\n\t\tthis.messages.push(responseMessage);\n\n\t\tif (responseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0) {\n\t\t\tfor (const toolCall of responseMessage.tool_calls) {\n\t\t\t\tswitch (toolCall.name) {\n\t\t\t\t\tcase editingTool.name: {\n\t\t\t\t\t\tconst toolResult = await editingTool.invoke(toolCall);\n\t\t\t\t\t\tthis.messages.push(toolResult);\n\t\t\t\t\t\tconst editResult: unknown = JSON.parse(toolResult.text);\n\t\t\t\t\t\tif (isEditResult(editResult) && editResult.type === \"tooManyEditsError\") {\n\t\t\t\t\t\t\treturn editResult.message;\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn this.queryEdit(edit);\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthis.messages.push(new HumanMessage(`Unrecognized tool call: ${toolCall.name}`));\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn responseMessage.text;\n\t}\n}\n\n/**\n * Type guard for {@link EditResult}.\n */\nfunction isEditResult(value: unknown): value is EditResult {\n\tif (value === null || typeof value !== \"object\") {\n\t\treturn false;\n\t}\n\treturn (\n\t\ttypeof (value as EditResult).type === \"string\" &&\n\t\ttypeof (value as EditResult).message === \"string\"\n\t);\n}\n"]}
+{"version":3,"file":"chatModel.js","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH,uEAAsE;AAUtE,uDAA+F,CAAC,mDAAmD;AACnJ,iDAA6C,CAAC,mDAAmD;AAEjG,uCAAuC;AAEvC;;;;;GAKG;AACH,SAAgB,wBAAwB,CAAC,cAA6B;IACrE,OAAO,IAAI,kBAAkB,CAAC,cAAc,CAAC,CAAC;AAC/C,CAAC;AAFD,4DAEC;AAED;;;;GAIG;AACH,MAAM,kBAAkB;IAGvB,YAAoC,KAAoB;QAApB,UAAK,GAAL,KAAK,CAAe;QAFxC,iBAAY,GAAG,yBAAyB,CAAC;IAEE,CAAC;IAE5D,IAAW,IAAI;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC;QAC5C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAEM,KAAK,CAAC,MAAM,CAClB,OAAwC;QAExC,4DAA4D;QAC5D,MAAM,QAAQ,GAAkB,0BAA0B,CAAC,OAAO,CAAC,CAAC;QAEpE,yEAAyE;QACzE,uEAAuE;QACvE,uEAAuE;QACvE,MAAM,WAAW,GAAG,IAAA,YAAI,EAAC,KAAK,EAAE,EAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YAClD,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,WAAW,EAAE,uEAAuE;SACpF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE;YACtD,WAAW,EAAE,MAAM;SACnB,CAAC,CAAC;QACH,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,qBAAU,CAAC,uDAAuD,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAExD,iDAAiD;QACjD,8EAA8E;QAC9E,+DAA+D;QAC/D,MAAM,aAAa,GAClB,eAAe,CAAC,UAAU,KAAK,SAAS,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;YAChF,CAAC,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC;YAC/B,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;YACjC,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,aAAa,CAAC,EAAE;gBAC5B,QAAQ,EAAE,aAAa,CAAC,IAAI;gBAC5B,QAAQ,EAAE,aAAa,CAAC,IAAI;aAC5B,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GACZ,OAAO,eAAe,CAAC,IAAI,KAAK,QAAQ;YACvC,CAAC,CAAC,eAAe,CAAC,IAAI;YACtB,CAAC,CAAC,OAAO,eAAe,CAAC,OAAO,KAAK,QAAQ;gBAC5C,CAAC,CAAC,eAAe,CAAC,OAAO;gBACzB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAC7C,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;IACvC,CAAC;CACD;AAED;;GAEG;AACH,SAAS,0BAA0B,CAAC,OAAwC;IAC3E,MAAM,QAAQ,GAAkB,EAAE,CAAC;IACnC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;QAC3B,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACf,QAAQ,CAAC,IAAI,CAAC,IAAI,wBAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC9C,MAAM;YACP,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACb,QAAQ,CAAC,IAAI,CAAC,IAAI,uBAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC7C,MAAM;YACP,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBAClB,QAAQ,CAAC,IAAI,CAAC,IAAI,oBAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC1C,MAAM;YACP,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBAClB,QAAQ,CAAC,IAAI,CACZ,IAAI,oBAAS,CAAC;oBACb,OAAO,EAAE,EAAE;oBACX,UAAU,EAAE;wBACX;4BACC,EAAE,EAAE,GAAG,CAAC,UAAU;4BAClB,IAAI,EAAE,GAAG,CAAC,QAAQ;4BAClB,IAAI,EAAE,GAAG,CAAC,QAAQ;yBAClB;qBACD;iBACD,CAAC,CACF,CAAC;gBACF,MAAM;YACP,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACpB,QAAQ,CAAC,IAAI,CACZ,IAAI,sBAAW,CAAC;oBACf,OAAO,EAAE,GAAG,CAAC,OAAO;oBACpB,YAAY,EAAE,GAAG,CAAC,UAAU,IAAI,EAAE;iBAClC,CAAC,CACF,CAAC;gBACF,MAAM;YACP,CAAC;YACD,aAAa;QACd,CAAC;IACF,CAAC;IACD,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,aAAa;AAEb,yCAAyC;AAEzC;;;;;;;;;GASG;AACH,SAAgB,8BAA8B,CAC7C,cAA6B;IAE7B,OAAO,IAAI,wBAAwB,CAAC,cAAc,CAAC,CAAC;AACrD,CAAC;AAJD,wEAIC;AAED,MAAM,wBAAwB;IAG7B,YAAoC,KAAoB;QAApB,UAAK,GAAL,KAAK,CAAe;QAFvC,aAAQ,GAAkB,EAAE,CAAC;QAI9B,iBAAY,GAAG,yBAAyB,CAAC;IAFE,CAAC;IAI5D,IAAW,IAAI;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC;QAC5C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAEM,aAAa,CAAC,IAAY;QAChC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,wBAAa,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED,kDAAkD;IAC3C,KAAK,CAAC,KAAK,CAAC,KAA0B;QAC5C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,uBAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,EAAU,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,KAAK,CAAC,SAAS,CACtB,IAAiC;QAEjC,MAAM,WAAW,GAAG,IAAA,YAAI,EAAC,IAAI,EAAE;YAC9B,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,WAAW,EAAE,uEAAuE;SACpF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE;YACtD,WAAW,EAAE,MAAM;SACnB,CAAC,CAAC;QACH,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,qBAAU,CAAC,uDAAuD,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAEpC,IAAI,eAAe,CAAC,UAAU,KAAK,SAAS,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvF,KAAK,MAAM,QAAQ,IAAI,eAAe,CAAC,UAAU,EAAE,CAAC;gBACnD,QAAQ,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACvB,KAAK,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;wBACtD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;wBAC/B,MAAM,UAAU,GAAY,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;wBACxD,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;4BACzE,OAAO,UAAU,CAAC,OAAO,CAAC;wBAC3B,CAAC;wBACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC7B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,uBAAY,CAAC,2BAA2B,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,eAAe,CAAC,IAAI,CAAC;IAC7B,CAAC;CACD;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,KAAc;IACnC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACjD,OAAO,KAAK,CAAC;IACd,CAAC;IACD,OAAO,CACN,OAAQ,KAAoB,CAAC,IAAI,KAAK,QAAQ;QAC9C,OAAQ,KAAoB,CAAC,OAAO,KAAK,QAAQ,CACjD,CAAC;AACH,CAAC;AAED,aAAa","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport type {\n\tEditResult,\n\tSharedTreeChatModel,\n\tTreeAgentChatMessage,\n\tTreeAgentChatResponse,\n\tSharedTreeChatQuery, // eslint-disable-line import-x/no-deprecated\n} from \"@fluidframework/tree-agent/alpha\";\nimport type { BaseChatModel } from \"@langchain/core/language_models/chat_models\"; // eslint-disable-line import-x/no-internal-modules\nimport type { BaseMessage } from \"@langchain/core/messages\"; // eslint-disable-line import-x/no-internal-modules\nimport { AIMessage, HumanMessage, SystemMessage, ToolMessage } from \"@langchain/core/messages\"; // eslint-disable-line import-x/no-internal-modules\nimport { tool } from \"@langchain/core/tools\"; // eslint-disable-line import-x/no-internal-modules\n\n// #region New stateless implementation\n\n/**\n * Creates a stateless {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.\n * @remarks Use with {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent}.\n * @param langchainModel - The LangChain chat model to use.\n * @alpha\n */\nexport function createLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel {\n\treturn new LangchainChatModel(langchainModel);\n}\n\n/**\n * Stateless LangChain adapter for {@link @fluidframework/tree-agent#SharedTreeChatModel}.\n * @remarks This class does not maintain internal message history. All context is provided\n * via the `history` parameter to {@link LangchainChatModel.invoke}.\n */\nclass LangchainChatModel implements SharedTreeChatModel {\n\tpublic readonly editToolName = \"GenerateTreeEditingCode\";\n\n\tpublic constructor(private readonly model: BaseChatModel) {}\n\n\tpublic get name(): string | undefined {\n\t\tconst name = this.model.metadata?.modelName;\n\t\treturn typeof name === \"string\" ? name : undefined;\n\t}\n\n\tpublic async invoke(\n\t\thistory: readonly TreeAgentChatMessage[],\n\t): Promise<TreeAgentChatResponse> {\n\t\t// Convert TreeAgentChatMessage[] to LangChain BaseMessage[]\n\t\tconst messages: BaseMessage[] = convertToLangchainMessages(history);\n\n\t\t// Create a placeholder tool definition so the LLM knows the tool exists.\n\t\t// The actual execution is handled by the agent's edit loop — this tool\n\t\t// is never directly invoked, it only tells the LLM the tool signature.\n\t\tconst editingTool = tool(async (js: string) => js, {\n\t\t\tname: this.editToolName,\n\t\t\tdescription: \"Invokes a JavaScript code snippet to edit a tree of application data.\",\n\t\t});\n\n\t\tconst runnable = this.model.bindTools?.([editingTool], {\n\t\t\ttool_choice: \"auto\",\n\t\t});\n\t\tif (runnable === undefined) {\n\t\t\tthrow new UsageError(\"LLM client must support function calling or tool use.\");\n\t\t}\n\n\t\tconst responseMessage = await runnable.invoke(messages);\n\n\t\t// Parse the response into TreeAgentChatResponse.\n\t\t// Return the first tool call as a tool_call message, preserving the raw args.\n\t\t// Arg parsing (extracting code) is the agent's responsibility.\n\t\tconst firstToolCall =\n\t\t\tresponseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0\n\t\t\t\t? responseMessage.tool_calls[0]\n\t\t\t\t: undefined;\n\t\tif (firstToolCall !== undefined) {\n\t\t\treturn {\n\t\t\t\trole: \"tool_call\",\n\t\t\t\ttoolCallId: firstToolCall.id,\n\t\t\t\ttoolName: firstToolCall.name,\n\t\t\t\ttoolArgs: firstToolCall.args,\n\t\t\t};\n\t\t}\n\n\t\tconst content =\n\t\t\ttypeof responseMessage.text === \"string\"\n\t\t\t\t? responseMessage.text\n\t\t\t\t: typeof responseMessage.content === \"string\"\n\t\t\t\t\t? responseMessage.content\n\t\t\t\t\t: JSON.stringify(responseMessage.content);\n\t\treturn { role: \"assistant\", content };\n\t}\n}\n\n/**\n * Converts an array of {@link TreeAgentChatMessage} to LangChain {@link BaseMessage} format.\n */\nfunction convertToLangchainMessages(history: readonly TreeAgentChatMessage[]): BaseMessage[] {\n\tconst messages: BaseMessage[] = [];\n\tfor (const msg of history) {\n\t\tswitch (msg.role) {\n\t\t\tcase \"system\": {\n\t\t\t\tmessages.push(new SystemMessage(msg.content));\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"user\": {\n\t\t\t\tmessages.push(new HumanMessage(msg.content));\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"assistant\": {\n\t\t\t\tmessages.push(new AIMessage(msg.content));\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"tool_call\": {\n\t\t\t\tmessages.push(\n\t\t\t\t\tnew AIMessage({\n\t\t\t\t\t\tcontent: \"\",\n\t\t\t\t\t\ttool_calls: [\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tid: msg.toolCallId,\n\t\t\t\t\t\t\t\tname: msg.toolName,\n\t\t\t\t\t\t\t\targs: msg.toolArgs,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t],\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"tool_result\": {\n\t\t\t\tmessages.push(\n\t\t\t\t\tnew ToolMessage({\n\t\t\t\t\t\tcontent: msg.content,\n\t\t\t\t\t\ttool_call_id: msg.toolCallId ?? \"\",\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\t// No default\n\t\t}\n\t}\n\treturn messages;\n}\n\n// #endregion\n\n// #region Legacy stateful implementation\n\n/**\n * Creates a legacy stateful {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.\n * @remarks This implementation maintains internal message history and manages the edit loop via the\n * {@link @fluidframework/tree-agent#SharedTreeChatQuery.edit | edit} callback pattern.\n * Use with {@link @fluidframework/tree-agent#SharedTreeSemanticAgent}.\n * @param langchainModel - The LangChain chat model to use.\n * @deprecated Use {@link createLangchainChatModel} with\n * {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent} instead.\n * @alpha\n */\nexport function createLegacyLangchainChatModel(\n\tlangchainModel: BaseChatModel,\n): SharedTreeChatModel {\n\treturn new LegacyLangchainChatModel(langchainModel);\n}\n\nclass LegacyLangchainChatModel implements SharedTreeChatModel {\n\tprivate readonly messages: BaseMessage[] = [];\n\n\tpublic constructor(private readonly model: BaseChatModel) {}\n\n\tpublic readonly editToolName = \"GenerateTreeEditingCode\";\n\n\tpublic get name(): string | undefined {\n\t\tconst name = this.model.metadata?.modelName;\n\t\treturn typeof name === \"string\" ? name : undefined;\n\t}\n\n\tpublic appendContext(text: string): void {\n\t\tthis.messages.push(new SystemMessage(text));\n\t}\n\n\t// eslint-disable-next-line import-x/no-deprecated\n\tpublic async query(query: SharedTreeChatQuery): Promise<string> {\n\t\tthis.messages.push(new HumanMessage(query.text));\n\t\treturn this.queryEdit(async (js: string) => query.edit(js));\n\t}\n\n\tprivate async queryEdit(\n\t\tedit: SharedTreeChatQuery[\"edit\"], // eslint-disable-line import-x/no-deprecated\n\t): Promise<string> {\n\t\tconst editingTool = tool(edit, {\n\t\t\tname: this.editToolName,\n\t\t\tdescription: \"Invokes a JavaScript code snippet to edit a tree of application data.\",\n\t\t});\n\t\tconst runnable = this.model.bindTools?.([editingTool], {\n\t\t\ttool_choice: \"auto\",\n\t\t});\n\t\tif (runnable === undefined) {\n\t\t\tthrow new UsageError(\"LLM client must support function calling or tool use.\");\n\t\t}\n\n\t\tconst responseMessage = await runnable.invoke(this.messages);\n\t\tthis.messages.push(responseMessage);\n\n\t\tif (responseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0) {\n\t\t\tfor (const toolCall of responseMessage.tool_calls) {\n\t\t\t\tswitch (toolCall.name) {\n\t\t\t\t\tcase editingTool.name: {\n\t\t\t\t\t\tconst toolResult = await editingTool.invoke(toolCall);\n\t\t\t\t\t\tthis.messages.push(toolResult);\n\t\t\t\t\t\tconst editResult: unknown = JSON.parse(toolResult.text);\n\t\t\t\t\t\tif (isEditResult(editResult) && editResult.type === \"tooManyEditsError\") {\n\t\t\t\t\t\t\treturn editResult.message;\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn this.queryEdit(edit);\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthis.messages.push(new HumanMessage(`Unrecognized tool call: ${toolCall.name}`));\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn responseMessage.text;\n\t}\n}\n\n/**\n * Type guard for {@link EditResult}.\n */\nfunction isEditResult(value: unknown): value is EditResult {\n\tif (value === null || typeof value !== \"object\") {\n\t\treturn false;\n\t}\n\treturn (\n\t\ttypeof (value as EditResult).type === \"string\" &&\n\t\ttypeof (value as EditResult).message === \"string\"\n\t);\n}\n\n// #endregion\n"]}

package/dist/index.d.ts

@@ -7,5 +7,5 @@
  *
  * @packageDocumentation
  */
-export { createLangchainChatModel } from "./chatModel.js";
+export { createLangchainChatModel, createLegacyLangchainChatModel } from "./chatModel.js";
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAE,8BAA8B,EAAE,MAAM,gBAAgB,CAAC"}

package/dist/index.js

@@ -4,7 +4,7 @@
  * Licensed under the MIT License.
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.createLangchainChatModel = void 0;
+exports.createLegacyLangchainChatModel = exports.createLangchainChatModel = void 0;
 /**
  * LangChain integration helpers for the {@link @fluidframework/tree-agent#SharedTreeSemanticAgent | SharedTreeSemanticAgent}.
  *
@@ -12,4 +12,5 @@ exports.createLangchainChatModel = void 0;
  */
 var chatModel_js_1 = require("./chatModel.js");
 Object.defineProperty(exports, "createLangchainChatModel", { enumerable: true, get: function () { return chatModel_js_1.createLangchainChatModel; } });
+Object.defineProperty(exports, "createLegacyLangchainChatModel", { enumerable: true, get: function () { return chatModel_js_1.createLegacyLangchainChatModel; } });
 //# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH;;;;GAIG;AAEH,+CAA0D;AAAjD,wHAAA,wBAAwB,OAAA","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\n/**\n * LangChain integration helpers for the {@link @fluidframework/tree-agent#SharedTreeSemanticAgent | SharedTreeSemanticAgent}.\n *\n * @packageDocumentation\n */\n\nexport { createLangchainChatModel } from \"./chatModel.js\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AAAA;;;GAGG;;;AAEH;;;;GAIG;AAEH,+CAA0F;AAAjF,wHAAA,wBAAwB,OAAA;AAAE,8HAAA,8BAA8B,OAAA","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\n/**\n * LangChain integration helpers for the {@link @fluidframework/tree-agent#SharedTreeSemanticAgent | SharedTreeSemanticAgent}.\n *\n * @packageDocumentation\n */\n\nexport { createLangchainChatModel, createLegacyLangchainChatModel } from \"./chatModel.js\";\n"]}

package/lib/alpha.d.ts

@@ -16,6 +16,7 @@
 
 export {
 	// #region @alpha APIs
-	createLangchainChatModel
+	createLangchainChatModel, 
+	createLegacyLangchainChatModel
 	// #endregion
 } from "./index.js";

package/lib/chatModel.d.ts

@@ -5,9 +5,21 @@
 import type { SharedTreeChatModel } from "@fluidframework/tree-agent/alpha";
 import type { BaseChatModel } from "@langchain/core/language_models/chat_models";
 /**
- * Creates a `SharedTreeChatModel` that uses the LangChain library to connect to the underlying LLM.
+ * Creates a stateless {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks Use with {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent}.
  * @param langchainModel - The LangChain chat model to use.
  * @alpha
  */
 export declare function createLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel;
+/**
+ * Creates a legacy stateful {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks This implementation maintains internal message history and manages the edit loop via the
+ * {@link @fluidframework/tree-agent#SharedTreeChatQuery.edit | edit} callback pattern.
+ * Use with {@link @fluidframework/tree-agent#SharedTreeSemanticAgent}.
+ * @param langchainModel - The LangChain chat model to use.
+ * @deprecated Use {@link createLangchainChatModel} with
+ * {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent} instead.
+ * @alpha
+ */
+export declare function createLegacyLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel;
 //# sourceMappingURL=chatModel.d.ts.map

package/lib/chatModel.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"chatModel.d.ts","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAKH,OAAO,KAAK,EAEX,mBAAmB,EAEnB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6CAA6C,CAAC;AAKjF;;;;GAIG;AACH,wBAAgB,wBAAwB,CAAC,cAAc,EAAE,aAAa,GAAG,mBAAmB,CAE3F"}
+{"version":3,"file":"chatModel.d.ts","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,OAAO,KAAK,EAEX,mBAAmB,EAInB,MAAM,kCAAkC,CAAC;AAC1C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,6CAA6C,CAAC;AAOjF;;;;;GAKG;AACH,wBAAgB,wBAAwB,CAAC,cAAc,EAAE,aAAa,GAAG,mBAAmB,CAE3F;AAuHD;;;;;;;;;GASG;AACH,wBAAgB,8BAA8B,CAC7C,cAAc,EAAE,aAAa,GAC3B,mBAAmB,CAErB"}

package/lib/chatModel.js

@@ -2,19 +2,132 @@
  * Copyright (c) Microsoft Corporation and contributors. All rights reserved.
  * Licensed under the MIT License.
  */
-/* eslint-disable import-x/no-internal-modules */
 import { UsageError } from "@fluidframework/telemetry-utils/internal";
-import { HumanMessage, SystemMessage } from "@langchain/core/messages";
-import { tool } from "@langchain/core/tools";
+import { AIMessage, HumanMessage, SystemMessage, ToolMessage } from "@langchain/core/messages"; // eslint-disable-line import-x/no-internal-modules
+import { tool } from "@langchain/core/tools"; // eslint-disable-line import-x/no-internal-modules
+// #region New stateless implementation
 /**
- * Creates a `SharedTreeChatModel` that uses the LangChain library to connect to the underlying LLM.
+ * Creates a stateless {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks Use with {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent}.
  * @param langchainModel - The LangChain chat model to use.
  * @alpha
  */
 export function createLangchainChatModel(langchainModel) {
     return new LangchainChatModel(langchainModel);
 }
+/**
+ * Stateless LangChain adapter for {@link @fluidframework/tree-agent#SharedTreeChatModel}.
+ * @remarks This class does not maintain internal message history. All context is provided
+ * via the `history` parameter to {@link LangchainChatModel.invoke}.
+ */
 class LangchainChatModel {
+    constructor(model) {
+        this.model = model;
+        this.editToolName = "GenerateTreeEditingCode";
+    }
+    get name() {
+        const name = this.model.metadata?.modelName;
+        return typeof name === "string" ? name : undefined;
+    }
+    async invoke(history) {
+        // Convert TreeAgentChatMessage[] to LangChain BaseMessage[]
+        const messages = convertToLangchainMessages(history);
+        // Create a placeholder tool definition so the LLM knows the tool exists.
+        // The actual execution is handled by the agent's edit loop — this tool
+        // is never directly invoked, it only tells the LLM the tool signature.
+        const editingTool = tool(async (js) => js, {
+            name: this.editToolName,
+            description: "Invokes a JavaScript code snippet to edit a tree of application data.",
+        });
+        const runnable = this.model.bindTools?.([editingTool], {
+            tool_choice: "auto",
+        });
+        if (runnable === undefined) {
+            throw new UsageError("LLM client must support function calling or tool use.");
+        }
+        const responseMessage = await runnable.invoke(messages);
+        // Parse the response into TreeAgentChatResponse.
+        // Return the first tool call as a tool_call message, preserving the raw args.
+        // Arg parsing (extracting code) is the agent's responsibility.
+        const firstToolCall = responseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0
+            ? responseMessage.tool_calls[0]
+            : undefined;
+        if (firstToolCall !== undefined) {
+            return {
+                role: "tool_call",
+                toolCallId: firstToolCall.id,
+                toolName: firstToolCall.name,
+                toolArgs: firstToolCall.args,
+            };
+        }
+        const content = typeof responseMessage.text === "string"
+            ? responseMessage.text
+            : typeof responseMessage.content === "string"
+                ? responseMessage.content
+                : JSON.stringify(responseMessage.content);
+        return { role: "assistant", content };
+    }
+}
+/**
+ * Converts an array of {@link TreeAgentChatMessage} to LangChain {@link BaseMessage} format.
+ */
+function convertToLangchainMessages(history) {
+    const messages = [];
+    for (const msg of history) {
+        switch (msg.role) {
+            case "system": {
+                messages.push(new SystemMessage(msg.content));
+                break;
+            }
+            case "user": {
+                messages.push(new HumanMessage(msg.content));
+                break;
+            }
+            case "assistant": {
+                messages.push(new AIMessage(msg.content));
+                break;
+            }
+            case "tool_call": {
+                messages.push(new AIMessage({
+                    content: "",
+                    tool_calls: [
+                        {
+                            id: msg.toolCallId,
+                            name: msg.toolName,
+                            args: msg.toolArgs,
+                        },
+                    ],
+                }));
+                break;
+            }
+            case "tool_result": {
+                messages.push(new ToolMessage({
+                    content: msg.content,
+                    tool_call_id: msg.toolCallId ?? "",
+                }));
+                break;
+            }
+            // No default
+        }
+    }
+    return messages;
+}
+// #endregion
+// #region Legacy stateful implementation
+/**
+ * Creates a legacy stateful {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks This implementation maintains internal message history and manages the edit loop via the
+ * {@link @fluidframework/tree-agent#SharedTreeChatQuery.edit | edit} callback pattern.
+ * Use with {@link @fluidframework/tree-agent#SharedTreeSemanticAgent}.
+ * @param langchainModel - The LangChain chat model to use.
+ * @deprecated Use {@link createLangchainChatModel} with
+ * {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent} instead.
+ * @alpha
+ */
+export function createLegacyLangchainChatModel(langchainModel) {
+    return new LegacyLangchainChatModel(langchainModel);
+}
+class LegacyLangchainChatModel {
     constructor(model) {
         this.model = model;
         this.messages = [];
@@ -27,6 +140,7 @@ class LangchainChatModel {
     appendContext(text) {
         this.messages.push(new SystemMessage(text));
     }
+    // eslint-disable-next-line import-x/no-deprecated
     async query(query) {
         this.messages.push(new HumanMessage(query.text));
         return this.queryEdit(async (js) => query.edit(js));
@@ -36,7 +150,9 @@ class LangchainChatModel {
             name: this.editToolName,
             description: "Invokes a JavaScript code snippet to edit a tree of application data.",
         });
-        const runnable = this.model.bindTools?.([editingTool], { tool_choice: "auto" });
+        const runnable = this.model.bindTools?.([editingTool], {
+            tool_choice: "auto",
+        });
         if (runnable === undefined) {
             throw new UsageError("LLM client must support function calling or tool use.");
         }
@@ -73,4 +189,5 @@ function isEditResult(value) {
     return (typeof value.type === "string" &&
         typeof value.message === "string");
 }
+// #endregion
 //# sourceMappingURL=chatModel.js.map

package/lib/chatModel.js.map

@@ -1 +1 @@
-{"version":3,"file":"chatModel.js","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,iDAAiD;AAEjD,OAAO,EAAE,UAAU,EAAE,MAAM,0CAA0C,CAAC;AAQtE,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AACvE,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAC;AAE7C;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CAAC,cAA6B;IACrE,OAAO,IAAI,kBAAkB,CAAC,cAAc,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,kBAAkB;IAGvB,YAAoC,KAAoB;QAApB,UAAK,GAAL,KAAK,CAAe;QAFvC,aAAQ,GAAkB,EAAE,CAAC;QAI9B,iBAAY,GAAG,yBAAyB,CAAC;IAFE,CAAC;IAI5D,IAAW,IAAI;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC;QAC5C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAEM,aAAa,CAAC,IAAY;QAChC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7C,CAAC;IAEM,KAAK,CAAC,KAAK,CAAC,KAA0B;QAC5C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,EAAU,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,IAAiC;QACxD,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE;YAC9B,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,WAAW,EAAE,uEAAuE;SACpF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE,EAAE,WAAW,EAAE,MAAM,EAAE,CAAC,CAAC;QAChF,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,UAAU,CAAC,uDAAuD,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAEpC,IAAI,eAAe,CAAC,UAAU,KAAK,SAAS,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvF,KAAK,MAAM,QAAQ,IAAI,eAAe,CAAC,UAAU,EAAE,CAAC;gBACnD,QAAQ,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACvB,KAAK,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;wBACtD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;wBAC/B,MAAM,UAAU,GAAY,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;wBACxD,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;4BACzE,OAAO,UAAU,CAAC,OAAO,CAAC;wBAC3B,CAAC;wBACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC7B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,2BAA2B,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,eAAe,CAAC,IAAI,CAAC;IAC7B,CAAC;CACD;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,KAAc;IACnC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACjD,OAAO,KAAK,CAAC;IACd,CAAC;IACD,OAAO,CACN,OAAQ,KAAoB,CAAC,IAAI,KAAK,QAAQ;QAC9C,OAAQ,KAAoB,CAAC,OAAO,KAAK,QAAQ,CACjD,CAAC;AACH,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\n/* eslint-disable import-x/no-internal-modules */\n\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport type {\n\tEditResult,\n\tSharedTreeChatModel,\n\tSharedTreeChatQuery,\n} from \"@fluidframework/tree-agent/alpha\";\nimport type { BaseChatModel } from \"@langchain/core/language_models/chat_models\";\nimport type { BaseMessage } from \"@langchain/core/messages\";\nimport { HumanMessage, SystemMessage } from \"@langchain/core/messages\";\nimport { tool } from \"@langchain/core/tools\";\n\n/**\n * Creates a `SharedTreeChatModel` that uses the LangChain library to connect to the underlying LLM.\n * @param langchainModel - The LangChain chat model to use.\n * @alpha\n */\nexport function createLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel {\n\treturn new LangchainChatModel(langchainModel);\n}\n\nclass LangchainChatModel implements SharedTreeChatModel {\n\tprivate readonly messages: BaseMessage[] = [];\n\n\tpublic constructor(private readonly model: BaseChatModel) {}\n\n\tpublic readonly editToolName = \"GenerateTreeEditingCode\";\n\n\tpublic get name(): string | undefined {\n\t\tconst name = this.model.metadata?.modelName;\n\t\treturn typeof name === \"string\" ? name : undefined;\n\t}\n\n\tpublic appendContext(text: string): void {\n\t\tthis.messages.push(new SystemMessage(text));\n\t}\n\n\tpublic async query(query: SharedTreeChatQuery): Promise<string> {\n\t\tthis.messages.push(new HumanMessage(query.text));\n\t\treturn this.queryEdit(async (js: string) => query.edit(js));\n\t}\n\n\tprivate async queryEdit(edit: SharedTreeChatQuery[\"edit\"]): Promise<string> {\n\t\tconst editingTool = tool(edit, {\n\t\t\tname: this.editToolName,\n\t\t\tdescription: \"Invokes a JavaScript code snippet to edit a tree of application data.\",\n\t\t});\n\t\tconst runnable = this.model.bindTools?.([editingTool], { tool_choice: \"auto\" });\n\t\tif (runnable === undefined) {\n\t\t\tthrow new UsageError(\"LLM client must support function calling or tool use.\");\n\t\t}\n\n\t\tconst responseMessage = await runnable.invoke(this.messages);\n\t\tthis.messages.push(responseMessage);\n\n\t\tif (responseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0) {\n\t\t\tfor (const toolCall of responseMessage.tool_calls) {\n\t\t\t\tswitch (toolCall.name) {\n\t\t\t\t\tcase editingTool.name: {\n\t\t\t\t\t\tconst toolResult = await editingTool.invoke(toolCall);\n\t\t\t\t\t\tthis.messages.push(toolResult);\n\t\t\t\t\t\tconst editResult: unknown = JSON.parse(toolResult.text);\n\t\t\t\t\t\tif (isEditResult(editResult) && editResult.type === \"tooManyEditsError\") {\n\t\t\t\t\t\t\treturn editResult.message;\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn this.queryEdit(edit);\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthis.messages.push(new HumanMessage(`Unrecognized tool call: ${toolCall.name}`));\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn responseMessage.text;\n\t}\n}\n\n/**\n * Type guard for {@link EditResult}.\n */\nfunction isEditResult(value: unknown): value is EditResult {\n\tif (value === null || typeof value !== \"object\") {\n\t\treturn false;\n\t}\n\treturn (\n\t\ttypeof (value as EditResult).type === \"string\" &&\n\t\ttypeof (value as EditResult).message === \"string\"\n\t);\n}\n"]}
+{"version":3,"file":"chatModel.js","sourceRoot":"","sources":["../src/chatModel.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,0CAA0C,CAAC;AAUtE,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,aAAa,EAAE,WAAW,EAAE,MAAM,0BAA0B,CAAC,CAAC,mDAAmD;AACnJ,OAAO,EAAE,IAAI,EAAE,MAAM,uBAAuB,CAAC,CAAC,mDAAmD;AAEjG,uCAAuC;AAEvC;;;;;GAKG;AACH,MAAM,UAAU,wBAAwB,CAAC,cAA6B;IACrE,OAAO,IAAI,kBAAkB,CAAC,cAAc,CAAC,CAAC;AAC/C,CAAC;AAED;;;;GAIG;AACH,MAAM,kBAAkB;IAGvB,YAAoC,KAAoB;QAApB,UAAK,GAAL,KAAK,CAAe;QAFxC,iBAAY,GAAG,yBAAyB,CAAC;IAEE,CAAC;IAE5D,IAAW,IAAI;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC;QAC5C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAEM,KAAK,CAAC,MAAM,CAClB,OAAwC;QAExC,4DAA4D;QAC5D,MAAM,QAAQ,GAAkB,0BAA0B,CAAC,OAAO,CAAC,CAAC;QAEpE,yEAAyE;QACzE,uEAAuE;QACvE,uEAAuE;QACvE,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,EAAE,EAAU,EAAE,EAAE,CAAC,EAAE,EAAE;YAClD,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,WAAW,EAAE,uEAAuE;SACpF,CAAC,CAAC;QAEH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE;YACtD,WAAW,EAAE,MAAM;SACnB,CAAC,CAAC;QACH,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,UAAU,CAAC,uDAAuD,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAExD,iDAAiD;QACjD,8EAA8E;QAC9E,+DAA+D;QAC/D,MAAM,aAAa,GAClB,eAAe,CAAC,UAAU,KAAK,SAAS,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC;YAChF,CAAC,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC;YAC/B,CAAC,CAAC,SAAS,CAAC;QACd,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;YACjC,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,UAAU,EAAE,aAAa,CAAC,EAAE;gBAC5B,QAAQ,EAAE,aAAa,CAAC,IAAI;gBAC5B,QAAQ,EAAE,aAAa,CAAC,IAAI;aAC5B,CAAC;QACH,CAAC;QAED,MAAM,OAAO,GACZ,OAAO,eAAe,CAAC,IAAI,KAAK,QAAQ;YACvC,CAAC,CAAC,eAAe,CAAC,IAAI;YACtB,CAAC,CAAC,OAAO,eAAe,CAAC,OAAO,KAAK,QAAQ;gBAC5C,CAAC,CAAC,eAAe,CAAC,OAAO;gBACzB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;QAC7C,OAAO,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;IACvC,CAAC;CACD;AAED;;GAEG;AACH,SAAS,0BAA0B,CAAC,OAAwC;IAC3E,MAAM,QAAQ,GAAkB,EAAE,CAAC;IACnC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;QAC3B,QAAQ,GAAG,CAAC,IAAI,EAAE,CAAC;YAClB,KAAK,QAAQ,CAAC,CAAC,CAAC;gBACf,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC9C,MAAM;YACP,CAAC;YACD,KAAK,MAAM,CAAC,CAAC,CAAC;gBACb,QAAQ,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC7C,MAAM;YACP,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBAClB,QAAQ,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;gBAC1C,MAAM;YACP,CAAC;YACD,KAAK,WAAW,CAAC,CAAC,CAAC;gBAClB,QAAQ,CAAC,IAAI,CACZ,IAAI,SAAS,CAAC;oBACb,OAAO,EAAE,EAAE;oBACX,UAAU,EAAE;wBACX;4BACC,EAAE,EAAE,GAAG,CAAC,UAAU;4BAClB,IAAI,EAAE,GAAG,CAAC,QAAQ;4BAClB,IAAI,EAAE,GAAG,CAAC,QAAQ;yBAClB;qBACD;iBACD,CAAC,CACF,CAAC;gBACF,MAAM;YACP,CAAC;YACD,KAAK,aAAa,CAAC,CAAC,CAAC;gBACpB,QAAQ,CAAC,IAAI,CACZ,IAAI,WAAW,CAAC;oBACf,OAAO,EAAE,GAAG,CAAC,OAAO;oBACpB,YAAY,EAAE,GAAG,CAAC,UAAU,IAAI,EAAE;iBAClC,CAAC,CACF,CAAC;gBACF,MAAM;YACP,CAAC;YACD,aAAa;QACd,CAAC;IACF,CAAC;IACD,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,aAAa;AAEb,yCAAyC;AAEzC;;;;;;;;;GASG;AACH,MAAM,UAAU,8BAA8B,CAC7C,cAA6B;IAE7B,OAAO,IAAI,wBAAwB,CAAC,cAAc,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,wBAAwB;IAG7B,YAAoC,KAAoB;QAApB,UAAK,GAAL,KAAK,CAAe;QAFvC,aAAQ,GAAkB,EAAE,CAAC;QAI9B,iBAAY,GAAG,yBAAyB,CAAC;IAFE,CAAC;IAI5D,IAAW,IAAI;QACd,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,SAAS,CAAC;QAC5C,OAAO,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACpD,CAAC;IAEM,aAAa,CAAC,IAAY;QAChC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC;IAC7C,CAAC;IAED,kDAAkD;IAC3C,KAAK,CAAC,KAAK,CAAC,KAA0B;QAC5C,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QACjD,OAAO,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,EAAU,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;IAC7D,CAAC;IAEO,KAAK,CAAC,SAAS,CACtB,IAAiC;QAEjC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE;YAC9B,IAAI,EAAE,IAAI,CAAC,YAAY;YACvB,WAAW,EAAE,uEAAuE;SACpF,CAAC,CAAC;QACH,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC,WAAW,CAAC,EAAE;YACtD,WAAW,EAAE,MAAM;SACnB,CAAC,CAAC;QACH,IAAI,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC5B,MAAM,IAAI,UAAU,CAAC,uDAAuD,CAAC,CAAC;QAC/E,CAAC;QAED,MAAM,eAAe,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;QAEpC,IAAI,eAAe,CAAC,UAAU,KAAK,SAAS,IAAI,eAAe,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACvF,KAAK,MAAM,QAAQ,IAAI,eAAe,CAAC,UAAU,EAAE,CAAC;gBACnD,QAAQ,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACvB,KAAK,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;wBACvB,MAAM,UAAU,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;wBACtD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;wBAC/B,MAAM,UAAU,GAAY,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;wBACxD,IAAI,YAAY,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;4BACzE,OAAO,UAAU,CAAC,OAAO,CAAC;wBAC3B,CAAC;wBACD,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;oBAC7B,CAAC;oBACD,OAAO,CAAC,CAAC,CAAC;wBACT,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,2BAA2B,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;oBAClF,CAAC;gBACF,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,eAAe,CAAC,IAAI,CAAC;IAC7B,CAAC;CACD;AAED;;GAEG;AACH,SAAS,YAAY,CAAC,KAAc;IACnC,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QACjD,OAAO,KAAK,CAAC;IACd,CAAC;IACD,OAAO,CACN,OAAQ,KAAoB,CAAC,IAAI,KAAK,QAAQ;QAC9C,OAAQ,KAAoB,CAAC,OAAO,KAAK,QAAQ,CACjD,CAAC;AACH,CAAC;AAED,aAAa","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\nimport { UsageError } from \"@fluidframework/telemetry-utils/internal\";\nimport type {\n\tEditResult,\n\tSharedTreeChatModel,\n\tTreeAgentChatMessage,\n\tTreeAgentChatResponse,\n\tSharedTreeChatQuery, // eslint-disable-line import-x/no-deprecated\n} from \"@fluidframework/tree-agent/alpha\";\nimport type { BaseChatModel } from \"@langchain/core/language_models/chat_models\"; // eslint-disable-line import-x/no-internal-modules\nimport type { BaseMessage } from \"@langchain/core/messages\"; // eslint-disable-line import-x/no-internal-modules\nimport { AIMessage, HumanMessage, SystemMessage, ToolMessage } from \"@langchain/core/messages\"; // eslint-disable-line import-x/no-internal-modules\nimport { tool } from \"@langchain/core/tools\"; // eslint-disable-line import-x/no-internal-modules\n\n// #region New stateless implementation\n\n/**\n * Creates a stateless {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.\n * @remarks Use with {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent}.\n * @param langchainModel - The LangChain chat model to use.\n * @alpha\n */\nexport function createLangchainChatModel(langchainModel: BaseChatModel): SharedTreeChatModel {\n\treturn new LangchainChatModel(langchainModel);\n}\n\n/**\n * Stateless LangChain adapter for {@link @fluidframework/tree-agent#SharedTreeChatModel}.\n * @remarks This class does not maintain internal message history. All context is provided\n * via the `history` parameter to {@link LangchainChatModel.invoke}.\n */\nclass LangchainChatModel implements SharedTreeChatModel {\n\tpublic readonly editToolName = \"GenerateTreeEditingCode\";\n\n\tpublic constructor(private readonly model: BaseChatModel) {}\n\n\tpublic get name(): string | undefined {\n\t\tconst name = this.model.metadata?.modelName;\n\t\treturn typeof name === \"string\" ? name : undefined;\n\t}\n\n\tpublic async invoke(\n\t\thistory: readonly TreeAgentChatMessage[],\n\t): Promise<TreeAgentChatResponse> {\n\t\t// Convert TreeAgentChatMessage[] to LangChain BaseMessage[]\n\t\tconst messages: BaseMessage[] = convertToLangchainMessages(history);\n\n\t\t// Create a placeholder tool definition so the LLM knows the tool exists.\n\t\t// The actual execution is handled by the agent's edit loop — this tool\n\t\t// is never directly invoked, it only tells the LLM the tool signature.\n\t\tconst editingTool = tool(async (js: string) => js, {\n\t\t\tname: this.editToolName,\n\t\t\tdescription: \"Invokes a JavaScript code snippet to edit a tree of application data.\",\n\t\t});\n\n\t\tconst runnable = this.model.bindTools?.([editingTool], {\n\t\t\ttool_choice: \"auto\",\n\t\t});\n\t\tif (runnable === undefined) {\n\t\t\tthrow new UsageError(\"LLM client must support function calling or tool use.\");\n\t\t}\n\n\t\tconst responseMessage = await runnable.invoke(messages);\n\n\t\t// Parse the response into TreeAgentChatResponse.\n\t\t// Return the first tool call as a tool_call message, preserving the raw args.\n\t\t// Arg parsing (extracting code) is the agent's responsibility.\n\t\tconst firstToolCall =\n\t\t\tresponseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0\n\t\t\t\t? responseMessage.tool_calls[0]\n\t\t\t\t: undefined;\n\t\tif (firstToolCall !== undefined) {\n\t\t\treturn {\n\t\t\t\trole: \"tool_call\",\n\t\t\t\ttoolCallId: firstToolCall.id,\n\t\t\t\ttoolName: firstToolCall.name,\n\t\t\t\ttoolArgs: firstToolCall.args,\n\t\t\t};\n\t\t}\n\n\t\tconst content =\n\t\t\ttypeof responseMessage.text === \"string\"\n\t\t\t\t? responseMessage.text\n\t\t\t\t: typeof responseMessage.content === \"string\"\n\t\t\t\t\t? responseMessage.content\n\t\t\t\t\t: JSON.stringify(responseMessage.content);\n\t\treturn { role: \"assistant\", content };\n\t}\n}\n\n/**\n * Converts an array of {@link TreeAgentChatMessage} to LangChain {@link BaseMessage} format.\n */\nfunction convertToLangchainMessages(history: readonly TreeAgentChatMessage[]): BaseMessage[] {\n\tconst messages: BaseMessage[] = [];\n\tfor (const msg of history) {\n\t\tswitch (msg.role) {\n\t\t\tcase \"system\": {\n\t\t\t\tmessages.push(new SystemMessage(msg.content));\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"user\": {\n\t\t\t\tmessages.push(new HumanMessage(msg.content));\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"assistant\": {\n\t\t\t\tmessages.push(new AIMessage(msg.content));\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"tool_call\": {\n\t\t\t\tmessages.push(\n\t\t\t\t\tnew AIMessage({\n\t\t\t\t\t\tcontent: \"\",\n\t\t\t\t\t\ttool_calls: [\n\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\tid: msg.toolCallId,\n\t\t\t\t\t\t\t\tname: msg.toolName,\n\t\t\t\t\t\t\t\targs: msg.toolArgs,\n\t\t\t\t\t\t\t},\n\t\t\t\t\t\t],\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\tcase \"tool_result\": {\n\t\t\t\tmessages.push(\n\t\t\t\t\tnew ToolMessage({\n\t\t\t\t\t\tcontent: msg.content,\n\t\t\t\t\t\ttool_call_id: msg.toolCallId ?? \"\",\n\t\t\t\t\t}),\n\t\t\t\t);\n\t\t\t\tbreak;\n\t\t\t}\n\t\t\t// No default\n\t\t}\n\t}\n\treturn messages;\n}\n\n// #endregion\n\n// #region Legacy stateful implementation\n\n/**\n * Creates a legacy stateful {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.\n * @remarks This implementation maintains internal message history and manages the edit loop via the\n * {@link @fluidframework/tree-agent#SharedTreeChatQuery.edit | edit} callback pattern.\n * Use with {@link @fluidframework/tree-agent#SharedTreeSemanticAgent}.\n * @param langchainModel - The LangChain chat model to use.\n * @deprecated Use {@link createLangchainChatModel} with\n * {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent} instead.\n * @alpha\n */\nexport function createLegacyLangchainChatModel(\n\tlangchainModel: BaseChatModel,\n): SharedTreeChatModel {\n\treturn new LegacyLangchainChatModel(langchainModel);\n}\n\nclass LegacyLangchainChatModel implements SharedTreeChatModel {\n\tprivate readonly messages: BaseMessage[] = [];\n\n\tpublic constructor(private readonly model: BaseChatModel) {}\n\n\tpublic readonly editToolName = \"GenerateTreeEditingCode\";\n\n\tpublic get name(): string | undefined {\n\t\tconst name = this.model.metadata?.modelName;\n\t\treturn typeof name === \"string\" ? name : undefined;\n\t}\n\n\tpublic appendContext(text: string): void {\n\t\tthis.messages.push(new SystemMessage(text));\n\t}\n\n\t// eslint-disable-next-line import-x/no-deprecated\n\tpublic async query(query: SharedTreeChatQuery): Promise<string> {\n\t\tthis.messages.push(new HumanMessage(query.text));\n\t\treturn this.queryEdit(async (js: string) => query.edit(js));\n\t}\n\n\tprivate async queryEdit(\n\t\tedit: SharedTreeChatQuery[\"edit\"], // eslint-disable-line import-x/no-deprecated\n\t): Promise<string> {\n\t\tconst editingTool = tool(edit, {\n\t\t\tname: this.editToolName,\n\t\t\tdescription: \"Invokes a JavaScript code snippet to edit a tree of application data.\",\n\t\t});\n\t\tconst runnable = this.model.bindTools?.([editingTool], {\n\t\t\ttool_choice: \"auto\",\n\t\t});\n\t\tif (runnable === undefined) {\n\t\t\tthrow new UsageError(\"LLM client must support function calling or tool use.\");\n\t\t}\n\n\t\tconst responseMessage = await runnable.invoke(this.messages);\n\t\tthis.messages.push(responseMessage);\n\n\t\tif (responseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0) {\n\t\t\tfor (const toolCall of responseMessage.tool_calls) {\n\t\t\t\tswitch (toolCall.name) {\n\t\t\t\t\tcase editingTool.name: {\n\t\t\t\t\t\tconst toolResult = await editingTool.invoke(toolCall);\n\t\t\t\t\t\tthis.messages.push(toolResult);\n\t\t\t\t\t\tconst editResult: unknown = JSON.parse(toolResult.text);\n\t\t\t\t\t\tif (isEditResult(editResult) && editResult.type === \"tooManyEditsError\") {\n\t\t\t\t\t\t\treturn editResult.message;\n\t\t\t\t\t\t}\n\t\t\t\t\t\treturn this.queryEdit(edit);\n\t\t\t\t\t}\n\t\t\t\t\tdefault: {\n\t\t\t\t\t\tthis.messages.push(new HumanMessage(`Unrecognized tool call: ${toolCall.name}`));\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\treturn responseMessage.text;\n\t}\n}\n\n/**\n * Type guard for {@link EditResult}.\n */\nfunction isEditResult(value: unknown): value is EditResult {\n\tif (value === null || typeof value !== \"object\") {\n\t\treturn false;\n\t}\n\treturn (\n\t\ttypeof (value as EditResult).type === \"string\" &&\n\t\ttypeof (value as EditResult).message === \"string\"\n\t);\n}\n\n// #endregion\n"]}

package/lib/index.d.ts

@@ -7,5 +7,5 @@
  *
  * @packageDocumentation
  */
-export { createLangchainChatModel } from "./chatModel.js";
+export { createLangchainChatModel, createLegacyLangchainChatModel } from "./chatModel.js";
 //# sourceMappingURL=index.d.ts.map

package/lib/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAE,8BAA8B,EAAE,MAAM,gBAAgB,CAAC"}

package/lib/index.js

@@ -7,5 +7,5 @@
  *
  * @packageDocumentation
  */
-export { createLangchainChatModel } from "./chatModel.js";
+export { createLangchainChatModel, createLegacyLangchainChatModel } from "./chatModel.js";
 //# sourceMappingURL=index.js.map

package/lib/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\n/**\n * LangChain integration helpers for the {@link @fluidframework/tree-agent#SharedTreeSemanticAgent | SharedTreeSemanticAgent}.\n *\n * @packageDocumentation\n */\n\nexport { createLangchainChatModel } from \"./chatModel.js\";\n"]}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAE,8BAA8B,EAAE,MAAM,gBAAgB,CAAC","sourcesContent":["/*!\n * Copyright (c) Microsoft Corporation and contributors. All rights reserved.\n * Licensed under the MIT License.\n */\n\n/**\n * LangChain integration helpers for the {@link @fluidframework/tree-agent#SharedTreeSemanticAgent | SharedTreeSemanticAgent}.\n *\n * @packageDocumentation\n */\n\nexport { createLangchainChatModel, createLegacyLangchainChatModel } from \"./chatModel.js\";\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@fluidframework/tree-agent-langchain",
-  "version": "2.90.0",
+  "version": "2.91.0",
   "description": "LangChain integration helpers for @fluidframework/tree-agent",
   "homepage": "https://fluidframework.com",
   "repository": {
@@ -69,21 +69,21 @@
     "temp-directory": "nyc/.nyc_output"
   },
   "dependencies": {
-    "@fluidframework/telemetry-utils": "~2.90.0",
-    "@fluidframework/tree-agent": "~2.90.0",
+    "@fluidframework/telemetry-utils": "~2.91.0",
+    "@fluidframework/tree-agent": "~2.91.0",
     "@langchain/core": "^0.3.80"
   },
   "devDependencies": {
     "@arethetypeswrong/cli": "^0.18.2",
-    "@biomejs/biome": "~1.9.3",
-    "@fluid-internal/mocha-test-setup": "~2.90.0",
+    "@biomejs/biome": "~2.4.5",
+    "@fluid-internal/mocha-test-setup": "~2.91.0",
     "@fluid-tools/build-cli": "^0.63.0",
     "@fluidframework/build-common": "^2.0.3",
     "@fluidframework/build-tools": "^0.63.0",
-    "@fluidframework/core-utils": "~2.90.0",
-    "@fluidframework/eslint-config-fluid": "~2.90.0",
-    "@fluidframework/runtime-utils": "~2.90.0",
-    "@fluidframework/tree": "~2.90.0",
+    "@fluidframework/core-utils": "~2.91.0",
+    "@fluidframework/eslint-config-fluid": "~2.91.0",
+    "@fluidframework/runtime-utils": "~2.91.0",
+    "@fluidframework/tree": "~2.91.0",
     "@langchain/anthropic": "^0.3.24",
     "@langchain/google-genai": "^0.2.16",
     "@langchain/openai": "^0.6.12",

package/src/chatModel.ts

@@ -3,21 +3,24 @@
  * Licensed under the MIT License.
  */
 
-/* eslint-disable import-x/no-internal-modules */
-
 import { UsageError } from "@fluidframework/telemetry-utils/internal";
 import type {
 	EditResult,
 	SharedTreeChatModel,
-	SharedTreeChatQuery,
+	TreeAgentChatMessage,
+	TreeAgentChatResponse,
+	SharedTreeChatQuery, // eslint-disable-line import-x/no-deprecated
 } from "@fluidframework/tree-agent/alpha";
-import type { BaseChatModel } from "@langchain/core/language_models/chat_models";
-import type { BaseMessage } from "@langchain/core/messages";
-import { HumanMessage, SystemMessage } from "@langchain/core/messages";
-import { tool } from "@langchain/core/tools";
+import type { BaseChatModel } from "@langchain/core/language_models/chat_models"; // eslint-disable-line import-x/no-internal-modules
+import type { BaseMessage } from "@langchain/core/messages"; // eslint-disable-line import-x/no-internal-modules
+import { AIMessage, HumanMessage, SystemMessage, ToolMessage } from "@langchain/core/messages"; // eslint-disable-line import-x/no-internal-modules
+import { tool } from "@langchain/core/tools"; // eslint-disable-line import-x/no-internal-modules
+
+// #region New stateless implementation
 
 /**
- * Creates a `SharedTreeChatModel` that uses the LangChain library to connect to the underlying LLM.
+ * Creates a stateless {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks Use with {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent}.
  * @param langchainModel - The LangChain chat model to use.
  * @alpha
  */
@@ -25,7 +28,140 @@ export function createLangchainChatModel(langchainModel: BaseChatModel): SharedT
 	return new LangchainChatModel(langchainModel);
 }
 
+/**
+ * Stateless LangChain adapter for {@link @fluidframework/tree-agent#SharedTreeChatModel}.
+ * @remarks This class does not maintain internal message history. All context is provided
+ * via the `history` parameter to {@link LangchainChatModel.invoke}.
+ */
 class LangchainChatModel implements SharedTreeChatModel {
+	public readonly editToolName = "GenerateTreeEditingCode";
+
+	public constructor(private readonly model: BaseChatModel) {}
+
+	public get name(): string | undefined {
+		const name = this.model.metadata?.modelName;
+		return typeof name === "string" ? name : undefined;
+	}
+
+	public async invoke(
+		history: readonly TreeAgentChatMessage[],
+	): Promise<TreeAgentChatResponse> {
+		// Convert TreeAgentChatMessage[] to LangChain BaseMessage[]
+		const messages: BaseMessage[] = convertToLangchainMessages(history);
+
+		// Create a placeholder tool definition so the LLM knows the tool exists.
+		// The actual execution is handled by the agent's edit loop — this tool
+		// is never directly invoked, it only tells the LLM the tool signature.
+		const editingTool = tool(async (js: string) => js, {
+			name: this.editToolName,
+			description: "Invokes a JavaScript code snippet to edit a tree of application data.",
+		});
+
+		const runnable = this.model.bindTools?.([editingTool], {
+			tool_choice: "auto",
+		});
+		if (runnable === undefined) {
+			throw new UsageError("LLM client must support function calling or tool use.");
+		}
+
+		const responseMessage = await runnable.invoke(messages);
+
+		// Parse the response into TreeAgentChatResponse.
+		// Return the first tool call as a tool_call message, preserving the raw args.
+		// Arg parsing (extracting code) is the agent's responsibility.
+		const firstToolCall =
+			responseMessage.tool_calls !== undefined && responseMessage.tool_calls.length > 0
+				? responseMessage.tool_calls[0]
+				: undefined;
+		if (firstToolCall !== undefined) {
+			return {
+				role: "tool_call",
+				toolCallId: firstToolCall.id,
+				toolName: firstToolCall.name,
+				toolArgs: firstToolCall.args,
+			};
+		}
+
+		const content =
+			typeof responseMessage.text === "string"
+				? responseMessage.text
+				: typeof responseMessage.content === "string"
+					? responseMessage.content
+					: JSON.stringify(responseMessage.content);
+		return { role: "assistant", content };
+	}
+}
+
+/**
+ * Converts an array of {@link TreeAgentChatMessage} to LangChain {@link BaseMessage} format.
+ */
+function convertToLangchainMessages(history: readonly TreeAgentChatMessage[]): BaseMessage[] {
+	const messages: BaseMessage[] = [];
+	for (const msg of history) {
+		switch (msg.role) {
+			case "system": {
+				messages.push(new SystemMessage(msg.content));
+				break;
+			}
+			case "user": {
+				messages.push(new HumanMessage(msg.content));
+				break;
+			}
+			case "assistant": {
+				messages.push(new AIMessage(msg.content));
+				break;
+			}
+			case "tool_call": {
+				messages.push(
+					new AIMessage({
+						content: "",
+						tool_calls: [
+							{
+								id: msg.toolCallId,
+								name: msg.toolName,
+								args: msg.toolArgs,
+							},
+						],
+					}),
+				);
+				break;
+			}
+			case "tool_result": {
+				messages.push(
+					new ToolMessage({
+						content: msg.content,
+						tool_call_id: msg.toolCallId ?? "",
+					}),
+				);
+				break;
+			}
+			// No default
+		}
+	}
+	return messages;
+}
+
+// #endregion
+
+// #region Legacy stateful implementation
+
+/**
+ * Creates a legacy stateful {@link @fluidframework/tree-agent#SharedTreeChatModel} backed by LangChain.
+ * @remarks This implementation maintains internal message history and manages the edit loop via the
+ * {@link @fluidframework/tree-agent#SharedTreeChatQuery.edit | edit} callback pattern.
+ * Use with {@link @fluidframework/tree-agent#SharedTreeSemanticAgent}.
+ * @param langchainModel - The LangChain chat model to use.
+ * @deprecated Use {@link createLangchainChatModel} with
+ * {@link @fluidframework/tree-agent#createTreeAgent | createTreeAgent} instead.
+ * @alpha
+ */
+export function createLegacyLangchainChatModel(
+	langchainModel: BaseChatModel,
+): SharedTreeChatModel {
+	return new LegacyLangchainChatModel(langchainModel);
+}
+
+class LegacyLangchainChatModel implements SharedTreeChatModel {
 	private readonly messages: BaseMessage[] = [];
 
 	public constructor(private readonly model: BaseChatModel) {}
@@ -41,17 +177,22 @@ class LangchainChatModel implements SharedTreeChatModel {
 		this.messages.push(new SystemMessage(text));
 	}
 
+	// eslint-disable-next-line import-x/no-deprecated
 	public async query(query: SharedTreeChatQuery): Promise<string> {
 		this.messages.push(new HumanMessage(query.text));
 		return this.queryEdit(async (js: string) => query.edit(js));
 	}
 
-	private async queryEdit(edit: SharedTreeChatQuery["edit"]): Promise<string> {
+	private async queryEdit(
+		edit: SharedTreeChatQuery["edit"], // eslint-disable-line import-x/no-deprecated
+	): Promise<string> {
 		const editingTool = tool(edit, {
 			name: this.editToolName,
 			description: "Invokes a JavaScript code snippet to edit a tree of application data.",
 		});
-		const runnable = this.model.bindTools?.([editingTool], { tool_choice: "auto" });
+		const runnable = this.model.bindTools?.([editingTool], {
+			tool_choice: "auto",
+		});
 		if (runnable === undefined) {
 			throw new UsageError("LLM client must support function calling or tool use.");
 		}
@@ -94,3 +235,5 @@ function isEditResult(value: unknown): value is EditResult {
 		typeof (value as EditResult).message === "string"
 	);
 }
+
+// #endregion

package/src/index.ts

@@ -9,4 +9,4 @@
  * @packageDocumentation
  */
 
-export { createLangchainChatModel } from "./chatModel.js";
+export { createLangchainChatModel, createLegacyLangchainChatModel } from "./chatModel.js";