trish06 commited on
Commit
e68c878
·
verified ·
1 Parent(s): bc8aca9

Update backend/main.py

Browse files
Files changed (1) hide show
  1. backend/main.py +7 -2
backend/main.py CHANGED
@@ -191,8 +191,13 @@ def extract_protocol(client: Groq, pdf_path: str, filename: str) -> dict:
191
  "reagents, antibodies, and equipment settings.\n\nPAPER TEXT:\n" + paper_text
192
  }
193
  ]
194
- raw = call_groq_with_backoff(client, messages)
195
- result = parse_json_response(raw)
 
 
 
 
 
196
  result["_filename"] = filename
197
  return result
198
 
 
191
  "reagents, antibodies, and equipment settings.\n\nPAPER TEXT:\n" + paper_text
192
  }
193
  ]
194
+ raw = call_groq_with_backoff(client, messages)
195
+
196
+ try:
197
+ result = parse_json_response(raw)
198
+ except Exception as e:
199
+ raise ValueError(f"Invalid JSON returned by LLM: {str(e)}")
200
+
201
  result["_filename"] = filename
202
  return result
203