Spaces:
Sleeping
Sleeping
Update backend/main.py
Browse files- 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
|
| 195 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|