@super_studio/ecforce-ai-agent-server 1.2.0 → 1.3.0

package/README.md

@@ -0,0 +1,292 @@
+# `@super_studio/ecforce-ai-agent-server`
+
+ecforce AI Agent のサーバー向け SDK です。
+
+主に次の 2 つを提供します。
+
+- API SDK: `createClient()` で AI Agent の内部 API を呼ぶ
+- Proxy provider: `spst()` で Vercel AI SDK から内部プロキシ経由でモデルを使う
+
+## インストール
+
+```bash
+pnpm add @super_studio/ecforce-ai-agent-server
+```
+
+`spst()` は Vercel AI SDK と一緒に使います。
+
+```bash
+pnpm add ai
+```
+
+Google Search などで `@ai-sdk/google-vertex` のツールも使う場合は追加で入れてください。
+
+```bash
+pnpm add @ai-sdk/google-vertex
+```
+
+## 環境変数
+
+通常は `AI_AGENT_API_KEY` だけ設定すれば動きます。
+
+```env
+AI_AGENT_API_KEY=your-api-key
+```
+
+dev / preview 環境へ向けたい場合だけ、接続先を上書きします。
+
+```env
+AI_AGENT_API_ENDPOINT=https://dev-agent.ec-force.com
+```
+
+- `AI_AGENT_API_KEY`: 必須。管理画面で発行した API キー
+- `AI_AGENT_API_ENDPOINT`: 任意。省略時は本番 `https://agent.ec-force.com`
+
+## 使い分け
+
+### `createClient()`
+
+内部 API を直接呼びたいときに使います。
+
+用途:
+
+- AI Agent セッション発行
+- 管理用の内部 API 呼び出し
+- 利用量サマリー取得
+
+### `spst()`
+
+Vercel AI SDK の `generateText()` / `streamText()` などから、ecforce AI Agent の proxy provider を使いたいときに使います。
+
+用途:
+
+- OpenAI / Google / Anthropic を内部プロキシ経由で呼ぶ
+- `projectId` / `email` / `callerApp` をヘッダーに載せて課金・利用元を紐付ける
+- アプリ側から AI SDK のままテキスト生成やストリームを実行する
+
+## API SDK
+
+### `createClient()` の基本
+
+```ts
+import { createClient } from "@super_studio/ecforce-ai-agent-server";
+
+const client = createClient();
+```
+
+必要なら API キーや接続先を明示できます。
+
+```ts
+import { createClient } from "@super_studio/ecforce-ai-agent-server";
+
+const client = createClient({
+  apiKey: process.env.AI_AGENT_API_KEY,
+  baseUrl: process.env.AI_AGENT_API_ENDPOINT,
+});
+```
+
+### AI Agent セッションを発行する
+
+React パッケージの `ChatbotSheet` / `ChatbotFrame` に渡すセッションは、通常この SDK で発行します。
+
+```ts
+import { NextResponse } from "next/server";
+import { createClient } from "@super_studio/ecforce-ai-agent-server";
+
+const client = createClient();
+
+export async function POST() {
+  const currentUser = {
+    email: "user@example.com",
+    projectId: "project_123",
+  };
+
+  const session = await client.internalChat.createSession({
+    email: currentUser.email,
+    projectId: currentUser.projectId,
+  });
+
+  return NextResponse.json(session);
+}
+```
+
+返り値は次の形です。
+
+```ts
+type AgentSession = {
+  token: string;
+  expiresAt: string;
+};
+```
+
+### 内部 usage summary を取得する
+
+内部 API の利用例として、`/v1/internal/usage/summary` に対応する usage summary 取得は次のように呼べます。
+
+```ts
+import { createClient } from "@super_studio/ecforce-ai-agent-server";
+
+const client = createClient();
+
+const summary = await client.internalUsage.getUsageSummary({
+  projectId: "project_123",
+  period: "2026-04",
+});
+
+console.log(summary);
+/*
+{
+  plan: "pro",
+  totalUsedCredits: 12345,
+  creditLimit: 50000,
+  remainingCredits: 37655,
+  billingPeriod: "2026-04"
+}
+*/
+```
+
+`period` を省略すると、サーバー側の現在 billing period が使われます。
+
+想定用途:
+
+- 自社管理画面で利用量サマリーを表示する
+- `spst()` 経由の利用量を含めた残クレジットを確認する
+- 定期ジョブや監視でプロジェクトごとの消費量を取る
+
+## Proxy Provider
+
+`spst()` は `@super_studio/ecforce-ai-agent-server/provider` から import します。
+
+```ts
+import { spst } from "@super_studio/ecforce-ai-agent-server/provider";
+```
+
+### 最小例
+
+```ts
+import { generateText } from "ai";
+import { spst } from "@super_studio/ecforce-ai-agent-server/provider";
+
+const result = await generateText({
+  model: spst("openai/gpt-5.4"),
+  prompt: "売上の改善案を3つ提案して",
+  providerOptions: {
+    spst: {
+      projectId: "project_123",
+      email: "user@example.com",
+      callerApp: "admin-dashboard",
+    },
+  },
+});
+
+console.log(result.text);
+console.log(result.usage);
+```
+
+`providerOptions.spst` は内部利用時の識別情報です。
+
+- `projectId`: どのプロジェクトの利用か
+- `email`: 誰の利用か
+- `callerApp`: どのアプリからの利用か
+
+通常の内部利用では、この 3 つを毎回渡してください。
+
+### モデル ID の形式
+
+`spst()` は provider 付きの model id を受け取ります。
+
+```ts
+spst("openai/gpt-5.4");
+spst("google/gemini-2.5-flash");
+spst("anthropic/claude-sonnet-4-6");
+```
+
+現状サポートしている prefix は次です。
+
+- `openai/`
+- `google/`
+- `anthropic/`
+
+### `callerApp` を固定した provider を作る
+
+同じアプリから何度も呼ぶなら `createSpstProvider()` が便利です。
+
+```ts
+import { generateText } from "ai";
+import { createSpstProvider } from "@super_studio/ecforce-ai-agent-server/provider";
+
+const provider = createSpstProvider({
+  callerApp: "admin-dashboard",
+});
+
+const result = await generateText({
+  model: provider("google/gemini-2.5-flash"),
+  prompt: "今月の広告成果を要約して",
+  providerOptions: {
+    spst: {
+      projectId: "project_123",
+      email: "user@example.com",
+    },
+  },
+});
+```
+
+この場合、`callerApp` は `createSpstProvider()` 側の値が使われます。
+
+必要なら API キーや接続先もここで上書きできます。
+
+```ts
+import { createSpstProvider } from "@super_studio/ecforce-ai-agent-server/provider";
+
+const provider = createSpstProvider({
+  apiKey: process.env.AI_AGENT_API_KEY,
+  baseUrl: process.env.AI_AGENT_API_ENDPOINT,
+  callerApp: "admin-dashboard",
+});
+```
+
+### ストリーミング
+
+`streamText()` でも同じです。
+
+```ts
+import { streamText } from "ai";
+import { spst } from "@super_studio/ecforce-ai-agent-server/provider";
+
+const result = streamText({
+  model: spst("openai/gpt-5.4-mini"),
+  prompt: "FAQ の草案を作成して",
+  providerOptions: {
+    spst: {
+      projectId: "project_123",
+      email: "user@example.com",
+      callerApp: "admin-dashboard",
+    },
+  },
+});
+```
+
+### Google Search tool を併用する
+
+Google 系モデルでは、AI SDK 側のツールを組み合わせて使えます。
+
+```ts
+import { vertex } from "@ai-sdk/google-vertex";
+import { generateText } from "ai";
+import { spst } from "@super_studio/ecforce-ai-agent-server/provider";
+
+const result = await generateText({
+  model: spst("google/gemini-3.1-pro-preview-customtools"),
+  prompt: "今日の生成AI関連ニュースを3件まとめて",
+  tools: {
+    google_search: vertex.tools.googleSearch({}),
+  },
+  providerOptions: {
+    spst: {
+      projectId: "project_123",
+      email: "user@example.com",
+      callerApp: "admin-dashboard",
+    },
+  },
+});
+```

package/dist/lib/constants.cjs

@@ -1,6 +1,6 @@
 
 //#region src/lib/constants.ts
-const API_ENDPOINT = "http://localhost:4043";
+const API_ENDPOINT = "https://agent.ec-force.com";
 
 //#endregion
 exports.API_ENDPOINT = API_ENDPOINT;

package/dist/lib/constants.mjs

@@ -1,5 +1,5 @@
 //#region src/lib/constants.ts
-const API_ENDPOINT = "http://localhost:4043";
+const API_ENDPOINT = "https://agent.ec-force.com";
 
 //#endregion
 export { API_ENDPOINT };

package/dist/lib/constants.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"constants.mjs","names":[],"sources":["../../src/lib/constants.ts"],"sourcesContent":["export const API_ENDPOINT = \"http://localhost:4043\";\n"],"mappings":";AAAA,MAAa,eAAe"}
+{"version":3,"file":"constants.mjs","names":[],"sources":["../../src/lib/constants.ts"],"sourcesContent":["export const API_ENDPOINT = \"https://agent.ec-force.com\";\n"],"mappings":";AAAA,MAAa,eAAe"}

package/dist/lib/google-proxy-fetch.cjs

@@ -0,0 +1,34 @@
+const require_objectWithoutProperties = require('../_virtual/_@oxc-project_runtime@0.103.0/helpers/objectWithoutProperties.cjs');
+const require_objectSpread2 = require('../_virtual/_@oxc-project_runtime@0.103.0/helpers/objectSpread2.cjs');
+
+//#region src/lib/google-proxy-fetch.ts
+const _excluded = ["streamFunctionCallArguments"];
+function createGoogleProxyFetch(apiKey) {
+	return async (input, init) => {
+		const url = typeof input === "string" ? input : input.toString();
+		const headers = new Headers(init === null || init === void 0 ? void 0 : init.headers);
+		headers.delete("x-goog-api-key");
+		headers.delete("X-Goog-Api-Key");
+		if (apiKey) headers.set("authorization", `Bearer ${apiKey}`);
+		const body = normalizeGoogleProxyRequestBody(url, init === null || init === void 0 ? void 0 : init.body);
+		return fetch(input, require_objectSpread2._objectSpread2(require_objectSpread2._objectSpread2(require_objectSpread2._objectSpread2({}, init), body ? { body } : {}), {}, { headers }));
+	};
+}
+function normalizeGoogleProxyRequestBody(url, body) {
+	if (!body || typeof body !== "string") return body;
+	if (!url.includes(":generateContent") && !url.includes(":streamGenerateContent")) return body;
+	try {
+		const payload = JSON.parse(body);
+		const toolConfig = payload.toolConfig;
+		const functionCallingConfig = toolConfig === null || toolConfig === void 0 ? void 0 : toolConfig.functionCallingConfig;
+		if (!(functionCallingConfig === null || functionCallingConfig === void 0 ? void 0 : functionCallingConfig.streamFunctionCallArguments)) return body;
+		const { streamFunctionCallArguments: _streamFunctionCallArguments } = functionCallingConfig, restFunctionCallingConfig = require_objectWithoutProperties._objectWithoutProperties(functionCallingConfig, _excluded);
+		payload.toolConfig = require_objectSpread2._objectSpread2(require_objectSpread2._objectSpread2({}, toolConfig), {}, { functionCallingConfig: restFunctionCallingConfig });
+		return JSON.stringify(payload);
+	} catch (_unused) {
+		return body;
+	}
+}
+
+//#endregion
+exports.createGoogleProxyFetch = createGoogleProxyFetch;

package/dist/lib/google-proxy-fetch.mjs

@@ -0,0 +1,35 @@
+import { _objectWithoutProperties } from "../_virtual/_@oxc-project_runtime@0.103.0/helpers/objectWithoutProperties.mjs";
+import { _objectSpread2 } from "../_virtual/_@oxc-project_runtime@0.103.0/helpers/objectSpread2.mjs";
+
+//#region src/lib/google-proxy-fetch.ts
+const _excluded = ["streamFunctionCallArguments"];
+function createGoogleProxyFetch(apiKey) {
+	return async (input, init) => {
+		const url = typeof input === "string" ? input : input.toString();
+		const headers = new Headers(init === null || init === void 0 ? void 0 : init.headers);
+		headers.delete("x-goog-api-key");
+		headers.delete("X-Goog-Api-Key");
+		if (apiKey) headers.set("authorization", `Bearer ${apiKey}`);
+		const body = normalizeGoogleProxyRequestBody(url, init === null || init === void 0 ? void 0 : init.body);
+		return fetch(input, _objectSpread2(_objectSpread2(_objectSpread2({}, init), body ? { body } : {}), {}, { headers }));
+	};
+}
+function normalizeGoogleProxyRequestBody(url, body) {
+	if (!body || typeof body !== "string") return body;
+	if (!url.includes(":generateContent") && !url.includes(":streamGenerateContent")) return body;
+	try {
+		const payload = JSON.parse(body);
+		const toolConfig = payload.toolConfig;
+		const functionCallingConfig = toolConfig === null || toolConfig === void 0 ? void 0 : toolConfig.functionCallingConfig;
+		if (!(functionCallingConfig === null || functionCallingConfig === void 0 ? void 0 : functionCallingConfig.streamFunctionCallArguments)) return body;
+		const { streamFunctionCallArguments: _streamFunctionCallArguments } = functionCallingConfig, restFunctionCallingConfig = _objectWithoutProperties(functionCallingConfig, _excluded);
+		payload.toolConfig = _objectSpread2(_objectSpread2({}, toolConfig), {}, { functionCallingConfig: restFunctionCallingConfig });
+		return JSON.stringify(payload);
+	} catch (_unused) {
+		return body;
+	}
+}
+
+//#endregion
+export { createGoogleProxyFetch };
+//# sourceMappingURL=google-proxy-fetch.mjs.map

package/dist/lib/google-proxy-fetch.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-proxy-fetch.mjs","names":[],"sources":["../../src/lib/google-proxy-fetch.ts"],"sourcesContent":["export function createGoogleProxyFetch(apiKey: string | undefined) {\n  return async (input: RequestInfo | URL, init?: RequestInit) => {\n    const url = typeof input === \"string\" ? input : input.toString();\n    const headers = new Headers(init?.headers);\n    headers.delete(\"x-goog-api-key\");\n    headers.delete(\"X-Goog-Api-Key\");\n\n    if (apiKey) {\n      headers.set(\"authorization\", `Bearer ${apiKey}`);\n    }\n\n    const body = normalizeGoogleProxyRequestBody(url, init?.body);\n\n    return fetch(input, {\n      ...init,\n      ...(body ? { body } : {}),\n      headers,\n    });\n  };\n}\n\nfunction normalizeGoogleProxyRequestBody(\n  url: string,\n  body: RequestInit[\"body\"],\n): RequestInit[\"body\"] {\n  if (!body || typeof body !== \"string\") {\n    return body;\n  }\n\n  if (\n    !url.includes(\":generateContent\") &&\n    !url.includes(\":streamGenerateContent\")\n  ) {\n    return body;\n  }\n\n  try {\n    const payload = JSON.parse(body) as Record<string, unknown>;\n    const toolConfig = payload.toolConfig as\n      | Record<string, unknown>\n      | undefined;\n    const functionCallingConfig = toolConfig?.functionCallingConfig as\n      | Record<string, unknown>\n      | undefined;\n\n    if (!functionCallingConfig?.streamFunctionCallArguments) {\n      return body;\n    }\n\n    const {\n      streamFunctionCallArguments: _streamFunctionCallArguments,\n      ...restFunctionCallingConfig\n    } = functionCallingConfig;\n\n    payload.toolConfig = {\n      ...toolConfig,\n      functionCallingConfig: restFunctionCallingConfig,\n    };\n\n    return JSON.stringify(payload);\n  } catch {\n    return body;\n  }\n}\n"],"mappings":";;;;mBAkDM;AAlDN,SAAgB,uBAAuB,QAA4B;AACjE,QAAO,OAAO,OAA0B,SAAuB;EAC7D,MAAM,MAAM,OAAO,UAAU,WAAW,QAAQ,MAAM,UAAU;EAChE,MAAM,UAAU,IAAI,oDAAQ,KAAM,QAAQ;AAC1C,UAAQ,OAAO,iBAAiB;AAChC,UAAQ,OAAO,iBAAiB;AAEhC,MAAI,OACF,SAAQ,IAAI,iBAAiB,UAAU,SAAS;EAGlD,MAAM,OAAO,gCAAgC,iDAAK,KAAM,KAAK;AAE7D,SAAO,MAAM,wDACR,OACC,OAAO,EAAE,MAAM,GAAG,EAAE,SACxB,WACA;;;AAIN,SAAS,gCACP,KACA,MACqB;AACrB,KAAI,CAAC,QAAQ,OAAO,SAAS,SAC3B,QAAO;AAGT,KACE,CAAC,IAAI,SAAS,mBAAmB,IACjC,CAAC,IAAI,SAAS,yBAAyB,CAEvC,QAAO;AAGT,KAAI;EACF,MAAM,UAAU,KAAK,MAAM,KAAK;EAChC,MAAM,aAAa,QAAQ;EAG3B,MAAM,gFAAwB,WAAY;AAI1C,MAAI,gFAAC,sBAAuB,6BAC1B,QAAO;EAGT,MAAM,EACJ,6BAA6B,wDAC1B,qDACD;AAEJ,UAAQ,+CACH,mBACH,uBAAuB;AAGzB,SAAO,KAAK,UAAU,QAAQ;mBACxB;AACN,SAAO"}

package/dist/provider.cjs

@@ -0,0 +1,107 @@
+const require_constants = require('./lib/constants.cjs');
+const require_objectWithoutProperties = require('./_virtual/_@oxc-project_runtime@0.103.0/helpers/objectWithoutProperties.cjs');
+const require_objectSpread2 = require('./_virtual/_@oxc-project_runtime@0.103.0/helpers/objectSpread2.cjs');
+const require_google_proxy_fetch = require('./lib/google-proxy-fetch.cjs');
+let _ai_sdk_anthropic = require("@ai-sdk/anthropic");
+let _ai_sdk_google_vertex = require("@ai-sdk/google-vertex");
+let _ai_sdk_openai = require("@ai-sdk/openai");
+
+//#region src/provider.ts
+const _excluded = ["spst"];
+const spstProviders = createSpstProviders({});
+function spst(modelId) {
+	return createSpstLanguageModel(modelId, spstProviders);
+}
+function createSpstProvider(params) {
+	const providers = createSpstProviders(params);
+	return (modelId) => {
+		return createSpstLanguageModel(modelId, providers, params.callerApp);
+	};
+}
+function createSpstProviders(config) {
+	var _config$apiKey;
+	const apiKey = (_config$apiKey = config.apiKey) !== null && _config$apiKey !== void 0 ? _config$apiKey : process.env.AI_AGENT_API_KEY;
+	return {
+		openai: (0, _ai_sdk_openai.createOpenAI)({
+			name: "spst",
+			baseURL: getOpenAIProxyBaseUrl(config.baseUrl),
+			apiKey
+		}),
+		google: (0, _ai_sdk_google_vertex.createVertex)({
+			apiKey,
+			baseURL: getGoogleProxyBaseUrl(config.baseUrl),
+			fetch: require_google_proxy_fetch.createGoogleProxyFetch(apiKey)
+		}),
+		anthropic: (0, _ai_sdk_anthropic.createAnthropic)({
+			baseURL: getAnthropicProxyBaseUrl(config.baseUrl),
+			apiKey
+		})
+	};
+}
+function getResolvedBaseUrl(baseUrl) {
+	var _ref;
+	return ((_ref = baseUrl !== null && baseUrl !== void 0 ? baseUrl : process.env.AI_AGENT_API_ENDPOINT) !== null && _ref !== void 0 ? _ref : require_constants.API_ENDPOINT).replace(/\/$/, "");
+}
+function getOpenAIProxyBaseUrl(baseUrl) {
+	return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/openai`;
+}
+function getGoogleProxyBaseUrl(baseUrl) {
+	return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/google`;
+}
+function getAnthropicProxyBaseUrl(baseUrl) {
+	return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/anthropic`;
+}
+function createSpstLanguageModel(modelId, providers, defaultCallerApp) {
+	const parsedModelId = parseSpstModelId(modelId);
+	switch (parsedModelId.provider) {
+		case "openai": return wrapSpstModel(providers.openai.responses(parsedModelId.proxyModelId), defaultCallerApp);
+		case "google": return wrapSpstModel(providers.google(parsedModelId.proxyModelId), defaultCallerApp);
+		case "anthropic": return wrapSpstModel(providers.anthropic(parsedModelId.proxyModelId), defaultCallerApp);
+	}
+}
+function parseSpstModelId(modelId) {
+	if (modelId.startsWith("openai/")) return {
+		provider: "openai",
+		proxyModelId: modelId
+	};
+	if (modelId.startsWith("google/")) return {
+		provider: "google",
+		proxyModelId: modelId.slice(7)
+	};
+	if (modelId.startsWith("anthropic/")) return {
+		provider: "anthropic",
+		proxyModelId: modelId.slice(10)
+	};
+	if (modelId.includes("/")) throw new Error(`Provider '${modelId.split("/")[0]}' is not supported by spst() yet.`);
+	throw new Error("The spst() helper currently requires an 'openai/<model>', 'google/<model>', or 'anthropic/<model>' model id.");
+}
+function wrapSpstModel(model, defaultCallerApp) {
+	return {
+		specificationVersion: model.specificationVersion,
+		provider: model.provider,
+		modelId: model.modelId,
+		supportedUrls: model.supportedUrls,
+		doGenerate: (options) => model.doGenerate(withSpstHeaders(options, defaultCallerApp)),
+		doStream: (options) => model.doStream(withSpstHeaders(options, defaultCallerApp))
+	};
+}
+function withSpstHeaders(options, defaultCallerApp) {
+	var _options$providerOpti, _spstOptions$callerAp, _options$providerOpti2;
+	const spstOptions = (_options$providerOpti = options.providerOptions) === null || _options$providerOpti === void 0 ? void 0 : _options$providerOpti.spst;
+	if (!spstOptions) return options;
+	const callerApp = (_spstOptions$callerAp = spstOptions.callerApp) !== null && _spstOptions$callerAp !== void 0 ? _spstOptions$callerAp : defaultCallerApp;
+	if (!spstOptions.projectId || !spstOptions.email || !callerApp) throw new Error("The spst provider requires projectId, email, and callerApp in providerOptions.spst.");
+	const _ref2 = (_options$providerOpti2 = options.providerOptions) !== null && _options$providerOpti2 !== void 0 ? _options$providerOpti2 : {}, { spst: _spst } = _ref2, providerOptions = require_objectWithoutProperties._objectWithoutProperties(_ref2, _excluded);
+	return require_objectSpread2._objectSpread2(require_objectSpread2._objectSpread2({}, options), {}, {
+		headers: require_objectSpread2._objectSpread2(require_objectSpread2._objectSpread2({}, options.headers), {}, {
+			"x-spst-project-id": spstOptions.projectId,
+			"x-spst-user-email": spstOptions.email,
+			"x-spst-caller-app": callerApp
+		}),
+		providerOptions
+	});
+}
+
+//#endregion
+exports.createSpstProvider = createSpstProvider;
+exports.spst = spst;

package/dist/provider.d.cts

@@ -0,0 +1,18 @@
+import { LanguageModelV3 } from "@ai-sdk/provider";
+
+//#region src/provider.d.ts
+type SpstProviderOptions = {
+  projectId: string;
+  email: string;
+  callerApp: string;
+};
+type CreateSpstProviderParams = {
+  apiKey?: string;
+  baseUrl?: string;
+  callerApp: string;
+};
+declare function spst(modelId: string): LanguageModelV3;
+declare function createSpstProvider(params: CreateSpstProviderParams): typeof spst;
+//#endregion
+export { CreateSpstProviderParams, SpstProviderOptions, createSpstProvider, spst };
+//# sourceMappingURL=provider.d.cts.map

package/dist/provider.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.cts","names":[],"sources":["../src/provider.ts"],"sourcesContent":[],"mappings":";;;KAOY,mBAAA;;EAAA,KAAA,EAAA,MAAA;EAMA,SAAA,EAAA,MAAA;AA4BZ,CAAA;AAIgB,KAhCJ,wBAAA,GAiCF;;;;;iBALM,IAAA,mBAAuB;iBAIvB,kBAAA,SACN,kCACA"}

package/dist/provider.d.mts

@@ -0,0 +1,18 @@
+import { LanguageModelV3 } from "@ai-sdk/provider";
+
+//#region src/provider.d.ts
+type SpstProviderOptions = {
+  projectId: string;
+  email: string;
+  callerApp: string;
+};
+type CreateSpstProviderParams = {
+  apiKey?: string;
+  baseUrl?: string;
+  callerApp: string;
+};
+declare function spst(modelId: string): LanguageModelV3;
+declare function createSpstProvider(params: CreateSpstProviderParams): typeof spst;
+//#endregion
+export { CreateSpstProviderParams, SpstProviderOptions, createSpstProvider, spst };
+//# sourceMappingURL=provider.d.mts.map

package/dist/provider.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.d.mts","names":[],"sources":["../src/provider.ts"],"sourcesContent":[],"mappings":";;;KAOY,mBAAA;;EAAA,KAAA,EAAA,MAAA;EAMA,SAAA,EAAA,MAAA;AA4BZ,CAAA;AAIgB,KAhCJ,wBAAA,GAiCF;;;;;iBALM,IAAA,mBAAuB;iBAIvB,kBAAA,SACN,kCACA"}

package/dist/provider.mjs

@@ -0,0 +1,107 @@
+import { API_ENDPOINT } from "./lib/constants.mjs";
+import { _objectWithoutProperties } from "./_virtual/_@oxc-project_runtime@0.103.0/helpers/objectWithoutProperties.mjs";
+import { _objectSpread2 } from "./_virtual/_@oxc-project_runtime@0.103.0/helpers/objectSpread2.mjs";
+import { createGoogleProxyFetch } from "./lib/google-proxy-fetch.mjs";
+import { createAnthropic } from "@ai-sdk/anthropic";
+import { createVertex } from "@ai-sdk/google-vertex";
+import { createOpenAI } from "@ai-sdk/openai";
+
+//#region src/provider.ts
+const _excluded = ["spst"];
+const spstProviders = createSpstProviders({});
+function spst(modelId) {
+	return createSpstLanguageModel(modelId, spstProviders);
+}
+function createSpstProvider(params) {
+	const providers = createSpstProviders(params);
+	return (modelId) => {
+		return createSpstLanguageModel(modelId, providers, params.callerApp);
+	};
+}
+function createSpstProviders(config) {
+	var _config$apiKey;
+	const apiKey = (_config$apiKey = config.apiKey) !== null && _config$apiKey !== void 0 ? _config$apiKey : process.env.AI_AGENT_API_KEY;
+	return {
+		openai: createOpenAI({
+			name: "spst",
+			baseURL: getOpenAIProxyBaseUrl(config.baseUrl),
+			apiKey
+		}),
+		google: createVertex({
+			apiKey,
+			baseURL: getGoogleProxyBaseUrl(config.baseUrl),
+			fetch: createGoogleProxyFetch(apiKey)
+		}),
+		anthropic: createAnthropic({
+			baseURL: getAnthropicProxyBaseUrl(config.baseUrl),
+			apiKey
+		})
+	};
+}
+function getResolvedBaseUrl(baseUrl) {
+	var _ref;
+	return ((_ref = baseUrl !== null && baseUrl !== void 0 ? baseUrl : process.env.AI_AGENT_API_ENDPOINT) !== null && _ref !== void 0 ? _ref : API_ENDPOINT).replace(/\/$/, "");
+}
+function getOpenAIProxyBaseUrl(baseUrl) {
+	return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/openai`;
+}
+function getGoogleProxyBaseUrl(baseUrl) {
+	return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/google`;
+}
+function getAnthropicProxyBaseUrl(baseUrl) {
+	return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/anthropic`;
+}
+function createSpstLanguageModel(modelId, providers, defaultCallerApp) {
+	const parsedModelId = parseSpstModelId(modelId);
+	switch (parsedModelId.provider) {
+		case "openai": return wrapSpstModel(providers.openai.responses(parsedModelId.proxyModelId), defaultCallerApp);
+		case "google": return wrapSpstModel(providers.google(parsedModelId.proxyModelId), defaultCallerApp);
+		case "anthropic": return wrapSpstModel(providers.anthropic(parsedModelId.proxyModelId), defaultCallerApp);
+	}
+}
+function parseSpstModelId(modelId) {
+	if (modelId.startsWith("openai/")) return {
+		provider: "openai",
+		proxyModelId: modelId
+	};
+	if (modelId.startsWith("google/")) return {
+		provider: "google",
+		proxyModelId: modelId.slice(7)
+	};
+	if (modelId.startsWith("anthropic/")) return {
+		provider: "anthropic",
+		proxyModelId: modelId.slice(10)
+	};
+	if (modelId.includes("/")) throw new Error(`Provider '${modelId.split("/")[0]}' is not supported by spst() yet.`);
+	throw new Error("The spst() helper currently requires an 'openai/<model>', 'google/<model>', or 'anthropic/<model>' model id.");
+}
+function wrapSpstModel(model, defaultCallerApp) {
+	return {
+		specificationVersion: model.specificationVersion,
+		provider: model.provider,
+		modelId: model.modelId,
+		supportedUrls: model.supportedUrls,
+		doGenerate: (options) => model.doGenerate(withSpstHeaders(options, defaultCallerApp)),
+		doStream: (options) => model.doStream(withSpstHeaders(options, defaultCallerApp))
+	};
+}
+function withSpstHeaders(options, defaultCallerApp) {
+	var _options$providerOpti, _spstOptions$callerAp, _options$providerOpti2;
+	const spstOptions = (_options$providerOpti = options.providerOptions) === null || _options$providerOpti === void 0 ? void 0 : _options$providerOpti.spst;
+	if (!spstOptions) return options;
+	const callerApp = (_spstOptions$callerAp = spstOptions.callerApp) !== null && _spstOptions$callerAp !== void 0 ? _spstOptions$callerAp : defaultCallerApp;
+	if (!spstOptions.projectId || !spstOptions.email || !callerApp) throw new Error("The spst provider requires projectId, email, and callerApp in providerOptions.spst.");
+	const _ref2 = (_options$providerOpti2 = options.providerOptions) !== null && _options$providerOpti2 !== void 0 ? _options$providerOpti2 : {}, { spst: _spst } = _ref2, providerOptions = _objectWithoutProperties(_ref2, _excluded);
+	return _objectSpread2(_objectSpread2({}, options), {}, {
+		headers: _objectSpread2(_objectSpread2({}, options.headers), {}, {
+			"x-spst-project-id": spstOptions.projectId,
+			"x-spst-user-email": spstOptions.email,
+			"x-spst-caller-app": callerApp
+		}),
+		providerOptions
+	});
+}
+
+//#endregion
+export { createSpstProvider, spst };
+//# sourceMappingURL=provider.mjs.map

package/dist/provider.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"provider.mjs","names":[],"sources":["../src/provider.ts"],"sourcesContent":["import { createAnthropic } from \"@ai-sdk/anthropic\";\nimport { createVertex } from \"@ai-sdk/google-vertex\";\nimport { createOpenAI } from \"@ai-sdk/openai\";\nimport type { LanguageModelV3 } from \"@ai-sdk/provider\";\nimport { API_ENDPOINT } from \"@/lib/constants\";\nimport { createGoogleProxyFetch } from \"@/lib/google-proxy-fetch\";\n\nexport type SpstProviderOptions = {\n  projectId: string;\n  email: string;\n  callerApp: string;\n};\n\nexport type CreateSpstProviderParams = {\n  apiKey?: string;\n  baseUrl?: string;\n  callerApp: string;\n};\n\ntype SpstProviderConfig = Pick<CreateSpstProviderParams, \"apiKey\" | \"baseUrl\">;\ntype SpstProviderSet = {\n  openai: ReturnType<typeof createOpenAI>;\n  google: ReturnType<typeof createVertex>;\n  anthropic: ReturnType<typeof createAnthropic>;\n};\ntype ParsedSpstModelId =\n  | {\n      provider: \"openai\";\n      proxyModelId: string;\n    }\n  | {\n      provider: \"google\";\n      proxyModelId: string;\n    }\n  | {\n      provider: \"anthropic\";\n      proxyModelId: string;\n    };\n\nconst spstProviders = createSpstProviders({});\n\nexport function spst(modelId: string): LanguageModelV3 {\n  return createSpstLanguageModel(modelId, spstProviders);\n}\n\nexport function createSpstProvider(\n  params: CreateSpstProviderParams,\n): typeof spst {\n  const providers = createSpstProviders(params);\n\n  return (modelId: string) => {\n    return createSpstLanguageModel(modelId, providers, params.callerApp);\n  };\n}\n\nfunction createSpstProviders(config: SpstProviderConfig): SpstProviderSet {\n  const apiKey = config.apiKey ?? process.env.AI_AGENT_API_KEY;\n\n  return {\n    openai: createOpenAI({\n      name: \"spst\",\n      baseURL: getOpenAIProxyBaseUrl(config.baseUrl),\n      apiKey,\n    }),\n    google: createVertex({\n      apiKey,\n      baseURL: getGoogleProxyBaseUrl(config.baseUrl),\n      fetch: createGoogleProxyFetch(apiKey),\n    }),\n    anthropic: createAnthropic({\n      baseURL: getAnthropicProxyBaseUrl(config.baseUrl),\n      apiKey,\n    }),\n  };\n}\n\nfunction getResolvedBaseUrl(baseUrl?: string) {\n  const resolvedBaseUrl =\n    baseUrl ?? process.env.AI_AGENT_API_ENDPOINT ?? API_ENDPOINT;\n  return resolvedBaseUrl.replace(/\\/$/, \"\");\n}\n\nfunction getOpenAIProxyBaseUrl(baseUrl?: string) {\n  return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/openai`;\n}\n\nfunction getGoogleProxyBaseUrl(baseUrl?: string) {\n  return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/google`;\n}\n\nfunction getAnthropicProxyBaseUrl(baseUrl?: string) {\n  return `${getResolvedBaseUrl(baseUrl)}/api/proxy/v1/anthropic`;\n}\n\nfunction createSpstLanguageModel(\n  modelId: string,\n  providers: SpstProviderSet,\n  defaultCallerApp?: string,\n) {\n  const parsedModelId = parseSpstModelId(modelId);\n\n  switch (parsedModelId.provider) {\n    case \"openai\":\n      return wrapSpstModel(\n        providers.openai.responses(parsedModelId.proxyModelId),\n        defaultCallerApp,\n      );\n    case \"google\":\n      return wrapSpstModel(\n        providers.google(parsedModelId.proxyModelId),\n        defaultCallerApp,\n      );\n    case \"anthropic\":\n      return wrapSpstModel(\n        providers.anthropic(parsedModelId.proxyModelId),\n        defaultCallerApp,\n      );\n  }\n}\n\nfunction parseSpstModelId(modelId: string): ParsedSpstModelId {\n  if (modelId.startsWith(\"openai/\")) {\n    return {\n      provider: \"openai\",\n      proxyModelId: modelId,\n    };\n  }\n\n  if (modelId.startsWith(\"google/\")) {\n    return {\n      provider: \"google\",\n      proxyModelId: modelId.slice(\"google/\".length),\n    };\n  }\n\n  if (modelId.startsWith(\"anthropic/\")) {\n    return {\n      provider: \"anthropic\",\n      proxyModelId: modelId.slice(\"anthropic/\".length),\n    };\n  }\n\n  if (modelId.includes(\"/\")) {\n    throw new Error(\n      `Provider '${modelId.split(\"/\")[0]}' is not supported by spst() yet.`,\n    );\n  }\n\n  throw new Error(\n    \"The spst() helper currently requires an 'openai/<model>', 'google/<model>', or 'anthropic/<model>' model id.\",\n  );\n}\n\nfunction wrapSpstModel(\n  model: LanguageModelV3,\n  defaultCallerApp?: string,\n): LanguageModelV3 {\n  return {\n    specificationVersion: model.specificationVersion,\n    provider: model.provider,\n    modelId: model.modelId,\n    supportedUrls: model.supportedUrls,\n    doGenerate: (options) =>\n      model.doGenerate(withSpstHeaders(options, defaultCallerApp)),\n    doStream: (options) =>\n      model.doStream(withSpstHeaders(options, defaultCallerApp)),\n  };\n}\n\nfunction withSpstHeaders(\n  options: Parameters<LanguageModelV3[\"doGenerate\"]>[0],\n  defaultCallerApp?: string,\n): Parameters<LanguageModelV3[\"doGenerate\"]>[0] {\n  const spstOptions = options.providerOptions?.spst as\n    | Partial<SpstProviderOptions>\n    | undefined;\n\n  if (!spstOptions) {\n    return options;\n  }\n\n  const callerApp = spstOptions.callerApp ?? defaultCallerApp;\n\n  if (!spstOptions.projectId || !spstOptions.email || !callerApp) {\n    throw new Error(\n      \"The spst provider requires projectId, email, and callerApp in providerOptions.spst.\",\n    );\n  }\n\n  const { spst: _spst, ...providerOptions } = options.providerOptions ?? {};\n\n  return {\n    ...options,\n    headers: {\n      ...options.headers,\n      \"x-spst-project-id\": spstOptions.projectId,\n      \"x-spst-user-email\": spstOptions.email,\n      \"x-spst-caller-app\": callerApp,\n    },\n    providerOptions,\n  };\n}\n"],"mappings":";;;;;;;;;mBA6LU;AAtJV,MAAM,gBAAgB,oBAAoB,EAAE,CAAC;AAE7C,SAAgB,KAAK,SAAkC;AACrD,QAAO,wBAAwB,SAAS,cAAc;;AAGxD,SAAgB,mBACd,QACa;CACb,MAAM,YAAY,oBAAoB,OAAO;AAE7C,SAAQ,YAAoB;AAC1B,SAAO,wBAAwB,SAAS,WAAW,OAAO,UAAU;;;AAIxE,SAAS,oBAAoB,QAA6C;;CACxE,MAAM,2BAAS,OAAO,iEAAU,QAAQ,IAAI;AAE5C,QAAO;EACL,QAAQ,aAAa;GACnB,MAAM;GACN,SAAS,sBAAsB,OAAO,QAAQ;GAC9C;GACD,CAAC;EACF,QAAQ,aAAa;GACnB;GACA,SAAS,sBAAsB,OAAO,QAAQ;GAC9C,OAAO,uBAAuB,OAAO;GACtC,CAAC;EACF,WAAW,gBAAgB;GACzB,SAAS,yBAAyB,OAAO,QAAQ;GACjD;GACD,CAAC;EACH;;AAGH,SAAS,mBAAmB,SAAkB;;AAG5C,iBADE,mDAAW,QAAQ,IAAI,4DAAyB,cAC3B,QAAQ,OAAO,GAAG;;AAG3C,SAAS,sBAAsB,SAAkB;AAC/C,QAAO,GAAG,mBAAmB,QAAQ,CAAC;;AAGxC,SAAS,sBAAsB,SAAkB;AAC/C,QAAO,GAAG,mBAAmB,QAAQ,CAAC;;AAGxC,SAAS,yBAAyB,SAAkB;AAClD,QAAO,GAAG,mBAAmB,QAAQ,CAAC;;AAGxC,SAAS,wBACP,SACA,WACA,kBACA;CACA,MAAM,gBAAgB,iBAAiB,QAAQ;AAE/C,SAAQ,cAAc,UAAtB;EACE,KAAK,SACH,QAAO,cACL,UAAU,OAAO,UAAU,cAAc,aAAa,EACtD,iBACD;EACH,KAAK,SACH,QAAO,cACL,UAAU,OAAO,cAAc,aAAa,EAC5C,iBACD;EACH,KAAK,YACH,QAAO,cACL,UAAU,UAAU,cAAc,aAAa,EAC/C,iBACD;;;AAIP,SAAS,iBAAiB,SAAoC;AAC5D,KAAI,QAAQ,WAAW,UAAU,CAC/B,QAAO;EACL,UAAU;EACV,cAAc;EACf;AAGH,KAAI,QAAQ,WAAW,UAAU,CAC/B,QAAO;EACL,UAAU;EACV,cAAc,QAAQ,MAAM,EAAiB;EAC9C;AAGH,KAAI,QAAQ,WAAW,aAAa,CAClC,QAAO;EACL,UAAU;EACV,cAAc,QAAQ,MAAM,GAAoB;EACjD;AAGH,KAAI,QAAQ,SAAS,IAAI,CACvB,OAAM,IAAI,MACR,aAAa,QAAQ,MAAM,IAAI,CAAC,GAAG,mCACpC;AAGH,OAAM,IAAI,MACR,+GACD;;AAGH,SAAS,cACP,OACA,kBACiB;AACjB,QAAO;EACL,sBAAsB,MAAM;EAC5B,UAAU,MAAM;EAChB,SAAS,MAAM;EACf,eAAe,MAAM;EACrB,aAAa,YACX,MAAM,WAAW,gBAAgB,SAAS,iBAAiB,CAAC;EAC9D,WAAW,YACT,MAAM,SAAS,gBAAgB,SAAS,iBAAiB,CAAC;EAC7D;;AAGH,SAAS,gBACP,SACA,kBAC8C;;CAC9C,MAAM,uCAAc,QAAQ,+FAAiB;AAI7C,KAAI,CAAC,YACH,QAAO;CAGT,MAAM,qCAAY,YAAY,kFAAa;AAE3C,KAAI,CAAC,YAAY,aAAa,CAAC,YAAY,SAAS,CAAC,UACnD,OAAM,IAAI,MACR,sFACD;CAGH,wCAA4C,QAAQ,0FAAmB,EAAE,EAAnE,EAAE,MAAM,iBAAU;AAExB,0CACK;EACH,2CACK,QAAQ;GACX,qBAAqB,YAAY;GACjC,qBAAqB,YAAY;GACjC,qBAAqB;;EAEvB"}