m-ahmad-official commited on
Commit
ae9c0d8
·
1 Parent(s): 5f90488
Files changed (2) hide show
  1. config.py +2 -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", "book-chunks"),
26
  }
27
 
28
  # Validate required keys
29
- required_keys = ["openai_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:
 
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(f"Search failed: {e}", exc_info=True)
 
 
 
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