coosld/MoneyPrinterTurbo
1
0
Fork 0
mirror of https://github.com/harry0703/MoneyPrinterTurbo.git synced 2026-02-21 08:27:22 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #372 from harry0703/dev

Browse Source 
enhanced exception handling for generating terms
This commit is contained in:
Harry 2024-05-17 17:12:13 +08:00 committed by GitHub
commit a4692060a0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
app/services/llm.py
View File

@ -295,6 +295,7 @@ Please note that you must use English for generating video search terms; Chinese
logger.info(f"subject: {video_subject}")
search_terms = []
response = ""
for i in range(_max_retries):
try:
response = _generate_response(prompt)
@ -304,12 +305,15 @@ Please note that you must use English for generating video search terms; Chinese
continue
except Exception as e:
match = re.search(r'\[.*]', response)
if match:
try:
search_terms = json.loads(match.group())
except json.JSONDecodeError:
pass
logger.warning(f"failed to generate video terms: {str(e)}")
if response:
match = re.search(r'\[.*]', response)
if match:
try:
search_terms = json.loads(match.group())
except Exception as e:
logger.warning(f"failed to generate video terms: {str(e)}")
pass
if search_terms and len(search_terms) > 0:
break