@gram-ai/create-function 0.3.2 → 0.5.0

package/gram-template-gram/NEXT_STEPS.txt

@@ -2,6 +2,8 @@ All done! Jump in with `cd $DIR`.
 
 Some next steps:
 
+- Start a local development MCP server with `$PACKAGE_MANAGER run dev`
+
 - Ensure the `Gram CLI` is installed and authenticated (`gram auth`): https://docs.getgram.ai/command-line/installation
 
 - Build your function with `$PACKAGE_MANAGER run build`

package/gram-template-gram/README.md

@@ -40,20 +40,16 @@ pnpm build
 ## Testing Locally
 
 If you want to poke at the tools you've built during local development, you can
-start a Hono server with:
+start a local MCP server over stdio transport with:
 
 ```bash
 pnpm dev
 ```
 
-Now you can simulate tool calls with:
+Specifically, this command will spin up [MCP inspector][mcp-inspector] to let
+you interactively test your tools.
 
-```bash
-curl \
-  --data '{"name": "greet", "input": {"name": "Georges"}}' \
-  -H "Content-Type: application/json" \
-  http://localhost:3000/tool-call
-```
+[mcp-inspector]: https://github.com/modelcontextprotocol/inspector
 
 ## What next?
 

package/gram-template-gram/node_modules/.bin/mcp-inspector

@@ -0,0 +1,21 @@
+#!/bin/sh
+basedir=$(dirname "$(echo "$0" | sed -e 's,\\,/,g')")
+
+case `uname` in
+    *CYGWIN*|*MINGW*|*MSYS*)
+        if command -v cygpath > /dev/null 2>&1; then
+            basedir=`cygpath -w "$basedir"`
+        fi
+    ;;
+esac
+
+if [ -z "$NODE_PATH" ]; then
+  export NODE_PATH="/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/inspector/cli/build/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/inspector/cli/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/inspector/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/node_modules"
+else
+  export NODE_PATH="/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/inspector/cli/build/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/inspector/cli/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/inspector/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules/@modelcontextprotocol/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/@modelcontextprotocol+inspector@0.17.1_@types+node@22.18.6_@types+react-dom@19.1.9_@typ_cc52197392af7d7a50d2088d86700a7d/node_modules:/home/runner/work/gram/gram/node_modules/.pnpm/node_modules:$NODE_PATH"
+fi
+if [ -x "$basedir/node" ]; then
+  exec "$basedir/node"  "$basedir/../@modelcontextprotocol/inspector/cli/build/cli.js" "$@"
+else
+  exec node  "$basedir/../@modelcontextprotocol/inspector/cli/build/cli.js" "$@"
+fi

package/gram-template-gram/package.json

@@ -19,7 +19,7 @@
     "node": ">=22.18.0"
   },
   "scripts": {
-    "dev": "node ./src/server.ts",
+    "dev": "mcp-inspector node ./src/server.ts",
     "lint": "tsc --noEmit",
     "build": "node -e \"console.log('Stubbed out')\"",
     "_:build": "gram-build",
@@ -28,12 +28,12 @@
   },
   "dependencies": {
     "@gram-ai/functions": "workspace:",
+    "@modelcontextprotocol/sdk": "^1.20.1",
     "zod": "^4"
   },
   "devDependencies": {
-    "@hono/node-server": "^1.19.5",
+    "@modelcontextprotocol/inspector": "^0.17.1",
     "@types/node": "22.x",
-    "hono": "^4",
     "typescript": "^5"
   }
 }

package/gram-template-gram/src/server.ts

@@ -1,44 +1,125 @@
-import { serve } from "@hono/node-server";
-import { Hono } from "hono";
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import {
+  CallToolRequestSchema,
+  ListToolsRequestSchema,
+  type CallToolResult,
+  type ListToolsResult,
+} from "@modelcontextprotocol/sdk/types.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+
+import pkg from "../package.json" with { type: "json" };
 import { gram } from "./gram.ts";
 
-const app = new Hono();
-
-app.post("/tool-call", async (c) => {
-  const req = await c.req.json();
-  return await gram.handleToolCall(req, {
-    signal: AbortSignal.timeout(1 * 60 * 1000),
-  });
-});
-
-app.get("/manifest", async () => {
-  const manifest = gram.manifest();
-  return new Response(JSON.stringify(manifest, null, 2), {
-    headers: { "Content-Type": "application/json" },
-  });
-});
-
-export function startServer(port: number = 3000) {
-  const server = serve({ fetch: app.fetch, port }, (info) => {
-    console.log(`Listening on http://localhost:${info.port}`);
-  });
-
-  const onClose = () => {
-    console.log("Shutting down server...");
-    server.close((err) => {
-      if (err) {
-        console.error(err);
-        process.exit(1);
-      }
-      process.exit(0);
+const structuredLike = /\b(yaml|yml|json|toml|xml|xhtml)\b/i;
+const textLike = /^text\//i;
+const imageLike = /^image\//i;
+const audioLike = /^audio\//i;
+
+export const server = new Server(
+  {
+    name: pkg.name,
+    version: pkg.version,
+  },
+  {
+    capabilities: {
+      tools: {},
+    },
+  },
+);
+
+server.setRequestHandler(
+  ListToolsRequestSchema,
+  async (): Promise<ListToolsResult> => {
+    const tools = gram.manifest().tools.map((t) => {
+      return {
+        name: t.name,
+        description: t.description,
+        inputSchema: t.inputSchema,
+      };
+    }) as ListToolsResult["tools"];
+
+    return {
+      tools,
+    };
+  },
+);
+
+server.setRequestHandler(
+  CallToolRequestSchema,
+  async (req, extra): Promise<CallToolResult> => {
+    const { name, arguments: args } = req.params;
+
+    const resp = await gram.handleToolCall({ name, input: args } as any, {
+      signal: extra.signal,
     });
+
+    let ctype = resp.headers.get("Content-Type") || "";
+    ctype = ctype.split(";")[0]?.trim() || "";
+
+    switch (true) {
+      case textLike.test(ctype) || structuredLike.test(ctype): {
+        const text = await resp.text();
+        return {
+          content: [{ type: "text", text }],
+        };
+      }
+      case imageLike.test(ctype): {
+        return {
+          content: [
+            {
+              type: "image",
+              mimeType: ctype,
+              data: await responseToBase64(resp),
+            },
+          ],
+        };
+      }
+      case audioLike.test(ctype): {
+        return {
+          content: [
+            {
+              type: "audio",
+              mimeType: ctype,
+              data: await responseToBase64(resp),
+            },
+          ],
+        };
+      }
+      default: {
+        return {
+          isError: true,
+          content: [
+            {
+              type: "text",
+              text: `Unhandled content type: ${ctype}. Create a handler for this type in the MCP server.`,
+            },
+          ],
+        };
+      }
+    }
+  },
+);
+
+async function responseToBase64(resp: Response): Promise<string> {
+  const blob = await resp.arrayBuffer();
+  const buffer = Buffer.from(blob);
+  return buffer.toString("base64");
+}
+
+async function run() {
+  console.error("Starting MCP server with stdio...");
+  const stdio = new StdioServerTransport();
+  await server.connect(stdio);
+
+  const quit = async () => {
+    console.error("\nShutting down MCP server...");
+    await server.close();
+    process.exit(0);
   };
-  process.on("SIGINT", onClose);
-  process.on("SIGTERM", onClose);
+  process.once("SIGINT", quit);
+  process.once("SIGTERM", quit);
 }
 
 if (import.meta.main) {
-  startServer();
+  run();
 }
-
-export default app;

package/gram-template-mcp/src/functions.ts

@@ -26,3 +26,20 @@ export async function handleToolCall(call: {
     headers: { "Content-Type": "application/json; mcp=tools_call" },
   });
 }
+
+export async function handleResources(call: {
+  uri: string;
+  input: string;
+  _meta?: Record<string, unknown>;
+}): Promise<Response> {
+  const response = await client.readResource({
+    uri: call.uri,
+    _meta: call._meta,
+  });
+
+  const body = JSON.stringify(response);
+  return new Response(body, {
+    status: 200,
+    headers: { "Content-Type": "application/json" },
+  });
+}

package/gram-template-mcp/src/mcp.ts

@@ -20,3 +20,24 @@ server.registerTool(
     };
   },
 );
+
+server.registerResource(
+  "a-cool-photo",
+  "resources://a-cool-photo",
+  {
+    mimeType: "image/jpg",
+    description: "This photo is really something",
+    title: "A Cool Photo",
+  },
+  async (uri) => {
+    let res = await fetch("https://picsum.photos/200/300.jpg");
+    return {
+      contents: [
+        {
+          uri: uri.href,
+          blob: Buffer.from(await res.arrayBuffer()).toString("base64"),
+        },
+      ],
+    };
+  },
+);

package/package.json

@@ -1,7 +1,7 @@
 {
   "type": "module",
   "name": "@gram-ai/create-function",
-  "version": "0.3.2",
+  "version": "0.5.0",
   "description": "Build AI tools and deploy them to getgram.ai",
   "keywords": [],
   "homepage": "https://github.com/speakeasy-api/gram",
@@ -38,7 +38,7 @@
     "prettier": "^3.6.2",
     "typescript": "5.9.2",
     "zod": "^3.25.76",
-    "@gram-ai/functions": "^0.3.2"
+    "@gram-ai/functions": "^0.5.0"
   },
   "scripts": {
     "build": "tsc --noEmit false"