Spaces:
Sleeping
Sleeping
Claude commited on
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
- 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 |
-
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
| 160 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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)}")
|