Update api.py
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
try:
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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
|