aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGravatar Qetesh <4559341+Qetesh@users.noreply.github.com> 2024-08-19 22:04:17 +0800
committerGravatar Qetesh <4559341+Qetesh@users.noreply.github.com> 2024-08-19 22:04:17 +0800
commitff8e06f2d495497d111e45f368b6c3bb1f9088c1 (patch)
treeda422da8c9877f58ecab787c8c25769ac9ac1c22
parent9adf372d461ceee5d9fc3aef4429b262290636b4 (diff)
downloadminiflux-ai-ff8e06f2d495497d111e45f368b6c3bb1f9088c1.tar.gz
miniflux-ai-ff8e06f2d495497d111e45f368b6c3bb1f9088c1.tar.zst
miniflux-ai-ff8e06f2d495497d111e45f368b6c3bb1f9088c1.zip
add timeout, add err handlerv0.5
-rw-r--r--main.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/main.py b/main.py
index 22f213f..79bad01 100644
--- a/main.py
+++ b/main.py
@@ -27,7 +27,7 @@ def process_entry(entry):
(((agent[1]['whitelist'] is not None) and (entry['feed']['site_url'] in agent[1]['whitelist'])) or
(agent[1]['blacklist'] is not None and entry['feed']['site_url'] not in agent[1]['blacklist']) or
(agent[1]['whitelist'] is None and agent[1]['blacklist'] is None))):
- completion = llm_client.chat.completions.create( model=config['llm']['model'], messages= messages, timeout=60 )
+ completion = llm_client.chat.completions.create( model=config['llm']['model'], messages= messages, timeout=15 )
response_content = completion.choices[0].message.content
print(time.strftime("%Y/%m/%d %H:%M:%S", time.localtime()), agent[0], entry['feed']['feed_url'], response_content)