cacode commited on
Commit
9a71750
·
verified ·
1 Parent(s): 70fc399

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +11 -2
app/main.py CHANGED
@@ -10,6 +10,8 @@ from datetime import datetime
10
  from io import BytesIO
11
  from urllib.parse import quote_plus
12
  from jinja2 import Template
 
 
13
  from openai import OpenAI
14
 
15
  # —— 环境变量读取 —— #
@@ -17,8 +19,15 @@ BAIDU_API_KEY = os.getenv("BAIDU_API_KEY", "")
17
  BAIDU_SECRET_KEY = os.getenv("BAIDU_SECRET_KEY", "")
18
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
19
 
20
- # —— 初始化 OpenAI v1 客户端 —— #
21
- client = OpenAI(api_key=OPENAI_API_KEY)
 
 
 
 
 
 
 
22
 
23
  # —— Baidu OCR —— #
24
  def get_access_token(api_key, secret_key):
 
10
  from io import BytesIO
11
  from urllib.parse import quote_plus
12
  from jinja2 import Template
13
+
14
+ import openai
15
  from openai import OpenAI
16
 
17
  # —— 环境变量读取 —— #
 
19
  BAIDU_SECRET_KEY = os.getenv("BAIDU_SECRET_KEY", "")
20
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
21
 
22
+ # —— 指定代理站 Base URL —— #
23
+ # 设置全局 module 级别(旧接口兼容)
24
+ openai.api_base = "https://free.v36.cm/v1/"
25
+
26
+ # 初始化 v1 客户端,并显式传入 api_base
27
+ client = OpenAI(
28
+ api_key=OPENAI_API_KEY,
29
+ api_base="https://free.v36.cm/v1/"
30
+ )
31
 
32
  # —— Baidu OCR —— #
33
  def get_access_token(api_key, secret_key):