Update app.py

app.py

@@ -7,6 +7,7 @@ from typing import List, Dict, Tuple, Any
 from pgvector.psycopg2 import register_vector
 import numpy as np
 from datetime import datetime
+from sklearn.preprocessing import normalize
 
 # 가중치 및 임계값 설정
 DEFAULT_FULL_WEIGHT = 0.2
@@ -65,7 +66,10 @@ def search_similar_chat(query: str, max_results: int = 100) -> List[Dict]:
     
     try:
         # 쿼리 임베딩 생성
-        query_embedding = np.array(get_embedding(query))
+        raw_embedding = np.array(get_embedding(query))
+        # L2 정규화 적용
+        query_embedding = normalize(raw_embedding.reshape(1, -1), norm='l2')[0]
+        print(f"임베딩 정규화 전/후 첫 5개 요소: {raw_embedding[:5]} -> {query_embedding[:5]}")
         
         # DB 연결
         conn = get_db_conn()
@@ -217,7 +221,10 @@ def search_similar_chat_by_date(
                 return []
         
         # 쿼리 임베딩 생성
-        query_embedding = np.array(get_embedding(query))
+        raw_embedding = np.array(get_embedding(query))
+        # L2 정규화 적용
+        query_embedding = normalize(raw_embedding.reshape(1, -1), norm='l2')[0]
+        print(f"날짜 검색 - 임베딩 정규화 전/후 첫 5개 요소: {raw_embedding[:5]} -> {query_embedding[:5]}")
         
         # DB 연결
         conn = get_db_conn()