Merge branch 'main' into vb/update-get-state

Makefile

@@ -1,6 +1,14 @@
-.PHONY: build-docs serve-docs serve-clean-docs clean-docs codespell
+.PHONY: build-docs serve-docs serve-clean-docs clean-docs codespell build-typedoc
 
-build-docs:
+build-typedoc:
+	cd libs/sdk-js && yarn install --include-dev && yarn typedoc
+	cd libs/sdk-js && yarn --silent concat-md --decrease-title-levels --ignore=js_ts_sdk_ref.md --start-title-level-at 2 docs > ../../docs/docs/cloud/reference/sdk/js_ts_sdk_ref.md 2>/dev/null
+	#  Add links to the monorepo
+	sed -e '1,10s|@langchain/langgraph-sdk|[@langchain/langgraph-sdk](https://github.com/langchain-ai/langgraph/tree/main/libs/sdk-js)|g' docs/docs/cloud/reference/sdk/js_ts_sdk_ref.md > temp_file && mv temp_file docs/docs/cloud/reference/sdk/js_ts_sdk_ref.md
+
+
+
+build-docs: build-typedoc
 	poetry run python docs/_scripts/copy_notebooks.py
 	poetry run python -m mkdocs build --clean -f docs/mkdocs.yml --strict
 
@@ -8,7 +16,7 @@ serve-clean-docs: clean-docs
 	poetry run python docs/_scripts/copy_notebooks.py
 	poetry run python -m mkdocs serve -c -f docs/mkdocs.yml --strict -w ./libs/langgraph
 
-serve-docs:
+serve-docs: build-typedoc
 	poetry run python docs/_scripts/copy_notebooks.py
 	poetry run python -m mkdocs serve -f docs/mkdocs.yml -w ./libs/langgraph --dirty
 

docs/.gitignore

@@ -1,3 +1,4 @@
 *.ipynb
 site/
 docs/tutorials/**/*.png
+docs/cloud/reference/sdk/js_ts_sdk_ref.md

docs/_scripts/copy_notebooks.py

@@ -101,6 +101,7 @@ _HIDE = set(
         "docs/quickstart.ipynb",
         "tutorials/rag-agent-testing.ipynb",
         "tutorials/rag-agent-testing-local.ipynb",
+        "tutorials/tool-calling-agent-local.ipynb",
         "time-travel.ipynb",
         "code_assistant/langgraph_code_assistant_mistral.ipynb",
     ]

libs/langgraph/langgraph/prebuilt/tool_node.py

@@ -1,4 +1,5 @@
 import asyncio
+import json
 from copy import copy
 from typing import (
     Any,
@@ -28,6 +29,16 @@ INVALID_TOOL_NAME_ERROR_TEMPLATE = (
 TOOL_CALL_ERROR_TEMPLATE = "Error: {error}\n Please fix your mistakes."
 
 
+def str_output(output: Any) -> str:
+    if isinstance(output, str):
+        return output
+    else:
+        try:
+            return json.dumps(output)
+        except Exception:
+            return str(output)
+
+
 class ToolNode(RunnableCallable):
     """A node that runs the tools called in the last AIMessage.
 
@@ -94,7 +105,12 @@ class ToolNode(RunnableCallable):
 
         try:
             input = {**call, **{"type": "tool_call"}}
-            return self.tools_by_name[call["name"]].invoke(input, config)
+            tool_message: ToolMessage = self.tools_by_name[call["name"]].invoke(
+                input, config
+            )
+            # TODO: handle this properly in core
+            tool_message.content = str_output(tool_message.content)
+            return tool_message
         except Exception as e:
             if not self.handle_tool_errors:
                 raise e
@@ -106,7 +122,12 @@ class ToolNode(RunnableCallable):
             return invalid_tool_message
         try:
             input = {**call, **{"type": "tool_call"}}
-            return await self.tools_by_name[call["name"]].ainvoke(input, config)
+            tool_message: ToolMessage = await self.tools_by_name[call["name"]].ainvoke(
+                input, config
+            )
+            # TODO: handle this properly in core
+            tool_message.content = str_output(tool_message.content)
+            return tool_message
         except Exception as e:
             if not self.handle_tool_errors:
                 raise e

libs/langgraph/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langgraph"
-version = "0.1.11"
+version = "0.1.12"
 description = "Building stateful, multi-actor applications with LLMs"
 authors = []
 license = "MIT"

libs/langgraph/tests/test_prebuilt.py

@@ -267,6 +267,13 @@ async def test_tool_node():
             raise ValueError("Test error")
         return f"tool2: {some_val} - {some_other_val}"
 
+    async def tool3(some_val: int, some_other_val: str) -> str:
+        """Tool 3 docstring."""
+        return [
+            {"key_1": some_val, "key_2": "foo"},
+            {"key_1": some_other_val, "key_2": "baz"},
+        ]
+
     result = ToolNode([tool1]).invoke(
         {
             "messages": [
@@ -377,6 +384,31 @@ async def test_tool_node():
     )
     assert tool_message.tool_call_id == "some 0"
 
+    # list of dicts tool content
+    result3 = await ToolNode([tool3]).ainvoke(
+        {
+            "messages": [
+                AIMessage(
+                    "hi?",
+                    tool_calls=[
+                        {
+                            "name": "tool3",
+                            "args": {"some_val": 2, "some_other_val": "bar"},
+                            "id": "some 0",
+                        }
+                    ],
+                )
+            ]
+        }
+    )
+    tool_message: ToolMessage = result3["messages"][-1]
+    assert tool_message.type == "tool"
+    assert (
+        tool_message.content
+        == '[{"key_1": 2, "key_2": "foo"}, {"key_1": "bar", "key_2": "baz"}]'
+    )
+    assert tool_message.tool_call_id == "some 0"
+
 
 def my_function(some_val: int, some_other_val: str) -> str:
     return f"{some_val} - {some_other_val}"

libs/sdk-js/scripts/create-entrypoints.js

@@ -38,9 +38,19 @@ const updateConfig = () => {
     ...json,
     typedocOptions: {
       ...json.typedocOptions,
-      entryPoints: [...Object.keys(entrypoints)].map(
-        (key) => `src/${entrypoints[key]}.ts`,
-      ),
+      entryPoints: [...Object.keys(entrypoints)].map((key) => {
+        const basePath = `src/${entrypoints[key]}`;
+        if (fs.existsSync(`${basePath}.mts`)) {
+          return `${basePath}.mts`;
+        } else if (fs.existsSync(`${basePath}.ts`)) {
+          return `${basePath}.ts`;
+        } else {
+          console.warn(
+            `Warning: Neither ${basePath}.mts nor ${basePath}.ts found for entrypoint ${key}`,
+          );
+          return `${basePath}.ts`; // Default to .ts if neither exists
+        }
+      }),
     },
   }));
 

libs/sdk-js/src/client.mts

@@ -107,7 +107,7 @@ class BaseClient {
   }
 }
 
-class CronsClient extends BaseClient {
+export class CronsClient extends BaseClient {
   /**
    *
    * @param threadId The ID of the thread.
@@ -195,7 +195,7 @@ class CronsClient extends BaseClient {
   }
 }
 
-class AssistantsClient extends BaseClient {
+export class AssistantsClient extends BaseClient {
   /**
    * Get an assistant by ID.
    *
@@ -302,7 +302,7 @@ class AssistantsClient extends BaseClient {
   }
 }
 
-class ThreadsClient extends BaseClient {
+export class ThreadsClient extends BaseClient {
   /**
    * Get a thread by ID.
    *
@@ -496,7 +496,7 @@ class ThreadsClient extends BaseClient {
   }
 }
 
-class RunsClient extends BaseClient {
+export class RunsClient extends BaseClient {
   stream(
     threadId: null,
     assistantId: string,

libs/sdk-js/tsconfig.json

@@ -2,7 +2,11 @@
   "extends": "@tsconfig/recommended",
   "compilerOptions": {
     "target": "ES2021",
-    "lib": ["ES2021", "ES2022.Object", "DOM"],
+    "lib": [
+      "ES2021",
+      "ES2022.Object",
+      "DOM"
+    ],
     "module": "NodeNext",
     "moduleResolution": "nodenext",
     "esModuleInterop": true,
@@ -17,19 +21,26 @@
     "strict": true,
     "outDir": "dist"
   },
-  "include": ["src/**/*"],
-  "exclude": ["node_modules", "dist", "coverage"],
+  "include": [
+    "src/**/*"
+  ],
+  "exclude": [
+    "node_modules",
+    "dist",
+    "coverage"
+  ],
   "includeVersion": true,
   "typedocOptions": {
     "entryPoints": [
-      "src/client.mts",
-      "src/schema.ts",
-      "src/types.mts"
+      "src/client.mts"
     ],
     "readme": "none",
     "out": "docs",
     "plugin": [
       "typedoc-plugin-markdown"
-    ]
+    ],
+    "excludePrivate": true,
+    "excludeProtected": true,
+    "excludeExternals": false
   }
 }