Spaces:
Running
Running
Commit ·
ae9c0d8
1
Parent(s): 5f90488
update
Browse files- config.py +2 -2
- retrieve.py +4 -1
config.py
CHANGED
|
@@ -22,11 +22,11 @@ def get_config() -> Dict[str, Any]:
|
|
| 22 |
"cohere_api_key": os.getenv("COHERE_API_KEY"),
|
| 23 |
"qdrant_url": os.getenv("QDRANT_URL"),
|
| 24 |
"qdrant_api_key": os.getenv("QDRANT_API_KEY"),
|
| 25 |
-
"qdrant_collection": os.getenv("QDRANT_COLLECTION", "
|
| 26 |
}
|
| 27 |
|
| 28 |
# Validate required keys
|
| 29 |
-
required_keys = ["
|
| 30 |
missing_keys = [key for key in required_keys if not config[key]]
|
| 31 |
|
| 32 |
if missing_keys:
|
|
|
|
| 22 |
"cohere_api_key": os.getenv("COHERE_API_KEY"),
|
| 23 |
"qdrant_url": os.getenv("QDRANT_URL"),
|
| 24 |
"qdrant_api_key": os.getenv("QDRANT_API_KEY"),
|
| 25 |
+
"qdrant_collection": os.getenv("QDRANT_COLLECTION", "book_embeddings"),
|
| 26 |
}
|
| 27 |
|
| 28 |
# Validate required keys
|
| 29 |
+
required_keys = ["cohere_api_key", "qdrant_url", "qdrant_api_key"]
|
| 30 |
missing_keys = [key for key in required_keys if not config[key]]
|
| 31 |
|
| 32 |
if missing_keys:
|
retrieve.py
CHANGED
|
@@ -64,7 +64,10 @@ def search(
|
|
| 64 |
return results
|
| 65 |
|
| 66 |
except Exception as e:
|
| 67 |
-
logger.error(
|
|
|
|
|
|
|
|
|
|
| 68 |
raise
|
| 69 |
|
| 70 |
|
|
|
|
| 64 |
return results
|
| 65 |
|
| 66 |
except Exception as e:
|
| 67 |
+
logger.error(
|
| 68 |
+
f"Search failed for query '{query_text[:100]}...': {type(e).__name__}: {e}",
|
| 69 |
+
exc_info=True,
|
| 70 |
+
)
|
| 71 |
raise
|
| 72 |
|
| 73 |
|