Jake-seong commited on
Commit
70b2365
·
verified ·
1 Parent(s): f17c04f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -89,10 +89,10 @@ def search_similar_chat(query: str, max_results: int = 100) -> List[Dict]:
89
  id,
90
  metadata,
91
  content,
92
- CASE WHEN full_embedding IS NOT NULL THEN 1 - (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
93
- CASE WHEN topic_embedding IS NOT NULL THEN 1 - (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
94
- CASE WHEN customer_embedding IS NOT NULL THEN 1 - (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
95
- CASE WHEN agent_embedding IS NOT NULL THEN 1 - (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
96
  FROM vector_store_multi_embeddings
97
  WHERE full_embedding IS NOT NULL
98
  OR topic_embedding IS NOT NULL
@@ -240,10 +240,10 @@ def search_similar_chat_by_date(
240
  id,
241
  metadata,
242
  content,
243
- CASE WHEN full_embedding IS NOT NULL THEN 1 - (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
244
- CASE WHEN topic_embedding IS NOT NULL THEN 1 - (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
245
- CASE WHEN customer_embedding IS NOT NULL THEN 1 - (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
246
- CASE WHEN agent_embedding IS NOT NULL THEN 1 - (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
247
  FROM vector_store_multi_embeddings
248
  WHERE full_embedding IS NOT NULL
249
  OR topic_embedding IS NOT NULL
 
89
  id,
90
  metadata,
91
  content,
92
+ CASE WHEN full_embedding IS NOT NULL THEN (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
93
+ CASE WHEN topic_embedding IS NOT NULL THEN (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
94
+ CASE WHEN customer_embedding IS NOT NULL THEN (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
95
+ CASE WHEN agent_embedding IS NOT NULL THEN (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
96
  FROM vector_store_multi_embeddings
97
  WHERE full_embedding IS NOT NULL
98
  OR topic_embedding IS NOT NULL
 
240
  id,
241
  metadata,
242
  content,
243
+ CASE WHEN full_embedding IS NOT NULL THEN (full_embedding <=> '{query_vector}'::vector) ELSE 0 END * {full_w} as full_sim,
244
+ CASE WHEN topic_embedding IS NOT NULL THEN (topic_embedding <=> '{query_vector}'::vector) ELSE 0 END * {topic_w} as topic_sim,
245
+ CASE WHEN customer_embedding IS NOT NULL THEN (customer_embedding <=> '{query_vector}'::vector) ELSE 0 END * {customer_w} as customer_sim,
246
+ CASE WHEN agent_embedding IS NOT NULL THEN (agent_embedding <=> '{query_vector}'::vector) ELSE 0 END * {agent_w} as agent_sim
247
  FROM vector_store_multi_embeddings
248
  WHERE full_embedding IS NOT NULL
249
  OR topic_embedding IS NOT NULL