Corin1998 commited on
Commit
cfc19b1
·
verified ·
1 Parent(s): aba9c67

Create openai_client.py

Browse files
Files changed (1) hide show
  1. core/openai_client.py +14 -0
core/openai_client.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from openai import OpenAI
3
+ import gradio as gr
4
+
5
+ VISION_MODEL = os.environ.get("OPENAI_VISION_MODEL", "gpt-4o-mini")
6
+ TEXT_MODEL = os.environ.get("OPENAI_TEXT_MODEL", "gpt-4o-mini")
7
+
8
+ def get_client() -> OpenAI:
9
+ key = os.environ.get("OPENAI_API_KEY")
10
+ if not key:
11
+ # UIに素直に出す
12
+ raise gr.Error("OPENAI_API_KEY が未設定です。Spaces → Settings → Variables and secrets で追加してください。")
13
+ # proxies を渡さない(httpx==0.27.2 で安定)
14
+ return OpenAI(api_key=key, timeout=60)