copilot: foodies サービスで pagination が必要な API はサポートして

Author: ks6088ts

template_fastapi/repositories/restaurants.py

@@ -138,16 +138,17 @@ def delete_restaurant(self, restaurant_id: str) -> None:
         """指定されたIDのレストランを削除する"""
         self.container.delete_item(item=restaurant_id, partition_key=restaurant_id)
 
-    def search_restaurants(self, query: str, k: int = 3) -> list[Restaurant]:
-        """キーワードによるレストランのベクトル検索を実行する"""
+    def search_restaurants(self, query: str, k: int = 3, offset: int = 0) -> list[Restaurant]:
+        """キーワードによるレストランのベクトル検索を実行する（ページネーション対応）"""
         # クエリテキストのベクトル埋め込みを生成
         query_embedding = self._get_embeddings(query)
 
-        # ベクトル検索クエリの実行
+        # ベクトル検索クエリの実行（オフセットとリミットを適用）
         query_text = f"""
-        SELECT TOP {k} *
+        SELECT *
         FROM c
         ORDER BY VectorDistance(c.vector, @queryVector)
+        OFFSET {offset} LIMIT {k}
         """
 
         parameters = [{"name": "@queryVector", "value": query_embedding}]
@@ -157,18 +158,19 @@ def search_restaurants(self, query: str, k: int = 3) -> list[Restaurant]:
         return [self._cosmos_item_to_restaurant(item) for item in items]
 
     def find_nearby_restaurants(
-        self, latitude: float, longitude: float, distance_km: float = 5.0, limit: int = 10
+        self, latitude: float, longitude: float, distance_km: float = 5.0, limit: int = 10, offset: int = 0
     ) -> list[Restaurant]:
-        """指定した位置の近くにあるレストランを検索する"""
+        """指定した位置の近くにあるレストランを検索する（ページネーション対応）"""
         # 地理空間クエリの実行（メートル単位で距離を指定）
         distance_meters = distance_km * 1000
         query_text = f"""
-        SELECT TOP {limit} *
+        SELECT *
         FROM c
         WHERE ST_DISTANCE(c.location, {{
             "type": "Point",
             "coordinates": [{longitude}, {latitude}]
         }}) < {distance_meters}
+        OFFSET {offset} LIMIT {limit}
         """
 
         items = list(self.container.query_items(query=query_text, enable_cross_partition_query=True))

template_fastapi/routers/foodies.py

@@ -104,13 +104,15 @@ async def delete_foodies_restaurant(restaurant_id: str) -> dict:
     operation_id="search_foodies_restaurants",
 )
 async def search_foodies_restaurants(
-    query: str, k: int = Query(3, description="取得する上位結果の数")
+    query: str,
+    k: int = Query(3, description="取得する上位結果の数"),
+    offset: int = Query(0, description="スキップする件数（ページネーション用）"),
 ) -> list[Restaurant]:
     """
-    キーワードによるレストランのベクトル検索を実行する
+    キーワードによるレストランのベクトル検索を実行する（ページネーション対応）
     """
     try:
-        return restaurant_repo.search_restaurants(query, k)
+        return restaurant_repo.search_restaurants(query, k, offset)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"検索に失敗しました: {str(e)}")
 
@@ -126,11 +128,12 @@ async def find_nearby_restaurants(
     longitude: float = Query(..., description="経度"),
     distance_km: float = Query(5.0, description="検索半径（キロメートル）"),
     limit: int = Query(10, description="取得する最大件数"),
+    offset: int = Query(0, description="スキップする件数（ページネーション用）"),
 ) -> list[Restaurant]:
     """
-    指定した位置の近くにあるレストランを検索する
+    指定した位置の近くにあるレストランを検索する（ページネーション対応）
     """
     try:
-        return restaurant_repo.find_nearby_restaurants(latitude, longitude, distance_km, limit)
+        return restaurant_repo.find_nearby_restaurants(latitude, longitude, distance_km, limit, offset)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"位置検索に失敗しました: {str(e)}")