不要なインポートを削除

api/main.py

@@ -1,7 +1,6 @@
 import os
 import json
 import gzip
-from typing import List, Optional, Literal
 
 from fastapi import FastAPI, Response, Depends, HTTPException, Security, Query
 from fastapi.responses import RedirectResponse, JSONResponse
@@ -11,7 +10,6 @@ from pydantic import BaseModel
 from dotenv import load_dotenv
 
 from utils.logger import setup_logger
-from utils.json import field_getter
 from api.search.engine import SearchEngine
 import schemas.projects as schema_projects
 
@@ -57,7 +55,7 @@ def health_check():
 
 @app.get(
     "/api/projects",
-    response_model=List[schema_projects.ProjectSummary],
+    response_model=list[schema_projects.ProjectSummary],
     dependencies=[Depends(get_api_key)],
 )
 def get_summary_data():
@@ -104,11 +102,15 @@ def search(request: SearchRequest):
     if request.debug:
         pairs, diag = result  # type: ignore
         ids = [pid for pid, _ in pairs]
-        return JSONResponse(content={
-            "projectIds": ids,
-            "scores": [{"projectId": pid, "score": float(score)} for pid, score in pairs],
-            "details": diag.get("details", []),
-        })
+        return JSONResponse(
+            content={
+                "projectIds": ids,
+                "scores": [
+                    {"projectId": pid, "score": float(score)} for pid, score in pairs
+                ],
+                "details": diag.get("details", []),
+            }
+        )
     pairs = result  # type: ignore
     ids = [pid for pid, _ in pairs]
     return schema_projects.ProjectIds(projectIds=ids)

scripts/6_build_word_embeddings.py

@@ -1,7 +1,6 @@
 import os
 import sys
 import json
-import math
 from typing import Dict, List, Tuple
 
 import numpy as np
@@ -9,14 +8,13 @@ import numpy as np
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
 from utils.logger import setup_logger
-from utils.json import get_file_path_from_config, field_getter, json_dumps
+from utils.json import field_getter, json_dumps
 
 log = setup_logger(__name__)
 
 
 def load_configs():
     files = field_getter("config/files.json")
-    search = field_getter("config/search_model.json")
 
     paths = {
         "tf_token": files("bm25.tf_token"),
@@ -38,7 +36,9 @@ def read_vocab_from_tf_token(tf_token_path: str) -> Tuple[List[dict], set[str]]:
     return docs, vocab
 
 
-def stream_fasttext_vec(vec_path: str, vocab: set[str]) -> Tuple[Dict[str, int], np.ndarray]:
+def stream_fasttext_vec(
+    vec_path: str, vocab: set[str]
+) -> Tuple[Dict[str, int], np.ndarray]:
     """fastText .vec から、必要語彙のみ抽出してベクトル行列を返す。"""
     token_to_idx: Dict[str, int] = {}
     vectors: List[np.ndarray] = []
@@ -95,7 +95,9 @@ def stream_fasttext_vec(vec_path: str, vocab: set[str]) -> Tuple[Dict[str, int],
 
 
 def main():
-    log.info("語彙/文書ベクトル(word_vocab.json, word_vectors.npz, doc_vectors.npy)を生成します")
+    log.info(
+        "語彙/文書ベクトル(word_vocab.json, word_vectors.npz, doc_vectors.npy)を生成します"
+    )
     try:
         paths = load_configs()
     except Exception as e: