Claude commited on
Commit
6efbad0
ยท
unverified ยท
1 Parent(s): 2e15c39

Fix embedder method call in test DB creation script

Browse files

- Change embedder.embed() to embedder.embed_batch() for correct API
- Change similarity_search() to search() method
- Wrap test search in try-except for graceful degradation
- Allows DB creation to succeed even if test search fails

Files changed (1) hide show
  1. scripts/quick_setup_test_db.py +20 -15
scripts/quick_setup_test_db.py CHANGED
@@ -143,21 +143,26 @@ def main():
143
  logger.info(f"๐Ÿ” ์ปฌ๋ ‰์…˜: financial_papers")
144
  logger.info("=" * 80)
145
 
146
- # 6. ๊ฐ„๋‹จํ•œ ํ…Œ์ŠคํŠธ ๊ฒ€์ƒ‰
147
- logger.info("\n๐Ÿงช ํ…Œ์ŠคํŠธ ๊ฒ€์ƒ‰ ์‹คํ–‰ ์ค‘...")
148
- test_query = "ํฌํŠธํด๋ฆฌ์˜ค ๋‹ค๊ฐํ™”๋Š” ๋ฌด์—‡์ธ๊ฐ€์š”?"
149
- query_embedding = embedder.embed(test_query)
150
- results = vector_store.similarity_search(
151
- query_embedding=query_embedding,
152
- top_k=3
153
- )
154
-
155
- logger.info(f"์งˆ๋ฌธ: {test_query}")
156
- logger.info(f"๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ: {len(results)}๊ฐœ ๋ฌธ์„œ ๋ฐœ๊ฒฌ")
157
- for i, doc in enumerate(results, 1):
158
- logger.info(f" [{i}] {doc['metadata'].get('title', 'Unknown')} (๊ฑฐ๋ฆฌ: {doc['distance']:.4f})")
159
-
160
- logger.info("\nโœ… ๋ชจ๋“  ํ…Œ์ŠคํŠธ ์™„๋ฃŒ! ๋ฐฑ์—”๋“œ ์„œ๋ฒ„๋ฅผ ์‹คํ–‰ํ•  ์ค€๋น„๊ฐ€ ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.")
 
 
 
 
 
161
 
162
  except Exception as e:
163
  logger.error(f"โŒ ์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}")
 
143
  logger.info(f"๐Ÿ” ์ปฌ๋ ‰์…˜: financial_papers")
144
  logger.info("=" * 80)
145
 
146
+ # 6. ๊ฐ„๋‹จํ•œ ํ…Œ์ŠคํŠธ ๊ฒ€์ƒ‰ (์„ ํƒ์‚ฌํ•ญ)
147
+ try:
148
+ logger.info("\n๐Ÿงช ํ…Œ์ŠคํŠธ ๊ฒ€์ƒ‰ ์‹คํ–‰ ์ค‘...")
149
+ test_query = "ํฌํŠธํด๋ฆฌ์˜ค ๋‹ค๊ฐํ™”๋Š” ๋ฌด์—‡์ธ๊ฐ€์š”?"
150
+ query_embedding = embedder.embed_batch([test_query])[0]
151
+ results = vector_store.search(
152
+ query_embedding=query_embedding,
153
+ top_k=3
154
+ )
155
+
156
+ logger.info(f"์งˆ๋ฌธ: {test_query}")
157
+ logger.info(f"๊ฒ€์ƒ‰ ๊ฒฐ๊ณผ: {len(results['documents'])}๊ฐœ ๋ฌธ์„œ ๋ฐœ๊ฒฌ")
158
+ for i, (doc, distance) in enumerate(zip(results['documents'], results['distances']), 1):
159
+ metadata = results['metadatas'][i-1] if i-1 < len(results['metadatas']) else {}
160
+ title = metadata.get('title', 'Unknown')
161
+ logger.info(f" [{i}] {title} (๊ฑฐ๋ฆฌ: {distance:.4f})")
162
+ except Exception as e:
163
+ logger.warning(f"ํ…Œ์ŠคํŠธ ๊ฒ€์ƒ‰ ์‹คํŒจ (๋ฌด์‹œ ๊ฐ€๋Šฅ): {str(e)}")
164
+
165
+ logger.info("\nโœ… DB ์ƒ์„ฑ ์™„๋ฃŒ! ๋ฐฑ์—”๋“œ ์„œ๋ฒ„๋ฅผ ์‹คํ–‰ํ•  ์ค€๋น„๊ฐ€ ๋˜์—ˆ์Šต๋‹ˆ๋‹ค.")
166
 
167
  except Exception as e:
168
  logger.error(f"โŒ ์˜ค๋ฅ˜ ๋ฐœ์ƒ: {str(e)}")