Merge remote-tracking branch 'origin/master' into feat/odedgo/atp-client-opt-in-refresh-token-behaviour

Author: encodedz

packages/langchain/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@mondaydotcomorg/atp-langchain",
-	"version": "0.23.0",
+	"version": "0.24.0",
 	"description": "LangChain integration for Agent Tool Protocol",
 	"type": "module",
 	"main": "./dist/index.cjs",

packages/langchain/src/langgraph-client.ts

@@ -20,6 +20,7 @@ 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 type { Embeddings } from '@langchain/core/embeddings';
+import { AgentToolProtocolServer } from '@mondaydotcomorg/atp-server';
 
 /**
  * Approval request that needs human decision
@@ -44,8 +45,10 @@ export interface ApprovalResponse {
  * Options for creating the LangGraph ATP client
  */
 export interface LangGraphATPClientOptions {
-	/** Base URL of ATP server */
-	serverUrl: string;
+    /** ATP server instance */
+    server?: AgentToolProtocolServer;
+    /** Base URL of ATP server used for http server (not inProcess) */
+	serverUrl?: string;
 	/** Custom headers for authentication (e.g., { Authorization: 'Bearer token' }) */
 	headers?: Record<string, string>;
 	/** LangChain LLM for atp.llm.call() sampling */
@@ -121,6 +124,7 @@ export class LangGraphATPClient {
 
 	constructor(options: LangGraphATPClientOptions) {
 		const {
+			server,
 			serverUrl,
 			headers,
 			llm,
@@ -132,7 +136,8 @@ export class LangGraphATPClient {
 		} = options;
 
 		this.client = new AgentToolProtocolClient({
-			baseUrl: serverUrl,
+			server,
+            baseUrl: serverUrl,
 			headers,
 			hooks,
 			serviceProviders: tools ? { tools } : undefined,

packages/langchain/src/langgraph-tools.ts

@@ -24,9 +24,7 @@ import { ExecutionStatus, type ExecutionConfig } from '@mondaydotcomorg/atp-prot
 /**
  * Options for creating ATP tools with LangGraph integration
  */
-export interface CreateATPToolsOptions extends Omit<LangGraphATPClientOptions, 'serverUrl'> {
-	/** ATP server URL */
-	serverUrl: string;
+export interface CreateATPToolsOptions extends LangGraphATPClientOptions {
 	/**
 	 * Default execution config for all ATP code executions
 	 */
@@ -83,11 +81,12 @@ export interface ATPToolsResult {
  * ```
  */
 export async function createATPTools(options: CreateATPToolsOptions): Promise<ATPToolsResult> {
-	const { serverUrl, defaultExecutionConfig, ...clientOptions } = options;
+	const { serverUrl, server, defaultExecutionConfig, ...clientOptions } = options;
 
 	const client = new LangGraphATPClient({
-		serverUrl,
-		...clientOptions,
+        server,
+        serverUrl,
+        ...clientOptions,
 	});
 
 	await client.connect();

packages/langchain/tsconfig.json

@@ -19,5 +19,5 @@
 	},
 	"include": ["src/**/*"],
 	"exclude": ["node_modules", "dist", "**/*.test.ts"],
-	"references": [{ "path": "../protocol" }, { "path": "../client" }, { "path": "../runtime" }]
+	"references": [{ "path": "../protocol" }, { "path": "../client" }, { "path": "../runtime" }, { "path": "../server" }]
 }