fix: timeout

This commit is contained in:
yihong0618 2024-08-02 14:14:40 +08:00
parent 34571633a2
commit 9ca9e1b602
2 changed files with 3 additions and 3 deletions

@ -22,10 +22,10 @@ ANTHROPIC_PRO_MODEL = "claude-3-opus-20240229"
if environ.get("ANTHROPIC_BASE_URL"):
client = Anthropic(
base_url=ANTHROPIC_BASE_URL, api_key=ANTHROPIC_API_KEY, timeout=20
base_url=ANTHROPIC_BASE_URL, api_key=ANTHROPIC_API_KEY
)
else:
client = Anthropic(api_key=ANTHROPIC_API_KEY, timeout=20)
client = Anthropic(api_key=ANTHROPIC_API_KEY)
# Global history cache

@ -80,7 +80,7 @@ if (CHATGPT_USE or CHATGPT_COMPLETE or CHATGPT_APPEND) and CHATGPT_API_KEY:
from openai import OpenAI
CHATGPT_PRO_MODEL = "gpt-4o-2024-05-13"
client = OpenAI(api_key=CHATGPT_API_KEY, base_url=CHATGPT_BASE_URL, timeout=300)
client = OpenAI(api_key=CHATGPT_API_KEY, base_url=CHATGPT_BASE_URL)
#### Gemini init ####