Mahmous commited on
Commit
c9a8d40
·
verified ·
1 Parent(s): e6d4b73

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +20 -7
api.py CHANGED
@@ -61,15 +61,28 @@ app = Flask(__name__)
61
  CORS(app, resources={r"/ask": {"origins": "*"}})
62
 
63
  # ---------- OpenAI Client ----------
 
 
64
  client = None
65
- if OPENAI_API_KEY:
 
 
 
 
66
  try:
67
- client = OpenAI(api_key=OPENAI_API_KEY)
68
- print("✅ OpenAI client initialized successfully.")
69
- except Exception as e:
70
- print("❌ Failed to initialize OpenAI client:", e)
71
- else:
72
- print("⚠️ OPENAI_API_KEY is missing in .env")
 
 
 
 
 
 
 
73
 
74
  # ---------- Retriever ----------
75
  retriever = None
 
61
  CORS(app, resources={r"/ask": {"origins": "*"}})
62
 
63
  # ---------- OpenAI Client ----------
64
+ import openai
65
+
66
  client = None
67
+ try:
68
+ if not OPENAI_API_KEY or OPENAI_API_KEY.strip() == "":
69
+ raise ValueError("Missing OPENAI_API_KEY in environment variables")
70
+
71
+ # Handle both old and new SDK structures safely
72
  try:
73
+ # New SDK style
74
+ client = openai.OpenAI(api_key=OPENAI_API_KEY)
75
+ print("✅ Using new OpenAI client interface.")
76
+ except Exception:
77
+ # Fallback to legacy interface
78
+ openai.api_key = OPENAI_API_KEY
79
+ client = openai
80
+ print("✅ Using legacy OpenAI interface.")
81
+
82
+ except Exception as e:
83
+ client = None
84
+ print(f"❌ Failed to initialize OpenAI client: {e}")
85
+
86
 
87
  # ---------- Retriever ----------
88
  retriever = None