Update app.py

app.py

@@ -4,6 +4,8 @@ from openai import OpenAI
 import json
 import os
 from typing import List, Dict
+from pgvector.psycopg2 import register_vector
+import numpy as np
 
 # DB 연결 설정
 def get_db_conn():
@@ -38,14 +40,15 @@ def search_similar_chats(query: str, maxResults: int = 10) -> List[Dict]:
     Returns:
         List[Dict]: 검색 결과 목록
     """
-    embedding = get_embedding(query)
+    embedding = np.array(get_embedding(query))  # numpy array로 변환
     conn = get_db_conn()
+    register_vector(conn)  # 벡터 타입 자동 변환 지원
     with conn.cursor() as cur:
         cur.execute("""
-            SELECT id, metadata, content, embedding <#> %s::vector AS distance
+            SELECT id, metadata, content, embedding <#> %s AS distance
             FROM vector_store
             WHERE metadata->>'documentType' = 'chatAnalysis'
-            ORDER BY embedding <#> %s::vector
+            ORDER BY embedding <#> %s
             LIMIT %s
         """, (embedding, embedding, maxResults))
         rows = cur.fetchall()