chore: Update dependencies and add tenacity library for retrying vali…

…dation errors
sheakes · Jun 9, 2024 · 5f34d90 · 5f34d90
1 parent 9d05f90
commit 5f34d90
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/main.py b/main.py
@@ -8,6 +8,8 @@
 from promptic import llm
 from pydantic import BaseModel
 from pypdf import PdfReader
+from tenacity import retry, retry_if_exception_type
+from pydantic import ValidationError
 
 
 class DialogueItem(BaseModel):
@@ -28,6 +30,7 @@ class Dialogue(BaseModel):
     dialogue: List[DialogueItem]
 
 
+@retry(retry=retry_if_exception_type(ValidationError))
 @llm(model="gemini/gemini-1.5-flash")
 def generate_dialogue(text: str) -> Dialogue:
     """

diff --git a/requirements.txt b/requirements.txt
@@ -4,3 +4,4 @@ pydantic~=2.7
 google-generativeai~=0.6
 loguru~=0.7
 pypdf~=4.1
+tenacity~=8.3