Jake-seong commited on
Commit
41d679b
ยท
verified ยท
1 Parent(s): d019bc4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -11
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
- raw_embedding = np.array(get_embedding(query))
74
- # L2 ์ •๊ทœํ™” ์ ์šฉ
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
- raw_embedding = np.array(get_embedding(query))
225
- # L2 ์ •๊ทœํ™” ์ ์šฉ
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)