Include error handling while fetching data

anasserhussien · Apr 28, 2024 · 36aff1e · 36aff1e
1 parent cc18e3d
commit 36aff1e
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/main.py b/main.py
@@ -99,9 +99,13 @@ async def read_wiki_url(request: Request, wiki: Wiki):
 
     url = wiki.wiki_url
     error_message = ""
-    docs = load_wiki_article(url)
-    splits = wiki_text_splitter(docs)
-    await create_db(splits)
+    try:
+        docs = load_wiki_article(url)
+        splits = wiki_text_splitter(docs)
+        await create_db(splits)
+    except Exception as e:
+        print(str(e))
+        raise Exception(str(e))
     #print('Finished execution')