davanstrien HF staff commited on
Commit
810d3a0
1 Parent(s): 4c88070

Fix error handling in get_summary function

Browse files
Files changed (1) hide show
  1. main.py +4 -2
main.py CHANGED
@@ -215,13 +215,15 @@ async def get_summary(dataset_id: str) -> str | None:
215
  try:
216
  # dataset_id = request.dataset_id
217
  try:
218
- card_text = await async_httpx_client.get(
219
  f"https://huggingface.co/datasets/{dataset_id}/raw/main/README.md"
220
  )
 
 
221
  except Exception as e:
222
  logger.error(e)
223
  return None
224
- card_text = card_text.text
225
  card = DatasetCard(card_text)
226
  text = card.text
227
  parsed_text = parse_markdown(text)
 
215
  try:
216
  # dataset_id = request.dataset_id
217
  try:
218
+ resp = await async_httpx_client.get(
219
  f"https://huggingface.co/datasets/{dataset_id}/raw/main/README.md"
220
  )
221
+ if resp.status_code != 200:
222
+ return None
223
  except Exception as e:
224
  logger.error(e)
225
  return None
226
+ card_text = resp.text
227
  card = DatasetCard(card_text)
228
  text = card.text
229
  parsed_text = parse_markdown(text)