Update app.py

app.py

@@ -14,7 +14,7 @@ DEFAULT_FULL_WEIGHT = 0.2
 DEFAULT_TOPIC_WEIGHT = 0.5
 DEFAULT_CUSTOMER_WEIGHT = 0.2
 DEFAULT_AGENT_WEIGHT = 0.1
-DEFAULT_SIMILARITY_THRESHOLD = 0
+DEFAULT_SIMILARITY_THRESHOLD = 0.7
 
 # DB 연결 설정
 def get_db_conn():
@@ -89,10 +89,10 @@ def search_similar_chat(query: str, max_results: int = 100) -> List[Dict]:
                     id, 
                     metadata, 
                     content,
-                    CASE WHEN full_embedding IS NOT NULL THEN (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
-                    CASE WHEN topic_embedding IS NOT NULL THEN (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
-                    CASE WHEN customer_embedding IS NOT NULL THEN (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
-                    CASE WHEN agent_embedding IS NOT NULL THEN (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
+                    CASE WHEN full_embedding IS NOT NULL THEN 1 - (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
+                    CASE WHEN topic_embedding IS NOT NULL THEN 1 - (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
+                    CASE WHEN customer_embedding IS NOT NULL THEN 1 - (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
+                    CASE WHEN agent_embedding IS NOT NULL THEN 1 - (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
                 FROM vector_store_multi_embeddings
                 WHERE full_embedding IS NOT NULL
                 OR topic_embedding IS NOT NULL
@@ -240,10 +240,10 @@ def search_similar_chat_by_date(
                     id, 
                     metadata, 
                     content,
-                    CASE WHEN full_embedding IS NOT NULL THEN (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
-                    CASE WHEN topic_embedding IS NOT NULL THEN (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
-                    CASE WHEN customer_embedding IS NOT NULL THEN (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
-                    CASE WHEN agent_embedding IS NOT NULL THEN (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
+                    CASE WHEN full_embedding IS NOT NULL THEN 1 - (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
+                    CASE WHEN topic_embedding IS NOT NULL THEN 1 - (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
+                    CASE WHEN customer_embedding IS NOT NULL THEN 1 - (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
+                    CASE WHEN agent_embedding IS NOT NULL THEN 1 - (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
                 FROM vector_store_multi_embeddings
                 WHERE full_embedding IS NOT NULL
                 OR topic_embedding IS NOT NULL