Skip to content

Commit 7792dd9

Browse files
committed
fixup! ✨(back) manage streaming with the ai service
1 parent 5e04984 commit 7792dd9

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

src/backend/core/api/viewsets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1369,7 +1369,7 @@ def ai_proxy(self, request, *args, **kwargs):
13691369
content_type="text/event-stream",
13701370
status=drf.status.HTTP_200_OK,
13711371
)
1372-
1372+
13731373
ai_response = ai_service.proxy(request.data)
13741374
return drf.response.Response(
13751375
ai_response.model_dump(),

src/backend/core/services/ai_services.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from django.conf import settings
77
from django.core.exceptions import ImproperlyConfigured
88

9-
from openai import OpenAI
9+
from openai import OpenAI, OpenAIError
1010

1111
log = logging.getLogger(__name__)
1212

@@ -27,7 +27,10 @@ def __init__(self):
2727
def proxy(self, data: dict, stream: bool = False) -> Generator[str, None, None]:
2828
"""Proxy AI API requests to the configured AI provider."""
2929
data["stream"] = stream
30-
return self.client.chat.completions.create(**data)
30+
try:
31+
return self.client.chat.completions.create(**data)
32+
except OpenAIError as e:
33+
raise RuntimeError(f"Failed to proxy AI request: {e}") from e
3134

3235
def stream(self, data: dict) -> Generator[str, None, None]:
3336
"""Stream AI API requests to the configured AI provider."""

0 commit comments

Comments
 (0)