feat: Improve OpenAI API integration and response handling (#5)

Co-authored-by: Hai Chang <haichang@microsoft.com>

Author: haha1903

src/app.controller.ts

@@ -37,6 +37,12 @@ export class ChatController {
       stream,
     );
 
+    if (response.status > 299 || response.status < 200) {
+      res.status(response.status);
+      res.send(response.data);
+      return;
+    }
+
     // set response headers
     for (const [key, value] of response.headers as AxiosHeaders) {
       res.header[key] = value;

src/app.service.ts

@@ -36,7 +36,11 @@ export class AppService {
       config['responseType'] = 'stream';
     }
     const ret = this.httpService.post(url, body, config);
-    return await firstValueFrom(ret);
+    try {
+      return await firstValueFrom(ret);
+    } catch (e) {
+      return e.response;
+    }
   }
   private getDeploymentId(mapping: string, model: string): string {
     this.logger.debug(`mapping: ${mapping}, model: ${model}`);