From 04232b31e32be815cb4050285a4e4edf84f04ac6 Mon Sep 17 00:00:00 2001
From: Charly Poly <1252066+charlypoly@users.noreply.github.com>
Date: Wed, 15 Jan 2025 19:04:19 +0100
Subject: [PATCH] Bundling fixes (#46)

* fix: import from `@modelcontextprotocol/sdk` should include `.js`

* fix: missing export for `createServer()`
---
 src/agent.ts | 6 +++---
 src/index.ts | 1 +
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/agent.ts b/src/agent.ts
index f23895d..cbe61a6 100644
--- a/src/agent.ts
+++ b/src/agent.ts
@@ -11,9 +11,9 @@ import {
 import { type Tool, type MCP } from "./types";
 import { getStepTools, type AnyZodType, type MaybePromise } from "./util";
 // MCP
-import { Client as MCPClient } from "@modelcontextprotocol/sdk/client/index";
-import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse";
-import { WebSocketClientTransport } from "@modelcontextprotocol/sdk/client/websocket";
+import { Client as MCPClient } from "@modelcontextprotocol/sdk/client/index.js";
+import { SSEClientTransport } from "@modelcontextprotocol/sdk/client/sse.js";
+import { WebSocketClientTransport } from "@modelcontextprotocol/sdk/client/websocket.js";
 import { ListToolsResultSchema } from "@modelcontextprotocol/sdk/types.js";
 import {
   type JSONSchema,
diff --git a/src/index.ts b/src/index.ts
index 4489b7f..83fb196 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -6,4 +6,5 @@ export * from "./network";
 export * from "./networkRun";
 export * from "./state";
 export * from "./types";
+export * from "./server";
 export * from "./util";