nemoris 0.1.0 → 0.1.2

package/src/providers/anthropic.js

@@ -1,168 +1,168 @@
-import { ProviderAdapter } from "./base.js";
-
-/**
- * Build correct auth headers for Anthropic API.
- * OAuth/setup-tokens (sk-ant-oat-*) use Authorization: Bearer.
- * API keys (sk-ant-api-* or sk-ant-*) use x-api-key.
- */
-export function buildAnthropicAuthHeaders(token) {
-  if (token && token.startsWith("sk-ant-oat")) {
-    return { "Authorization": `Bearer ${token}`, "anthropic-version": "2023-06-01" };
-  }
-  return { "x-api-key": token, "anthropic-version": "2023-06-01" };
-}
-
-export class AnthropicAdapter extends ProviderAdapter {
-  static providerId = "anthropic";
-
-  normalizeModelId(model) {
-    return typeof model === "string" && model.startsWith("anthropic/")
-      ? model.slice("anthropic/".length)
-      : model;
-  }
-
-  getCapabilities() {
-    return {
-      supportsToolDefinitions: true,
-      supportsToolCalls: true,
-      structuredOutputMode: "forced_tool",
-      supportsReasoningSchema: true,
-      tokenCountingMode: "exact",
-      toolReliabilityTier: "high",
-      supportsVision: true
-    };
-  }
-
-  buildStructuredOutputTool(input) {
-    if (!input?.responseSchema?.schema) return null;
-    return {
-      name: input.responseSchema.name || "record_structured_output",
-      description: "Return the final structured job result using the required schema.",
-      input_schema: input.responseSchema.schema
-    };
-  }
-
-  buildInvokePayload(input) {
-    const { model: rawModel, system, messages, maxTokens = 2048, tools } = input;
-    const model = this.normalizeModelId(rawModel);
-    const structuredOutputTool = this.buildStructuredOutputTool(input);
-    const combinedTools = [...(tools || [])];
-    if (structuredOutputTool) {
-      combinedTools.push(structuredOutputTool);
-    }
-
-    const payload = {
-      model,
-      max_tokens: maxTokens,
-      messages
-    };
-    if (system) {
-      payload.system = [{
-        type: "text",
-        text: typeof system === "string" ? system : JSON.stringify(system),
-        cache_control: { type: "ephemeral" }
-      }];
-    }
-    if (combinedTools.length) payload.tools = combinedTools;
-    if (structuredOutputTool) {
-      payload.tool_choice = {
-        type: "tool",
-        name: structuredOutputTool.name
-      };
-      payload.disable_parallel_tool_use = true;
-    }
-    return payload;
-  }
-
-  async invoke(input) {
-    const token = this.ensureAuthToken();
-    return this.postJson(
-      "v1/messages",
-      this.buildInvokePayload(input),
-      buildAnthropicAuthHeaders(token),
-      {
-        timeoutMs: input.timeoutMs
-      }
-    );
-  }
-
-  normalizeResponse(raw) {
-    const toolUseBlock =
-      raw?.content?.find((item) => item?.type === "tool_use" && item.input && typeof item.input === "object") || null;
-    if (toolUseBlock?.input) {
-      return {
-        summary: toolUseBlock.input.summary || "Provider returned structured tool output.",
-        output: toolUseBlock.input.output ?? toolUseBlock.input,
-        nextActions: Array.isArray(toolUseBlock.input.nextActions) ? toolUseBlock.input.nextActions : [],
-        reasoning: toolUseBlock.input.analysis || null,
-        raw
-      };
-    }
-
-    return super.normalizeResponse(raw);
-  }
-
-  buildCountTokensPayload(input) {
-    const { model: rawModel, system, messages, tools } = input;
-    const model = this.normalizeModelId(rawModel);
-    const structuredOutputTool = this.buildStructuredOutputTool(input);
-    const combinedTools = [...(tools || [])];
-    if (structuredOutputTool) {
-      combinedTools.push(structuredOutputTool);
-    }
-
-    const payload = {
-      model,
-      messages
-    };
-    if (system) {
-      payload.system = [{
-        type: "text",
-        text: typeof system === "string" ? system : JSON.stringify(system),
-        cache_control: { type: "ephemeral" }
-      }];
-    }
-    if (combinedTools.length) payload.tools = combinedTools;
-    if (structuredOutputTool) {
-      payload.tool_choice = {
-        type: "tool",
-        name: structuredOutputTool.name
-      };
-      payload.disable_parallel_tool_use = true;
-    }
-    return payload;
-  }
-
-  async countTokens(input) {
-    const token = this.ensureAuthToken();
-    const result = await this.postJson(
-      "v1/messages/count_tokens",
-      this.buildCountTokensPayload(input),
-      buildAnthropicAuthHeaders(token),
-      {
-        timeoutMs: input.timeoutMs
-      }
-    );
-    return Number(result?.input_tokens ?? result?.inputTokens ?? 0);
-  }
-
-  async healthCheck() {
-    const token = this.ensureAuthToken();
-    // OAuth tokens (sk-ant-oat*) don't have access to GET /v1/models.
-    // Use a minimal messages.count_tokens probe instead — works with all token types.
-    // The configured baseUrl still goes through the shared SSRF policy.
-    try {
-      const response = await this.fetchResponse("v1/messages/count_tokens", {
-        method: "POST",
-        headers: { "content-type": "application/json", ...buildAnthropicAuthHeaders(token) },
-        body: JSON.stringify({
-          model: "claude-haiku-4-5",
-          messages: [{ role: "user", content: "ping" }]
-        })
-      });
-      return { ok: response.ok, status: response.status };
-    } catch {
-      return { ok: false, status: 0 };
-    }
-  }
-}
+import { ProviderAdapter } from "./base.js";
+
+/**
+ * Build correct auth headers for Anthropic API.
+ * OAuth/setup-tokens (sk-ant-oat-*) use Authorization: Bearer.
+ * API keys (sk-ant-api-* or sk-ant-*) use x-api-key.
+ */
+export function buildAnthropicAuthHeaders(token) {
+  if (token && token.startsWith("sk-ant-oat")) {
+    return { "Authorization": `Bearer ${token}`, "anthropic-version": "2023-06-01" };
+  }
+  return { "x-api-key": token, "anthropic-version": "2023-06-01" };
+}
+
+export class AnthropicAdapter extends ProviderAdapter {
+  static providerId = "anthropic";
+
+  normalizeModelId(model) {
+    return typeof model === "string" && model.startsWith("anthropic/")
+      ? model.slice("anthropic/".length)
+      : model;
+  }
+
+  getCapabilities() {
+    return {
+      supportsToolDefinitions: true,
+      supportsToolCalls: true,
+      structuredOutputMode: "forced_tool",
+      supportsReasoningSchema: true,
+      tokenCountingMode: "exact",
+      toolReliabilityTier: "high",
+      supportsVision: true
+    };
+  }
+
+  buildStructuredOutputTool(input) {
+    if (!input?.responseSchema?.schema) return null;
+    return {
+      name: input.responseSchema.name || "record_structured_output",
+      description: "Return the final structured job result using the required schema.",
+      input_schema: input.responseSchema.schema
+    };
+  }
+
+  buildInvokePayload(input) {
+    const { model: rawModel, system, messages, maxTokens = 2048, tools } = input;
+    const model = this.normalizeModelId(rawModel);
+    const structuredOutputTool = this.buildStructuredOutputTool(input);
+    const combinedTools = [...(tools || [])];
+    if (structuredOutputTool) {
+      combinedTools.push(structuredOutputTool);
+    }
+
+    const payload = {
+      model,
+      max_tokens: maxTokens,
+      messages
+    };
+    if (system) {
+      payload.system = [{
+        type: "text",
+        text: typeof system === "string" ? system : JSON.stringify(system),
+        cache_control: { type: "ephemeral" }
+      }];
+    }
+    if (combinedTools.length) payload.tools = combinedTools;
+    if (structuredOutputTool) {
+      payload.tool_choice = {
+        type: "tool",
+        name: structuredOutputTool.name
+      };
+      payload.disable_parallel_tool_use = true;
+    }
+    return payload;
+  }
+
+  async invoke(input) {
+    const token = this.ensureAuthToken();
+    return this.postJson(
+      "v1/messages",
+      this.buildInvokePayload(input),
+      buildAnthropicAuthHeaders(token),
+      {
+        timeoutMs: input.timeoutMs
+      }
+    );
+  }
+
+  normalizeResponse(raw) {
+    const toolUseBlock =
+      raw?.content?.find((item) => item?.type === "tool_use" && item.input && typeof item.input === "object") || null;
+    if (toolUseBlock?.input) {
+      return {
+        summary: toolUseBlock.input.summary || "Provider returned structured tool output.",
+        output: toolUseBlock.input.output ?? toolUseBlock.input,
+        nextActions: Array.isArray(toolUseBlock.input.nextActions) ? toolUseBlock.input.nextActions : [],
+        reasoning: toolUseBlock.input.analysis || null,
+        raw
+      };
+    }
+
+    return super.normalizeResponse(raw);
+  }
+
+  buildCountTokensPayload(input) {
+    const { model: rawModel, system, messages, tools } = input;
+    const model = this.normalizeModelId(rawModel);
+    const structuredOutputTool = this.buildStructuredOutputTool(input);
+    const combinedTools = [...(tools || [])];
+    if (structuredOutputTool) {
+      combinedTools.push(structuredOutputTool);
+    }
+
+    const payload = {
+      model,
+      messages
+    };
+    if (system) {
+      payload.system = [{
+        type: "text",
+        text: typeof system === "string" ? system : JSON.stringify(system),
+        cache_control: { type: "ephemeral" }
+      }];
+    }
+    if (combinedTools.length) payload.tools = combinedTools;
+    if (structuredOutputTool) {
+      payload.tool_choice = {
+        type: "tool",
+        name: structuredOutputTool.name
+      };
+      payload.disable_parallel_tool_use = true;
+    }
+    return payload;
+  }
+
+  async countTokens(input) {
+    const token = this.ensureAuthToken();
+    const result = await this.postJson(
+      "v1/messages/count_tokens",
+      this.buildCountTokensPayload(input),
+      buildAnthropicAuthHeaders(token),
+      {
+        timeoutMs: input.timeoutMs
+      }
+    );
+    return Number(result?.input_tokens ?? result?.inputTokens ?? 0);
+  }
+
+  async healthCheck() {
+    const token = this.ensureAuthToken();
+    // OAuth tokens (sk-ant-oat*) don't have access to GET /v1/models.
+    // Use a minimal messages.count_tokens probe instead — works with all token types.
+    // The configured baseUrl still goes through the shared SSRF policy.
+    try {
+      const response = await this.fetchResponse("v1/messages/count_tokens", {
+        method: "POST",
+        headers: { "content-type": "application/json", ...buildAnthropicAuthHeaders(token) },
+        body: JSON.stringify({
+          model: "claude-haiku-4-5",
+          messages: [{ role: "user", content: "ping" }]
+        })
+      });
+      return { ok: response.ok, status: response.status };
+    } catch {
+      return { ok: false, status: 0 };
+    }
+  }
+}