feat: カテゴリIDに基づいてカテゴリ情報を取得するAPIエンドポイントを追加し、アイテム登録時にカテゴリ情報を使用

Author: nomhiro

frontend/src/app/api/category/getbycategoryid/route.ts

@@ -9,7 +9,7 @@ import { getCosmosItemsByCategoryId } from "../../../../util/cosmos/document";
  */
 export const GET = async (req: NextRequest) => {
   try {
-    const { searchParams } = new URL(req.url);
+    const { searchParams } = req.nextUrl; // 修正: req.url から req.nextUrl に変更
     const categoryId = searchParams.get("categoryId");
 
     if (!categoryId || typeof categoryId !== "string") {

frontend/src/app/api/document/route.ts

@@ -3,6 +3,7 @@ import { registerItem } from "../../../util/cosmos/document";
 import { getEmbedding, getChatCompletions } from "../../../util/openai";
 import { v4 as uuidv4 } from "uuid";
 import { deleteItem } from "../../../util/cosmos/document"; // 修正: deleteItemをインポート
+import { getCategoryById } from "../../../util/cosmos/category"
 
 /**
  * POSTメソッドで新しいアイテムを登録するAPIエンドポイント
@@ -19,8 +20,13 @@ export const POST = async (req: NextRequest) => {
       return NextResponse.json({ message: "Invalid input" }, { status: 400 });
     }
 
+    console.log("🚀Registering item with category ID:", category_id, "and content:", content);
+
+    // カテゴリIDからカテゴリ情報を取得
+    const category = await getCategoryById(category_id);
+
     // OpenAIのAPIを使用してコンテンツのベクトルを取得
-    const contentVector = await getEmbedding(content);
+    const contentVector = await getEmbedding(`${category?.category} \n${content}`);
 
     // OpenAIのAPIを使用してドキュメントタイトルを取得
     const systemMessage = `あなたは優秀なライターです。与えられた内容からタイトルを生成してください。\n\n- タイトルのみを出力してください。「」など不要です。`;

frontend/src/util/cosmos/category.ts

@@ -22,6 +22,33 @@ export const getAllCategories = async (): Promise<CategoryItem[]> => {
   });
 };
 
+/**
+ * categoryをIDで取得する
+ * @param id - カテゴリのID
+ * @returns - カテゴリ情報
+ */
+export const getCategoryById = async (id: string): Promise<CategoryItem | null> => {
+  return new Promise(async (resolve, reject) => {
+    const cosmosClient = new CosmosClient(process.env.COSMOS_CONNECTION_STRING!);
+    const database = cosmosClient.database(process.env.COSMOS_DATABASE_NAME!);
+    const container = database.container(process.env.COSMOS_CATEGORY_CONTAINER_NAME!);
+
+    try {
+      const { resource } = await container.item(id, id).read<CategoryItem>();
+      console.log("🚀Category retrieved successfully.");
+      resolve(resource || null); // カテゴリが見つからない場合はnullを返す
+    } catch (error) {
+      if ((error as any).code === 404) {
+        console.log("🚀Category not found.");
+        resolve(null);
+      } else {
+        console.error("🚀Error retrieving category:", error);
+        reject(error);
+      }
+    }
+  });
+}
+
 /**
  * categoryを登録する
  * @param category - 登録するカテゴリ