Spaces:
Sleeping
Sleeping
Commit
·
49982b5
1
Parent(s):
ad78edb
feat: add chat completions test endpoint to validate OpenAI integration and improve error handling
Browse files- backend/app/main.py +40 -4
backend/app/main.py
CHANGED
|
@@ -77,18 +77,18 @@ async def test_openai():
|
|
| 77 |
"""Test OpenAI connection without making actual API calls"""
|
| 78 |
try:
|
| 79 |
from .core.config import settings
|
| 80 |
-
|
| 81 |
if not settings.OPENAI_API_KEY:
|
| 82 |
return {"status": "error", "message": "OPENAI_API_KEY not configured"}
|
| 83 |
-
|
| 84 |
api_key_preview = (
|
| 85 |
settings.OPENAI_API_KEY[:10] + "..."
|
| 86 |
if len(settings.OPENAI_API_KEY) > 10
|
| 87 |
else "Too short"
|
| 88 |
)
|
| 89 |
-
|
| 90 |
return {
|
| 91 |
-
"status": "ok",
|
| 92 |
"message": "OpenAI API key is configured",
|
| 93 |
"api_key_preview": api_key_preview,
|
| 94 |
"environment": (
|
|
@@ -98,6 +98,42 @@ async def test_openai():
|
|
| 98 |
except Exception as e:
|
| 99 |
return {"status": "error", "message": f"OpenAI configuration error: {str(e)}"}
|
| 100 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 101 |
|
| 102 |
# Include API router
|
| 103 |
app.include_router(api_router, prefix=settings.API_V1_STR)
|
|
|
|
| 77 |
"""Test OpenAI connection without making actual API calls"""
|
| 78 |
try:
|
| 79 |
from .core.config import settings
|
| 80 |
+
|
| 81 |
if not settings.OPENAI_API_KEY:
|
| 82 |
return {"status": "error", "message": "OPENAI_API_KEY not configured"}
|
| 83 |
+
|
| 84 |
api_key_preview = (
|
| 85 |
settings.OPENAI_API_KEY[:10] + "..."
|
| 86 |
if len(settings.OPENAI_API_KEY) > 10
|
| 87 |
else "Too short"
|
| 88 |
)
|
| 89 |
+
|
| 90 |
return {
|
| 91 |
+
"status": "ok",
|
| 92 |
"message": "OpenAI API key is configured",
|
| 93 |
"api_key_preview": api_key_preview,
|
| 94 |
"environment": (
|
|
|
|
| 98 |
except Exception as e:
|
| 99 |
return {"status": "error", "message": f"OpenAI configuration error: {str(e)}"}
|
| 100 |
|
| 101 |
+
# Test chat completions endpoint
|
| 102 |
+
@app.get("/api/test-chat")
|
| 103 |
+
async def test_chat():
|
| 104 |
+
"""Test if OpenAI chat completions work on Hugging Face"""
|
| 105 |
+
try:
|
| 106 |
+
from .core.config import settings
|
| 107 |
+
from openai import OpenAI
|
| 108 |
+
|
| 109 |
+
if not settings.OPENAI_API_KEY:
|
| 110 |
+
return {"status": "error", "message": "OPENAI_API_KEY not configured"}
|
| 111 |
+
|
| 112 |
+
client = OpenAI(api_key=settings.OPENAI_API_KEY, timeout=30.0)
|
| 113 |
+
|
| 114 |
+
# Test a simple chat completion
|
| 115 |
+
response = client.chat.completions.create(
|
| 116 |
+
model="gpt-3.5-turbo",
|
| 117 |
+
messages=[
|
| 118 |
+
{"role": "user", "content": "Say 'Hello from Hugging Face!' in exactly those words."}
|
| 119 |
+
],
|
| 120 |
+
max_tokens=20
|
| 121 |
+
)
|
| 122 |
+
|
| 123 |
+
return {
|
| 124 |
+
"status": "success",
|
| 125 |
+
"message": "Chat completions work!",
|
| 126 |
+
"response": response.choices[0].message.content,
|
| 127 |
+
"model": response.model
|
| 128 |
+
}
|
| 129 |
+
|
| 130 |
+
except Exception as e:
|
| 131 |
+
return {
|
| 132 |
+
"status": "error",
|
| 133 |
+
"message": f"Chat completion failed: {str(e)}",
|
| 134 |
+
"error_type": type(e).__name__
|
| 135 |
+
}
|
| 136 |
+
|
| 137 |
|
| 138 |
# Include API router
|
| 139 |
app.include_router(api_router, prefix=settings.API_V1_STR)
|