@andreprado/agentkit 0.1.0-alpha.17 → 0.1.0-alpha.19

package/src/index.ts

@@ -2,9 +2,10 @@ export type AgentRuntime = "local" | "edge";
 export type AgentProviderName = "test" | "openai" | "anthropic" | "openrouter" | "custom";
 export type AccessMode = "private" | "token" | "public";
 export type ToolVisibility = "user" | "internal";
-export type ChannelType = "website" | "telegram" | "whatsapp";
-export type ChannelProvider = "agentkit" | "telegram" | "zapster" | "meta";
+export type ChannelType = "website" | "telegram" | "whatsapp" | "discord";
+export type ChannelProvider = "agentkit" | "telegram" | "zapster" | "meta" | "discord";
 export type WhatsappChannelProvider = "zapster" | "meta";
+export type DiscordChannelMode = "interactions" | "bot";
 export type TelegramAllowedUpdate = "message";
 export type ChannelAudioMode = "off" | "reject" | "transcribe";
 export type TranscriptionProviderName = "test" | "openai" | "groq";
@@ -143,7 +144,26 @@ export type WhatsappChannelConfig = {
   audio?: ChannelAudioConfig;
 };
 
-export type AgentChannel = WebsiteChannelConfig | TelegramChannelConfig | WhatsappChannelConfig;
+export type DiscordChannelInput = {
+  name: string;
+  mode?: DiscordChannelMode;
+  secrets?: string[];
+  limits?: ChannelLimits;
+  buffer?: ChannelBufferInput;
+};
+
+export type DiscordChannelConfig = {
+  type: "discord";
+  provider: "discord";
+  name: string;
+  mode: DiscordChannelMode;
+  secrets: string[];
+  limits?: ChannelLimits;
+  buffer?: ChannelBufferConfig;
+  audio?: never;
+};
+
+export type AgentChannel = WebsiteChannelConfig | TelegramChannelConfig | WhatsappChannelConfig | DiscordChannelConfig;
 
 export type IntegrationTagsConfig = {
   enable?: string[];
@@ -155,6 +175,7 @@ export type ComposioManagedIntegrationInput = {
   toolkits: string[];
   tools?: Record<string, string[]>;
   tags?: IntegrationTagsConfig;
+  confirmExternalWrites?: boolean;
 };
 
 export type ComposioManagedIntegration = {
@@ -165,6 +186,7 @@ export type ComposioManagedIntegration = {
   toolkits: string[];
   tools?: Record<string, string[]>;
   tags?: IntegrationTagsConfig;
+  confirmExternalWrites?: boolean;
 };
 
 export type AgentIntegration = ComposioManagedIntegration;
@@ -368,6 +390,7 @@ export function composioManaged(input: ComposioManagedIntegrationInput): Composi
   const toolkits = normalizeToolkitList(input.toolkits, "composioManaged.toolkits");
   const tools = input.tools ? normalizeComposioToolMap(input.tools, toolkits) : undefined;
   const tags = input.tags ? normalizeIntegrationTags(input.tags) : undefined;
+  const confirmExternalWrites = input.confirmExternalWrites;
 
   return {
     type: "composio",
@@ -377,6 +400,7 @@ export function composioManaged(input: ComposioManagedIntegrationInput): Composi
     toolkits,
     ...(tools ? { tools } : {}),
     ...(tags ? { tags } : {}),
+    ...(confirmExternalWrites === undefined ? {} : { confirmExternalWrites }),
   };
 }
 
@@ -424,6 +448,24 @@ export function whatsappChannel(input: WhatsappChannelInput): WhatsappChannelCon
   };
 }
 
+export function discordChannel(input: DiscordChannelInput): DiscordChannelConfig {
+  const mode = input.mode ?? "interactions";
+
+  return {
+    type: "discord",
+    provider: "discord",
+    name: input.name,
+    mode,
+    secrets: normalizeChannelSecrets(input.secrets ?? defaultDiscordChannelSecrets(mode), "discord channel"),
+    ...(input.limits ? { limits: input.limits } : {}),
+    ...(input.buffer ? { buffer: normalizeChannelBuffer(input.buffer, "discord channel") } : {}),
+  };
+}
+
+function defaultDiscordChannelSecrets(mode: DiscordChannelMode): string[] {
+  return mode === "bot" ? ["DISCORD_BOT_TOKEN"] : ["DISCORD_PUBLIC_KEY"];
+}
+
 function defaultWhatsappChannelSecrets(provider: WhatsappChannelProvider): string[] {
   if (provider === "zapster") {
     return ["ZAPSTER_API_KEY", "ZAPSTER_INSTANCE_ID", "ZAPSTER_WEBHOOK_ID"];
@@ -645,6 +687,7 @@ export type {
   ChannelMediaDownloadResult,
   ChannelMessageContentType,
   ChannelMessageRole,
+  ChannelReplyTarget,
   ChannelSendInput,
   ChannelSendResult,
   ChannelStatusInput,

package/src/providers/pi.ts

@@ -19,6 +19,9 @@ import type { ProviderAdapter, ProviderRunInput, ProviderRunResult } from "./typ
 
 const PI_PROVIDER_NAMES = ["openai", "anthropic", "openrouter"] as const satisfies AgentProviderName[];
 const MAX_TOOL_TURNS = 8;
+const OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1";
+const UNKNOWN_OPENROUTER_CONTEXT_WINDOW = 128000;
+const UNKNOWN_OPENROUTER_MAX_TOKENS = 8192;
 
 export const piProviderAdapters: ProviderAdapter[] = PI_PROVIDER_NAMES.map((name) => ({
   name,
@@ -189,25 +192,56 @@ function formatSecretList(names: string[]): string {
   return `one of: ${names.join(", ")}`;
 }
 
-function resolveModel(provider: Provider, modelId: string): Model<any> {
-  const modelExists = getModels(provider as any).some((model) => model.id === modelId);
+export function resolveModel(provider: Provider, modelId: string): Model<any> {
+  const model = getModel(provider as any, modelId as any) as Model<any> | undefined;
 
-  if (!modelExists) {
-    throw new AgentKitError(
-      "provider_model_unsupported",
-      `Pi SDK could not resolve provider "${provider}" model "${modelId}".`,
-    );
+  if (model) {
+    return model;
   }
 
-  try {
-    return getModel(provider as any, modelId as any) as Model<any>;
-  } catch (error) {
-    throw new AgentKitError(
-      "provider_model_unsupported",
-      `Pi SDK could not resolve provider "${provider}" model "${modelId}".`,
-      { cause: error },
-    );
+  if (provider === "openrouter") {
+    return createOpenRouterPassThroughModel(modelId);
   }
+
+  throw new AgentKitError(
+    "provider_model_unsupported",
+    `Pi SDK could not resolve provider "${provider}" model "${modelId}". ${knownModelHint(provider)}`,
+  );
+}
+
+function createOpenRouterPassThroughModel(modelId: string): Model<"openai-completions"> {
+  return {
+    id: modelId,
+    name: `OpenRouter ${modelId}`,
+    api: "openai-completions",
+    provider: "openrouter",
+    baseUrl: OPENROUTER_BASE_URL,
+    reasoning: false,
+    input: ["text"],
+    cost: {
+      input: 0,
+      output: 0,
+      cacheRead: 0,
+      cacheWrite: 0,
+    },
+    contextWindow: UNKNOWN_OPENROUTER_CONTEXT_WINDOW,
+    maxTokens: UNKNOWN_OPENROUTER_MAX_TOKENS,
+    compat: {
+      openRouterRouting: {},
+    },
+  };
+}
+
+function knownModelHint(provider: Provider): string {
+  const examples = getModels(provider as any)
+    .slice(0, 5)
+    .map((model) => model.id);
+
+  if (examples.length === 0) {
+    return "Use a model id supported by the installed Pi SDK for that provider.";
+  }
+
+  return `Use a model id supported by the installed Pi SDK for that provider. Examples: ${examples.join(", ")}.`;
 }
 
 function toPiMessages(messages: ProviderRunInput["messages"]): Message[] {

package/src/runtime/channel-test-harness.ts

@@ -14,7 +14,10 @@ export type ChannelFixtureName =
   | "zapster-message"
   | "zapster-audio-message"
   | "zapster-duplicate-message"
-  | "zapster-unsupported-media";
+  | "zapster-unsupported-media"
+  | "discord-command"
+  | "discord-message-create"
+  | "discord-ping";
 
 export type ChannelFixture = {
   name: ChannelFixtureName;