Rohan03 commited on
Commit
6048d78
·
verified ·
1 Parent(s): da8f259

Fix Issue 3: Add 60s default timeout to OpenAI client

Browse files
Files changed (1) hide show
  1. patches/openai_backend_init.py +15 -0
patches/openai_backend_init.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ def __init__(
2
+ self,
3
+ model: str = "gpt-4o",
4
+ base_url: str | None = None,
5
+ api_key: str | None = None,
6
+ timeout: float = 60.0,
7
+ ):
8
+ from openai import OpenAI
9
+
10
+ self.model = model
11
+ self.client = OpenAI(
12
+ base_url=base_url,
13
+ api_key=api_key or os.environ.get("OPENAI_API_KEY"),
14
+ timeout=timeout,
15
+ )