Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,6 @@ from typing import List, Dict, Tuple, Any
|
|
| 7 |
from pgvector.psycopg2 import register_vector
|
| 8 |
import numpy as np
|
| 9 |
from datetime import datetime
|
| 10 |
-
from sklearn.preprocessing import normalize
|
| 11 |
|
| 12 |
# ๊ฐ์ค์น ๋ฐ ์๊ณ๊ฐ ์ค์
|
| 13 |
DEFAULT_FULL_WEIGHT = 0.2
|
|
@@ -69,11 +68,9 @@ def search_similar_chat(query: str, max_results: int = 100) -> List[Dict]:
|
|
| 69 |
print(f"๋ค์ค ์๋ฒ ๋ฉ ๊ฒ์ ์์: ์ฟผ๋ฆฌ='{query}', ๊ฐ์ค์น=(full={full_w}, topic={topic_w}, customer={customer_w}, agent={agent_w}), ์ต๋ ๊ฒฐ๊ณผ={limit}")
|
| 70 |
|
| 71 |
try:
|
| 72 |
-
# ์ฟผ๋ฆฌ ์๋ฒ ๋ฉ ์์ฑ
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
query_embedding = normalize(raw_embedding.reshape(1, -1), norm='l2')[0]
|
| 76 |
-
print(f"์๋ฒ ๋ฉ ์ ๊ทํ ์ /ํ ์ฒซ 5๊ฐ ์์: {raw_embedding[:5]} -> {query_embedding[:5]}")
|
| 77 |
|
| 78 |
# Java ๋ฐฉ์: ๋ฒกํฐ๋ฅผ ๋ฌธ์์ด๋ก ๋ณํ
|
| 79 |
query_vector = format_vector_for_pg(query_embedding)
|
|
@@ -220,11 +217,9 @@ def search_similar_chat_by_date(
|
|
| 220 |
print(f"์ข
๋ฃ ๋ ์ง ํ์ ์ค๋ฅ: {str(e)}")
|
| 221 |
return []
|
| 222 |
|
| 223 |
-
# ์ฟผ๋ฆฌ ์๋ฒ ๋ฉ ์์ฑ
|
| 224 |
-
|
| 225 |
-
|
| 226 |
-
query_embedding = normalize(raw_embedding.reshape(1, -1), norm='l2')[0]
|
| 227 |
-
print(f"๋ ์ง ๊ฒ์ - ์๋ฒ ๋ฉ ์ ๊ทํ ์ /ํ ์ฒซ 5๊ฐ ์์: {raw_embedding[:5]} -> {query_embedding[:5]}")
|
| 228 |
|
| 229 |
# Java ๋ฐฉ์: ๋ฒกํฐ๋ฅผ ๋ฌธ์์ด๋ก ๋ณํ
|
| 230 |
query_vector = format_vector_for_pg(query_embedding)
|
|
|
|
| 7 |
from pgvector.psycopg2 import register_vector
|
| 8 |
import numpy as np
|
| 9 |
from datetime import datetime
|
|
|
|
| 10 |
|
| 11 |
# ๊ฐ์ค์น ๋ฐ ์๊ณ๊ฐ ์ค์
|
| 12 |
DEFAULT_FULL_WEIGHT = 0.2
|
|
|
|
| 68 |
print(f"๋ค์ค ์๋ฒ ๋ฉ ๊ฒ์ ์์: ์ฟผ๋ฆฌ='{query}', ๊ฐ์ค์น=(full={full_w}, topic={topic_w}, customer={customer_w}, agent={agent_w}), ์ต๋ ๊ฒฐ๊ณผ={limit}")
|
| 69 |
|
| 70 |
try:
|
| 71 |
+
# ์ฟผ๋ฆฌ ์๋ฒ ๋ฉ ์์ฑ - ์ ๊ทํ ์ ๊ฑฐ
|
| 72 |
+
query_embedding = get_embedding(query)
|
| 73 |
+
print(f"์๋ฒ ๋ฉ ์์ฑ ์๋ฃ: ์ฒซ 5๊ฐ ์์: {query_embedding[:5]}")
|
|
|
|
|
|
|
| 74 |
|
| 75 |
# Java ๋ฐฉ์: ๋ฒกํฐ๋ฅผ ๋ฌธ์์ด๋ก ๋ณํ
|
| 76 |
query_vector = format_vector_for_pg(query_embedding)
|
|
|
|
| 217 |
print(f"์ข
๋ฃ ๋ ์ง ํ์ ์ค๋ฅ: {str(e)}")
|
| 218 |
return []
|
| 219 |
|
| 220 |
+
# ์ฟผ๋ฆฌ ์๋ฒ ๋ฉ ์์ฑ - ์ ๊ทํ ์ ๊ฑฐ
|
| 221 |
+
query_embedding = get_embedding(query)
|
| 222 |
+
print(f"๋ ์ง ๊ฒ์ - ์๋ฒ ๋ฉ ์์ฑ ์๋ฃ: ์ฒซ 5๊ฐ ์์: {query_embedding[:5]}")
|
|
|
|
|
|
|
| 223 |
|
| 224 |
# Java ๋ฐฉ์: ๋ฒกํฐ๋ฅผ ๋ฌธ์์ด๋ก ๋ณํ
|
| 225 |
query_vector = format_vector_for_pg(query_embedding)
|