@super_studio/ecforce-ai-agent-server 0.1.0

package/src/sdk/generate.ts

@@ -0,0 +1,66 @@
+import fs from "fs";
+import path from "path";
+import { generateApi } from "swagger-typescript-api";
+import { API_ENDPOINT } from "@/lib/constants";
+
+const SCRIPT_DIR = path.dirname(new URL(import.meta.url).pathname);
+
+/**
+ * こちらのスクリプトは以下のコマンドで実行してください
+ *
+ * `pnpm --filter web ma:sdk:generate`
+ */
+void (async () => {
+  await updateApiSchema();
+
+  await generateApi({
+    fileName: "index.ts",
+    input: path.resolve(SCRIPT_DIR, "./schema/openapi.json"),
+    output: path.resolve(SCRIPT_DIR, "./__generated__"),
+    moduleNameIndex: 1,
+    unwrapResponseData: true,
+    hooks: {
+      onFormatRouteName(routeInfo) {
+        const { method, route, summary } = routeInfo;
+
+        const isValidSummary = !!summary && /^[a-zA-Z]+$/.test(summary);
+
+        const methodName = isValidSummary
+          ? summary
+          : getFallbackMethodName(method, route);
+
+        return methodName;
+      },
+    },
+  });
+})();
+
+/**
+ * `get: /cx/app-test/prisma-test` -> `getPrismaTest`
+ */
+function getFallbackMethodName(method: string, route: string) {
+  const [, , , ...pathParts] = route.split("/");
+  const methodName = pathParts
+    .map((part) => {
+      // ハイフンを削除してキャメルケースに変換
+      return part
+        .split("-")
+        .map((segment) => segment.charAt(0).toUpperCase() + segment.slice(1))
+        .join("");
+    })
+    .join("");
+  return `${method}${methodName}`;
+}
+
+async function updateApiSchema() {
+  // スキーマを取得
+  const res = await fetch(`${API_ENDPOINT}/api/openapi.json`);
+  const json = (await res.json()) as unknown;
+  // フォルダが存在しない場合は作成
+  fs.mkdirSync(path.resolve(SCRIPT_DIR, "./schema"), { recursive: true });
+  // スキーマを保存
+  fs.writeFileSync(
+    path.resolve(SCRIPT_DIR, "./schema/openapi.json"),
+    JSON.stringify(json, null, 2),
+  );
+}

package/src/sdk/index.ts

@@ -0,0 +1,46 @@
+import { API_ENDPOINT } from "@/lib/constants";
+import { Api } from "./__generated__";
+
+export * from "./__generated__";
+
+export type CreateClientParams = {
+  /**
+   * 管理画面から作られたAPIキー。
+   * `AI_AGENT_API_KEY`の環境変数からも設定可能。
+   */
+  apiKey?: string;
+  /**
+   * エージェントのAPIエンドポイント。
+   * `AI_AGENT_API_ENDPOINT`の環境変数からも設定可能。
+   * デフォルトは本番のAPIエンドポイント。
+   * 基本的には変更不要。
+   */
+  baseUrl?: string;
+};
+
+export function createClient(params?: CreateClientParams) {
+  const baseUrl =
+    params?.baseUrl ?? process.env.AI_AGENT_API_ENDPOINT ?? API_ENDPOINT;
+  const apiKey = params?.apiKey ?? process.env.AI_AGENT_API_KEY;
+
+  if (!apiKey) {
+    throw new Error("APIキーが設定されていません。");
+  }
+  if (!baseUrl) {
+    throw new Error("APIエンドポイントが設定されていません。");
+  }
+
+  return new Api({
+    baseUrl,
+    baseApiParams: {
+      secure: true,
+    },
+    async securityWorker() {
+      return {
+        headers: {
+          Authorization: `Bearer ${apiKey}`,
+        },
+      };
+    },
+  });
+}