@pantheon.ai/mcp 0.0.1

package/packages/sdk/src/schemas/project.ts

@@ -0,0 +1,208 @@
+import { StreamResponseParser } from "@pantheon/sdk/lib/stream-parser"
+import { zodJsonDate, ZodStream } from "@pantheon/sdk/lib/zod"
+import {
+  type MasterAgentMessage,
+  type MasterAgentMessageChunk,
+  type MasterAgentMessagePart,
+  messageMetadataSchema,
+} from "@pantheon/sdk/schemas/ai"
+import { z, type ZodType } from "zod"
+
+export type Project = {
+  id: string
+  name: string
+  display_name: string
+  category: string
+  description: string
+  user_id: string
+  root_branch_id: string | null
+  ongoing_stream_id: string | null
+  status?:
+    | "CREATED"
+    | "PROVISIONING"
+    | "READY"
+    | "RUNNING"
+    | "PAUSED"
+    | "FAILED"
+    | "ARCHIVED"
+    | null
+  error_message?: string | null
+  created_at: Date
+  updated_at: Date
+}
+
+export const projectSchema = z.object({
+  id: z.string(),
+  name: z.string(),
+  display_name: z.string(),
+  category: z.string(),
+  description: z.string(),
+  user_id: z.string(),
+  root_branch_id: z.string().nullable(),
+  ongoing_stream_id: z.string().nullable(),
+  status: z
+    .enum([
+      "CREATED",
+      "PROVISIONING",
+      "READY",
+      "RUNNING",
+      "PAUSED",
+      "FAILED",
+      "ARCHIVED",
+    ])
+    .nullable()
+    .optional(),
+  error_message: z.string().nullable().optional(),
+  created_at: zodJsonDate,
+  updated_at: zodJsonDate,
+})
+
+export interface ProjectMessage extends MasterAgentMessage {
+  ordinal: number
+  project_id: string
+  event_stream_id: string | null
+  source_request_id?: string | null
+  status: string | null
+  is_finished: boolean
+  created_at: Date
+  updated_at: Date
+}
+
+export interface ProjectQueuedRequest {
+  id: string
+  type: string
+  payload: unknown
+  requested_at: Date
+  summary: string
+}
+
+export const projectMessageSchema = z.object({
+  id: z.string(),
+  role: z.enum(["system", "user", "assistant"]),
+  metadata: messageMetadataSchema.nullable().optional(),
+  parts: z.array(
+    z.object().loose() as unknown as ZodType<MasterAgentMessagePart>
+  ),
+  ordinal: z.number(),
+  project_id: z.string(),
+  source_request_id: z.string().nullable().optional(),
+  event_stream_id: z.string().nullable(),
+  status: z.string().nullable(),
+  is_finished: z.boolean(),
+  created_at: zodJsonDate,
+  updated_at: zodJsonDate,
+})
+
+const projectMessageStreamParser = new StreamResponseParser({
+  pipe: (input) => {
+    return input
+      .pipeThrough(new TextDecoderStream())
+      .pipeThrough(new BufferedLinesStream("\n\n"))
+      .pipeThrough(new SSEDataDecoderStream())
+      .pipeThrough(
+        new ZodStream({
+          schema: z
+            .object()
+            .loose() as unknown as ZodType<MasterAgentMessageChunk>,
+        })
+      )
+  },
+})
+
+const projectRawMessageStreamParser = new StreamResponseParser({
+  pipe: (input) => {
+    return input
+      .pipeThrough(new TextDecoderStream())
+      .pipeThrough(new BufferedLinesStream("\n\n"))
+      .pipeThrough(new SSEDataDecoderStream())
+      .pipeThrough(
+        new ZodStream({
+          schema: z.object().loose() as unknown as ZodType<unknown>,
+        })
+      )
+  },
+})
+
+export const projectQueuedRequestSchema = z.object({
+  id: z.string(),
+  type: z.string(),
+  payload: z.unknown(),
+  requested_at: zodJsonDate,
+  summary: z.string(),
+})
+
+export const projectBackgroundTaskSchema = z.object({
+  task_id: z.string(),
+  name: z.string().nullable(),
+  action_type: z.enum(["execution", "exploration"]),
+  action_id: z.string(),
+  status: z.enum(["SUCCESS", "FAILURE", "PENDING", "STARTED"]).nullable(),
+  result: z.object().loose().nullable(),
+  retries: z.number().nullable(),
+  started_at: zodJsonDate.nullable(),
+  last_polled_at: zodJsonDate.nullable(),
+  ended_at: zodJsonDate.nullable(),
+})
+
+class BufferedLinesStream extends TransformStream<string, string> {
+  constructor(sep = "\n") {
+    let buffer = ""
+    super({
+      transform(chunk, controller) {
+        chunk = buffer += chunk
+
+        while (true) {
+          const index = chunk.indexOf(sep)
+          if (index === -1) {
+            buffer = chunk
+            break
+          }
+
+          const line = chunk.slice(0, index).trim()
+          if (line) {
+            controller.enqueue(line)
+          }
+          chunk = chunk.slice(index + sep.length)
+        }
+      },
+
+      flush() {
+        if (buffer) {
+          console.warn("buffered lines stream has unprocessed lines:", buffer)
+        }
+      },
+    })
+  }
+}
+
+export class SSEDataDecoderStream extends TransformStream<string, any> {
+  constructor() {
+    let handle: ReturnType<typeof setTimeout> | undefined = undefined
+
+    super({
+      transform: (chunk, controller) => {
+        clearTimeout(handle)
+        handle = setTimeout(() => {
+          controller.error(new Error("Read timeout, please retry."))
+        }, 15000)
+
+        if (/^data:/.test(chunk)) {
+          const part = chunk.replace(/^data: /, "").trim()
+          if (part === "[DONE]") {
+          } else {
+            try {
+              controller.enqueue(JSON.parse(part))
+            } catch (e) {
+              console.error("invalid sse chunk", chunk, e)
+              controller.error(e)
+            }
+          }
+        } else {
+          console.debug("ignore sse comment", chunk)
+        }
+      },
+    })
+  }
+}
+
+export { projectMessageStreamParser, projectRawMessageStreamParser }

package/packages/sdk/tsconfig.json

@@ -0,0 +1,17 @@
+{
+  "compilerOptions": {
+    "lib": ["ESNext", "DOM", "DOM.AsyncIterable"],
+    "target": "ESNext",
+    "strict": true,
+    "module": "NodeNext",
+    "moduleResolution": "bundler",
+    "esModuleInterop": true,
+    "skipDefaultLibCheck": true,
+    "isolatedModules": true,
+    "baseUrl": ".",
+    "paths": {
+      "@pantheon/sdk/*": ["./src/*"]
+    }
+  },
+  "include": ["src"]
+}

package/patches/ai+6.0.45.patch

@@ -0,0 +1,163 @@
+diff --git a/node_modules/ai/dist/index.d.mts b/node_modules/ai/dist/index.d.mts
+index f66d872..74d7d44 100644
+--- a/node_modules/ai/dist/index.d.mts
++++ b/node_modules/ai/dist/index.d.mts
+@@ -5633,3 +5633,55 @@ declare global {
+ }
+ 
+ export { AbstractChat, Agent, AgentCallParameters, AgentStreamParameters, AsyncIterableStream, CallSettings, CallWarning, ChatAddToolApproveResponseFunction, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, ContentPart, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DirectChatTransport, DirectChatTransportOptions, DynamicToolCall, DynamicToolError, DynamicToolResult, DynamicToolUIPart, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelMiddleware, EmbeddingModelUsage, ErrorHandler, ToolLoopAgent as Experimental_Agent, ToolLoopAgentSettings as Experimental_AgentSettings, DownloadFunction as Experimental_DownloadFunction, Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, InferAgentUIMessage as Experimental_InferAgentUIMessage, LogWarningsFunction as Experimental_LogWarningsFunction, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, FileUIPart, FinishReason, GenerateImageResult, GenerateObjectResult, GenerateTextOnFinishCallback, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageModelMiddleware, ImageModelProviderMetadata, ImageModelResponseMetadata, ImageModelUsage, InferAgentUIMessage, InferCompleteOutput as InferGenerateOutput, InferPartialOutput as InferStreamOutput, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolApprovalError, InvalidToolInputError, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelMiddleware, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, LogWarningsFunction, MessageConversionError, MissingToolResultsError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputGeneratedError, NoSpeechGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningOutput, ReasoningUIPart, RepairTextFunction, RerankResult, RerankingModel, RetryError, SafeValidateUIMessagesResult, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, StaticToolCall, StaticToolError, StaticToolOutputDenied, StaticToolResult, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, TimeoutConfiguration, ToolApprovalRequestOutput, ToolCallNotFoundForApprovalError, ToolCallRepairError, ToolCallRepairFunction, ToolChoice, ToolLoopAgent, ToolLoopAgentOnFinishCallback, ToolLoopAgentOnStepFinishCallback, ToolLoopAgentSettings, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TypedToolCall, TypedToolError, TypedToolOutputDenied, TypedToolResult, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamError, UIMessageStreamOnFinishCallback, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UIToolInvocation, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, Warning, addToolInputExamplesMiddleware, assistantModelMessageSchema, callCompletionApi, consumeStream, convertFileListToFileUIParts, convertToModelMessages, cosineSimilarity, createAgentUIStream, createAgentUIStreamResponse, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultEmbeddingSettingsMiddleware, defaultSettingsMiddleware, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractJsonMiddleware, extractReasoningMiddleware, generateImage, generateObject, generateText, getStaticToolName, getTextFromDataUrl, getToolName, getToolOrDynamicToolName, hasToolCall, isDataUIPart, isDeepEqualData, isFileUIPart, isReasoningUIPart, isStaticToolUIPart, isTextUIPart, isToolOrDynamicToolUIPart, isToolUIPart, lastAssistantMessageIsCompleteWithApprovalResponses, lastAssistantMessageIsCompleteWithToolCalls, modelMessageSchema, parsePartialJson, pipeAgentUIStreamToResponse, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, pruneMessages, readUIMessageStream, rerank, safeValidateUIMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, uiMessageChunkSchema, userModelMessageSchema, validateUIMessages, wrapEmbeddingModel, wrapImageModel, wrapLanguageModel, wrapProvider };
++
++
++export type StreamingUIMessageState<UI_MESSAGE extends UIMessage> = {
++    message: UI_MESSAGE;
++    activeTextParts: Record<string, TextUIPart>;
++    activeReasoningParts: Record<string, ReasoningUIPart>;
++    partialToolCalls: Record<
++      string,
++      {
++          text: string;
++          index: number;
++          toolName: string;
++          dynamic?: boolean;
++          title?: string;
++      }
++    >;
++    finishReason?: FinishReason;
++};
++
++export function createStreamingUIMessageState<UI_MESSAGE extends UIMessage>({
++    lastMessage,
++    messageId,
++}: {
++    lastMessage: UI_MESSAGE | undefined;
++    messageId: string;
++}): StreamingUIMessageState<UI_MESSAGE>
++
++export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
++    stream,
++    messageMetadataSchema,
++    dataPartSchemas,
++    runUpdateMessageJob,
++    onError,
++    onToolCall,
++    onData,
++}: {
++    // input stream is not fully typed yet:
++    stream: ReadableStream<UIMessageChunk>;
++    messageMetadataSchema?: FlexibleSchema<InferUIMessageMetadata<UI_MESSAGE>>;
++    dataPartSchemas?: UIDataTypesToSchemas<InferUIMessageData<UI_MESSAGE>>;
++    onToolCall?: (options: {
++        toolCall: InferUIMessageToolCall<UI_MESSAGE>;
++    }) => void | PromiseLike<void>;
++    onData?: (dataPart: DataUIPart<InferUIMessageData<UI_MESSAGE>>) => void;
++    runUpdateMessageJob: (
++      job: (options: {
++          state: StreamingUIMessageState<UI_MESSAGE>;
++          write: () => void;
++      }) => Promise<void>,
++    ) => Promise<void>;
++    onError: ErrorHandler;
++}): ReadableStream<InferUIMessageChunk<UI_MESSAGE>>
+diff --git a/node_modules/ai/dist/index.d.ts b/node_modules/ai/dist/index.d.ts
+index f66d872..fdce962 100644
+--- a/node_modules/ai/dist/index.d.ts
++++ b/node_modules/ai/dist/index.d.ts
+@@ -5633,3 +5633,54 @@ declare global {
+ }
+ 
+ export { AbstractChat, Agent, AgentCallParameters, AgentStreamParameters, AsyncIterableStream, CallSettings, CallWarning, ChatAddToolApproveResponseFunction, ChatInit, ChatOnDataCallback, ChatOnErrorCallback, ChatOnFinishCallback, ChatOnToolCallCallback, ChatRequestOptions, ChatState, ChatStatus, ChatTransport, ChunkDetector, CompletionRequestOptions, ContentPart, CreateUIMessage, DataUIPart, DeepPartial, DefaultChatTransport, DirectChatTransport, DirectChatTransportOptions, DynamicToolCall, DynamicToolError, DynamicToolResult, DynamicToolUIPart, EmbedManyResult, EmbedResult, Embedding, EmbeddingModel, EmbeddingModelMiddleware, EmbeddingModelUsage, ErrorHandler, ToolLoopAgent as Experimental_Agent, ToolLoopAgentSettings as Experimental_AgentSettings, DownloadFunction as Experimental_DownloadFunction, Experimental_GenerateImageResult, GeneratedFile as Experimental_GeneratedImage, InferAgentUIMessage as Experimental_InferAgentUIMessage, LogWarningsFunction as Experimental_LogWarningsFunction, SpeechResult as Experimental_SpeechResult, TranscriptionResult as Experimental_TranscriptionResult, FileUIPart, FinishReason, GenerateImageResult, GenerateObjectResult, GenerateTextOnFinishCallback, GenerateTextOnStepFinishCallback, GenerateTextResult, GeneratedAudioFile, GeneratedFile, HttpChatTransport, HttpChatTransportInitOptions, ImageModel, ImageModelMiddleware, ImageModelProviderMetadata, ImageModelResponseMetadata, ImageModelUsage, InferAgentUIMessage, InferCompleteOutput as InferGenerateOutput, InferPartialOutput as InferStreamOutput, InferUIDataParts, InferUIMessageChunk, InferUITool, InferUITools, InvalidArgumentError, InvalidDataContentError, InvalidMessageRoleError, InvalidStreamPartError, InvalidToolApprovalError, InvalidToolInputError, JSONValue, JsonToSseTransformStream, LanguageModel, LanguageModelMiddleware, LanguageModelRequestMetadata, LanguageModelResponseMetadata, LanguageModelUsage, LogWarningsFunction, MessageConversionError, MissingToolResultsError, NoImageGeneratedError, NoObjectGeneratedError, NoOutputGeneratedError, NoSpeechGeneratedError, NoSuchProviderError, NoSuchToolError, ObjectStreamPart, output as Output, PrepareReconnectToStreamRequest, PrepareSendMessagesRequest, PrepareStepFunction, PrepareStepResult, Prompt, Provider, ProviderMetadata, ProviderRegistryProvider, ReasoningOutput, ReasoningUIPart, RepairTextFunction, RerankResult, RerankingModel, RetryError, SafeValidateUIMessagesResult, SerialJobExecutor, SourceDocumentUIPart, SourceUrlUIPart, SpeechModel, SpeechModelResponseMetadata, StaticToolCall, StaticToolError, StaticToolOutputDenied, StaticToolResult, StepResult, StepStartUIPart, StopCondition, StreamObjectOnFinishCallback, StreamObjectResult, StreamTextOnChunkCallback, StreamTextOnErrorCallback, StreamTextOnFinishCallback, StreamTextOnStepFinishCallback, StreamTextResult, StreamTextTransform, TelemetrySettings, TextStreamChatTransport, TextStreamPart, TextUIPart, TimeoutConfiguration, ToolApprovalRequestOutput, ToolCallNotFoundForApprovalError, ToolCallRepairError, ToolCallRepairFunction, ToolChoice, ToolLoopAgent, ToolLoopAgentOnFinishCallback, ToolLoopAgentOnStepFinishCallback, ToolLoopAgentSettings, ToolSet, ToolUIPart, TranscriptionModel, TranscriptionModelResponseMetadata, TypedToolCall, TypedToolError, TypedToolOutputDenied, TypedToolResult, UIDataPartSchemas, UIDataTypes, UIMessage, UIMessageChunk, UIMessagePart, UIMessageStreamError, UIMessageStreamOnFinishCallback, UIMessageStreamOptions, UIMessageStreamWriter, UITool, UIToolInvocation, UITools, UI_MESSAGE_STREAM_HEADERS, UnsupportedModelVersionError, UseCompletionOptions, Warning, addToolInputExamplesMiddleware, assistantModelMessageSchema, callCompletionApi, consumeStream, convertFileListToFileUIParts, convertToModelMessages, cosineSimilarity, createAgentUIStream, createAgentUIStreamResponse, createProviderRegistry, createTextStreamResponse, createUIMessageStream, createUIMessageStreamResponse, customProvider, defaultEmbeddingSettingsMiddleware, defaultSettingsMiddleware, embed, embedMany, experimental_createProviderRegistry, experimental_customProvider, experimental_generateImage, generateSpeech as experimental_generateSpeech, transcribe as experimental_transcribe, extractJsonMiddleware, extractReasoningMiddleware, generateImage, generateObject, generateText, getStaticToolName, getTextFromDataUrl, getToolName, getToolOrDynamicToolName, hasToolCall, isDataUIPart, isDeepEqualData, isFileUIPart, isReasoningUIPart, isStaticToolUIPart, isTextUIPart, isToolOrDynamicToolUIPart, isToolUIPart, lastAssistantMessageIsCompleteWithApprovalResponses, lastAssistantMessageIsCompleteWithToolCalls, modelMessageSchema, parsePartialJson, pipeAgentUIStreamToResponse, pipeTextStreamToResponse, pipeUIMessageStreamToResponse, pruneMessages, readUIMessageStream, rerank, safeValidateUIMessages, simulateReadableStream, simulateStreamingMiddleware, smoothStream, stepCountIs, streamObject, streamText, systemModelMessageSchema, toolModelMessageSchema, uiMessageChunkSchema, userModelMessageSchema, validateUIMessages, wrapEmbeddingModel, wrapImageModel, wrapLanguageModel, wrapProvider };
++
++export type StreamingUIMessageState<UI_MESSAGE extends UIMessage> = {
++    message: UI_MESSAGE;
++    activeTextParts: Record<string, TextUIPart>;
++    activeReasoningParts: Record<string, ReasoningUIPart>;
++    partialToolCalls: Record<
++      string,
++      {
++          text: string;
++          index: number;
++          toolName: string;
++          dynamic?: boolean;
++          title?: string;
++      }
++    >;
++    finishReason?: FinishReason;
++};
++
++export function createStreamingUIMessageState<UI_MESSAGE extends UIMessage>({
++    lastMessage,
++    messageId,
++}: {
++    lastMessage: UI_MESSAGE | undefined;
++    messageId: string;
++}): StreamingUIMessageState<UI_MESSAGE>
++
++export function processUIMessageStream<UI_MESSAGE extends UIMessage>({
++    stream,
++    messageMetadataSchema,
++    dataPartSchemas,
++    runUpdateMessageJob,
++    onError,
++    onToolCall,
++    onData,
++}: {
++    // input stream is not fully typed yet:
++    stream: ReadableStream<UIMessageChunk>;
++    messageMetadataSchema?: FlexibleSchema<InferUIMessageMetadata<UI_MESSAGE>>;
++    dataPartSchemas?: UIDataTypesToSchemas<InferUIMessageData<UI_MESSAGE>>;
++    onToolCall?: (options: {
++        toolCall: InferUIMessageToolCall<UI_MESSAGE>;
++    }) => void | PromiseLike<void>;
++    onData?: (dataPart: DataUIPart<InferUIMessageData<UI_MESSAGE>>) => void;
++    runUpdateMessageJob: (
++      job: (options: {
++          state: StreamingUIMessageState<UI_MESSAGE>;
++          write: () => void;
++      }) => Promise<void>,
++    ) => Promise<void>;
++    onError: ErrorHandler;
++}): ReadableStream<InferUIMessageChunk<UI_MESSAGE>>
+diff --git a/node_modules/ai/dist/index.js b/node_modules/ai/dist/index.js
+index 114687a..8b65974 100644
+--- a/node_modules/ai/dist/index.js
++++ b/node_modules/ai/dist/index.js
+@@ -4889,7 +4889,7 @@ function getToolName(part) {
+ var getToolOrDynamicToolName = getToolName;
+ 
+ // src/ui/process-ui-message-stream.ts
+-function createStreamingUIMessageState({
++export function createStreamingUIMessageState({
+   lastMessage,
+   messageId
+ }) {
+@@ -4905,7 +4905,7 @@ function createStreamingUIMessageState({
+     partialToolCalls: {}
+   };
+ }
+-function processUIMessageStream({
++export function processUIMessageStream({
+   stream,
+   messageMetadataSchema,
+   dataPartSchemas,
+diff --git a/node_modules/ai/dist/index.mjs b/node_modules/ai/dist/index.mjs
+index e5278cb..f74b520 100644
+--- a/node_modules/ai/dist/index.mjs
++++ b/node_modules/ai/dist/index.mjs
+@@ -4812,7 +4812,7 @@ function getToolName(part) {
+ var getToolOrDynamicToolName = getToolName;
+ 
+ // src/ui/process-ui-message-stream.ts
+-function createStreamingUIMessageState({
++export function createStreamingUIMessageState({
+   lastMessage,
+   messageId
+ }) {
+@@ -4828,7 +4828,7 @@ function createStreamingUIMessageState({
+     partialToolCalls: {}
+   };
+ }
+-function processUIMessageStream({
++export function processUIMessageStream({
+   stream,
+   messageMetadataSchema,
+   dataPartSchemas,

package/src/index.ts

@@ -0,0 +1,34 @@
+#!/usr/bin/env node
+
+import { startStdio } from './stdio.ts';
+import { startStreamableHttpServer } from './streamableHttp.ts';
+
+const args = process.argv.slice(2);
+
+function printHelp (message?: string) {
+  console.error(`${message || 'Usage'}
+npx @pantheon.ai/mcp stdio <Api-Key>
+PORT=10086 npx @pantheon.ai/mcp http [Optional Api-Key]
+`);
+}
+
+if (args.includes('--help')) {
+  printHelp();
+}
+
+switch (args[0]) {
+  case 'stdio':
+    if (!args[1]) {
+      printHelp('Api-Key is required for stdio mode');
+      process.exit(1);
+    }
+    await startStdio({ apiKey: args[1] });
+    break;
+  case 'http':
+    if (!args[1]) console.error('API Key not provided, client connections must provide authentication header.');
+    await startStreamableHttpServer({ apiKey: args[1] });
+    break;
+  default:
+    printHelp();
+    process.exit(1);
+}

package/src/mcp/mcp.ts

@@ -0,0 +1,172 @@
+import { getErrorMessage } from '@ai-sdk/provider';
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { type Api, createProject, createProjectExploration, getProject, getProjectBranch, getProjectBranchOutput, getProjectExplorationResult, listProjectBranches, listProjectExplorations, listProjects } from '@pantheon/sdk';
+import { getMe } from '@pantheon/sdk/api/auth';
+import type { ApiParams, ApiRequestBody, ApiResponseBody } from '@pantheon/sdk/lib/api';
+import { createApiExecutor, validateResponse } from '@pantheon/sdk/lib/api-executor';
+import { branchSchema } from '@pantheon/sdk/schemas/branch';
+import { paged } from '@pantheon/sdk/schemas/common';
+import { explorationResultSchema, explorationSchema } from '@pantheon/sdk/schemas/exploration';
+import { projectSchema } from '@pantheon/sdk/schemas/project';
+import { z, type ZodType } from 'zod';
+import packageJson from '../../package.json' with { type: 'json' };
+
+export async function mcpServer ({
+  authorizationHeader,
+}: {
+  authorizationHeader?: string
+}) {
+
+  const executor = createApiExecutor({
+    baseUrl: 'https://pantheon-ai.tidb.ai/',
+    headers: authorizationHeader ? {
+      'Authorization': authorizationHeader,
+    } : undefined,
+    validateResponse: validateResponse,
+  });
+
+  try {
+    await executor.execute(getMe, null, null);
+  } catch {
+    throw 401;
+  }
+
+  const mcpServer = new McpServer({
+    name: 'Pantheon MCP',
+    version: packageJson.version,
+    websiteUrl: 'https://pantheon-ai.tidb.ai/',
+  });
+
+  function registerApiTool<
+    API extends Api<any, any, any, any>,
+  > (
+    name: string,
+    api: API,
+    { paramsSchema, bodySchema, ...options }: {
+      title: string,
+      description?: string,
+      paramsSchema: ZodType<ApiParams<API>, any>,
+      bodySchema: ZodType<ApiRequestBody<API> extends void ? null : ApiRequestBody<API>, any>,
+      outputSchema: ZodType<ApiResponseBody<API>, any>,
+    },
+  ) {
+    mcpServer.registerTool(name, {
+      ...options,
+      inputSchema: z.object({
+        params: paramsSchema,
+        body: bodySchema,
+      }),
+    }, async ({ params, body }) => {
+      try {
+        const result = await executor.execute(api, params, body);
+        return {
+          content: [{ type: 'text', text: JSON.stringify(result) }],
+          structuredContent: result,
+        };
+      } catch (e) {
+        return {
+          content: [{ type: 'text', text: getErrorMessage(e) }],
+        };
+      }
+    });
+  }
+
+  registerApiTool('listProjects', listProjects, {
+    title: 'List pantheon projects',
+    paramsSchema: pageParams,
+    bodySchema: z.null(),
+    outputSchema: paged(projectSchema),
+  });
+
+  registerApiTool('createProject', createProject, {
+    title: 'Create pantheon project',
+    paramsSchema: z.null(),
+    bodySchema: z.object({
+      first_prompt: z.string(),
+      project_type: z.enum(['general', 'github']),
+      github_repo_id: z.number().optional(),
+      github_installation_id: z.number().optional(),
+      env_id: z.string().optional(),
+    }),
+    outputSchema: projectSchema,
+  });
+
+  registerApiTool('listProjectBranches', listProjectBranches, {
+    title: 'List project branches',
+    paramsSchema: pageParams.extend({
+      projectId: z.string(),
+    }),
+    bodySchema: z.null(),
+    outputSchema: paged(branchSchema),
+  });
+
+  registerApiTool('getProject', getProject, {
+    title: 'Get pantheon project',
+    paramsSchema: z.object({
+      projectId: z.string(),
+    }),
+    bodySchema: z.null(),
+    outputSchema: projectSchema,
+  });
+
+  registerApiTool('getProjectBranch', getProjectBranch, {
+    title: 'Get pantheon project branch info',
+    paramsSchema: z.object({
+      projectId: z.string(),
+      branchId: z.string(),
+    }),
+    bodySchema: z.null(),
+    outputSchema: branchSchema,
+  });
+
+  registerApiTool('getProjectBranchOutput', getProjectBranchOutput, {
+    title: 'Get pantheon project branch output',
+    paramsSchema: z.object({
+      projectId: z.string(),
+      branchId: z.string(),
+      fullOutput: z.boolean().optional().default(false),
+    }),
+    bodySchema: z.null(),
+    outputSchema: z.string(),
+  });
+
+  registerApiTool('createProjectExploration', createProjectExploration, {
+    title: 'Create pantheon project exploration',
+    paramsSchema: z.object({
+      projectId: z.string(),
+    }),
+    bodySchema: z.object({
+      parent_branch_id: z.string(),
+      shared_prompt_sequence: z.string().array().min(1).max(4),
+      num_branches: z.number().int().min(1).max(50).optional().default(3),
+      agent: z.string().optional(),
+    }),
+    outputSchema: explorationSchema,
+  });
+
+  registerApiTool('listProjectExplorations', listProjectExplorations, {
+    title: 'List project explorations',
+    paramsSchema: z.object({
+      projectId: z.string(),
+    }),
+    bodySchema: z.null(),
+    outputSchema: explorationSchema.array(),
+  });
+
+  registerApiTool('getProjectExplorationResult', getProjectExplorationResult, {
+    title: 'Get project exploration result',
+    paramsSchema: z.object({
+      projectId: z.string(),
+      explorationId: z.string(),
+    }),
+    bodySchema: z.null(),
+    outputSchema: explorationResultSchema,
+  });
+
+  return mcpServer;
+}
+
+const pageParams = z.object({
+  page: z.number().int().min(1),
+  size: z.number().int().min(1).max(100),
+});

package/src/stdio.ts

@@ -0,0 +1,11 @@
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { mcpServer } from './mcp/mcp.ts';
+
+export async function startStdio ({ apiKey }: { apiKey: string }) {
+  const stdio = new StdioServerTransport();
+  const server = await mcpServer({
+    authorizationHeader: `Bearer ${apiKey}`,
+  });
+
+  await server.connect(stdio);
+}

package/src/streamableHttp.ts

@@ -0,0 +1,60 @@
+import { createMcpExpressApp } from '@modelcontextprotocol/sdk/server/express.js';
+import { StreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/streamableHttp.js';
+import { mcpServer } from './mcp/mcp.ts';
+
+const port = process.env.PORT ? parseInt(process.env.PORT) : 9986;
+
+export async function startStreamableHttpServer ({ apiKey }: { apiKey?: string }) {
+  const app = createMcpExpressApp();
+
+  app.all('/mcp', async (req, res) => {
+    const authorizationHeader = req.get('authorization') ?? (apiKey ? `Bearer ${apiKey}` : undefined);
+    let server: Awaited<ReturnType<typeof mcpServer>>;
+
+    try {
+      server = await mcpServer({ authorizationHeader });
+    } catch (e) {
+      if (e === 401) {
+        res.status(400).send('401 Need Login');
+        return;
+      } else {
+        console.error('Error creating MCP server:', e);
+        res.status(500).send('Internal server error');
+        return;
+      }
+    }
+
+    const transport = new StreamableHTTPServerTransport({
+      sessionIdGenerator: undefined,
+    });
+
+    try {
+      await server.connect(transport);
+      await transport.handleRequest(req, res, req.body);
+
+      res.on('close', () => {
+        transport.close();
+        server.close();
+      });
+    } catch (error) {
+      console.error('Error handling MCP request:', error);
+      if (!res.headersSent) {
+        res.status(500).json({
+          jsonrpc: '2.0',
+          error: {
+            code: -32603,
+            message: 'Internal server error',
+          },
+          id: null,
+        });
+      }
+      transport.close().catch(() => {});
+      server.close().catch(() => {});
+    }
+  });
+
+  app.listen(port);
+
+  console.log('Listening on port 3001');
+}
+