Adjust pool capacity and support local agent creation

src/index.ts

@@ -8,7 +8,7 @@ import { startOpenAiProxy } from "./openai-proxy.js";
 import { createProviderHook } from "./provider.js";
 import { STATIC_FALLBACK_MODELS, makeModelMeta } from "./models.js";
 
-const POOL_CAPACITY = 8;
+const POOL_CAPACITY = 10;
 
 /**
  * OpenCode client with extended properties used by this plugin.

src/openai-proxy.ts

@@ -152,7 +152,15 @@ async function handleChat(req: IncomingMessage, res: ServerResponse, log: Logger
   } else {
     const maxRetries = 3;
     let attempt = 0;
-    const { Agent } = (await import("@cursor/sdk")) as { Agent: typeof import("@cursor/sdk").Agent };
+    const { Agent } = (await import("@cursor/sdk")) as unknown as {
+      Agent: {
+        create: (opts: {
+          apiKey: string;
+          model: { id: string };
+          local: { cwd: string };
+        }) => Promise<any>;
+      };
+    };
 
     while (attempt < maxRetries) {
       attempt++;
@@ -162,6 +170,7 @@ async function handleChat(req: IncomingMessage, res: ServerResponse, log: Logger
         agent = await Agent.create({
           apiKey,
           model: { id: modelId },
+          local: { cwd: process.cwd() },
         });
         messageToSend = translated.fullPromptOnMiss;
         log.debug("cursor-openai-proxy: pool miss", { prefixHash: translated.prefixHash.slice(0, 8) });

src/provider.ts

@@ -309,11 +309,17 @@ async function runDoStream(opts: {
   return { stream };
 }
 
+type AgentCreateOpts = {
+  apiKey: string;
+  model: { id: string };
+  local: { cwd: string };
+};
+
 /**
  * Single attempt to create an agent with error classification and logging.
  */
 async function performAgentCreationAttempt(deps: {
-  Agent: { create: (opts: { apiKey: string; model: { id: string } }) => Promise<unknown> };
+  Agent: { create: (opts: AgentCreateOpts) => Promise<unknown> };
   apiKey: string;
   modelId: string;
   log: Logger;
@@ -322,7 +328,11 @@ async function performAgentCreationAttempt(deps: {
   const { Agent, apiKey, modelId, log, attempt } = deps;
   try {
     log.debug("cursor-provider: calling Agent.create", { modelId, attempt });
-    const agent = (await Agent.create({ apiKey, model: { id: modelId } })) as SDKAgent;
+    const agent = (await Agent.create({
+      apiKey,
+      model: { id: modelId },
+      local: { cwd: process.cwd() },
+    })) as SDKAgent;
     return { agent };
   } catch (err) {
     const decision = classifyError(err, { phase: "create" });
@@ -342,7 +352,7 @@ async function createAgentWithRetry(deps: { apiKey: string; modelId: string; log
   const { log } = deps;
 
   for (let attempt = 1; attempt <= 3; attempt++) {
-    const result = await performAgentCreationAttempt({ Agent: Agent as unknown as { create: (opts: { apiKey: string; model: { id: string } }) => Promise<unknown> }, ...deps, attempt });
+    const result = await performAgentCreationAttempt({ Agent: Agent as unknown as { create: (opts: AgentCreateOpts) => Promise<unknown> }, ...deps, attempt });
 
     if ("agent" in result) return result.agent;
     if (!result.canRetry) {

tests/provider.test.ts

@@ -138,7 +138,10 @@ describe("createProviderHook.models()", () => {
     // models1 から生成された doStream なので ctx1 を使うべき
     expect(resolveApiKey).toHaveBeenCalledWith(ctx1, expect.anything());
     expect(sdk.Agent.create).toHaveBeenCalledWith(
-      expect.objectContaining({ apiKey: "key-1" }),
+      expect.objectContaining({
+        apiKey: "key-1",
+        local: { cwd: process.cwd() },
+      }),
     );
   });