fix: 예외 처리

syk25 · Jul 24, 2024 · 6bc3fa1 · 6bc3fa1
1 parent c82e9d6
commit 6bc3fa1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/app/api/similarity.py b/app/api/similarity.py
@@ -40,9 +40,13 @@ def get_word_vector(word, model):
     AIvectorsA = [normalize_vector(get_word_vector(interest, model)) for interest in AIinterestsA]
     AIvectorsB = [normalize_vector(get_word_vector(interest, model)) for interest in AIinterestsB]
 
+    # 관심사 벡터가 비어있을 때 유사도를 0.0
     if not vectorsA or not vectorsB:
         return {"similarity": 0.0}
 
+    if not AIvectorsA or not AIvectorsB:
+        return {"similarity": 0.0}
+
     # 관심사 벡터의 평균 계산
     vectorA = normalize_vector(np.mean(vectorsA, axis=0))
     vectorB = normalize_vector(np.mean(vectorsB, axis=0))