@agentgrader/agent-openrouter 1.0.1 → 2.0.2

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { AgentAdapter, StepEvent, AgentResult } from '@agentgrader/core';
 
-declare class OpenRouterAgentAdapter implements AgentAdapter {
-    readonly name = "openrouter";
+declare class AiSdkAgentAdapter implements AgentAdapter {
+    readonly name = "ai-sdk";
     solve(input: {
         prompt: string;
         sandbox: any;
@@ -10,4 +10,4 @@ declare class OpenRouterAgentAdapter implements AgentAdapter {
     }): Promise<AgentResult>;
 }
 
-export { OpenRouterAgentAdapter };
+export { AiSdkAgentAdapter, AiSdkAgentAdapter as OpenRouterAgentAdapter };

package/dist/index.js

@@ -1,25 +1,54 @@
 import { tool, experimental_createMCPClient, generateText } from 'ai';
 import { Experimental_StdioMCPTransport } from 'ai/mcp-stdio';
 import { createOpenAI } from '@ai-sdk/openai';
+import { createAnthropic } from '@ai-sdk/anthropic';
 import { z } from 'zod';
 
 // src/index.ts
-var OpenRouterAgentAdapter = class {
-  name = "openrouter";
+var AiSdkAgentAdapter = class {
+  name = "ai-sdk";
   async solve(input) {
     const { prompt, sandbox, config, onStep } = input;
-    const apiKey = process.env.OPENROUTER_API_KEY || process.env.OPENAI_API_KEY || "mock-key";
-    const baseURL = process.env.OPENROUTER_API_KEY ? "https://openrouter.ai/api/v1" : void 0;
-    const client = createOpenAI({
-      baseURL,
-      apiKey,
-      headers: {
-        "HTTP-Referer": "https://github.com/agentgrader/agr",
-        "X-Title": "Agentgrader"
-      }
-    });
+    const provider = config.provider || "openrouter";
     const modelName = config.model || "gpt-4o-mini";
-    const model = client(modelName);
+    let model;
+    if (provider === "anthropic") {
+      if (!process.env.ANTHROPIC_API_KEY) {
+        throw new Error(
+          "ANTHROPIC_API_KEY is not set. Set it in your environment or .env file to use provider: anthropic."
+        );
+      }
+      const anthropic = createAnthropic({
+        apiKey: process.env.ANTHROPIC_API_KEY
+      });
+      model = anthropic(modelName);
+    } else if (provider === "openai") {
+      if (!process.env.OPENAI_API_KEY) {
+        throw new Error(
+          "OPENAI_API_KEY is not set. Set it in your environment or .env file to use provider: openai."
+        );
+      }
+      const openai = createOpenAI({
+        apiKey: process.env.OPENAI_API_KEY
+      });
+      model = openai(modelName);
+    } else {
+      const apiKey = process.env.OPENROUTER_API_KEY || process.env.OPENAI_API_KEY;
+      if (!apiKey) {
+        throw new Error(
+          "OPENROUTER_API_KEY (or OPENAI_API_KEY) is not set. Set it in your environment or .env file to use provider: openrouter."
+        );
+      }
+      const openrouter = createOpenAI({
+        baseURL: "https://openrouter.ai/api/v1",
+        apiKey,
+        headers: {
+          "HTTP-Referer": "https://github.com/agentgrader/agr",
+          "X-Title": "Agentgrader"
+        }
+      });
+      model = openrouter(modelName);
+    }
     let submitted = false;
     let stepIndex = 0;
     const localTools = {
@@ -91,9 +120,9 @@ var OpenRouterAgentAdapter = class {
           // remote transport type, regardless of the configured `type`
           { type: "sse", url: serverConfig.url, headers: serverConfig.headers }
         );
-        const client2 = await experimental_createMCPClient({ transport });
-        mcpClients.push(client2);
-        const serverTools = await client2.tools();
+        const client = await experimental_createMCPClient({ transport });
+        mcpClients.push(client);
+        const serverTools = await client.tools();
         for (const [toolName, toolDef] of Object.entries(serverTools)) {
           mcpTools[`${serverName}_${toolName}`] = toolDef;
         }
@@ -168,9 +197,9 @@ var OpenRouterAgentAdapter = class {
     } catch (err) {
       console.error(`Error in generateText agent loop: ${err.message}`);
     } finally {
-      for (const client2 of mcpClients) {
+      for (const client of mcpClients) {
         try {
-          await client2.close();
+          await client.close();
         } catch (e) {
         }
       }
@@ -183,4 +212,4 @@ var OpenRouterAgentAdapter = class {
   }
 };
 
-export { OpenRouterAgentAdapter };
+export { AiSdkAgentAdapter, AiSdkAgentAdapter as OpenRouterAgentAdapter };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentgrader/agent-openrouter",
-  "version": "1.0.1",
+  "version": "2.0.2",
   "description": "OpenRouter/AI SDK ReAct agent adapter for the Agentgrader framework",
   "license": "MIT",
   "type": "module",
@@ -21,8 +21,9 @@
     "build:watch": "tsup src/index.ts --format esm --dts --watch"
   },
   "dependencies": {
+    "@agentgrader/core": "^1.1.2",
+    "@ai-sdk/anthropic": "^1.2.12",
     "@ai-sdk/openai": "^1.1.0",
-    "@agentgrader/core": "^1.0.1",
     "ai": "^4.1.0",
     "zod": "^3.23.8"
   },
@@ -30,6 +31,6 @@
     "tsup": "^8.5.1"
   },
   "peerDependencies": {
-    "@agentgrader/core": "^1.0.1"
+    "@agentgrader/core": "^1.1.2"
   }
 }