Merge pull request #2 from juzibot/feat/videosupport

Feat/videosupport

Author: windmemory

src/google/google-proxy.controller.ts

@@ -66,7 +66,17 @@ export class GoogleProxyController {
       const queryString = new URLSearchParams(query).toString();
       const uploadUrl = `https://generativelanguage.googleapis.com/upload/v1beta/files?${queryString}`;
       const bufferBody = req.body;
-      return this.service.uploadFileData(uploadUrl, bufferBody, headers);
+      const result = await this.service.uploadFileData(uploadUrl, bufferBody, headers);
+            if (result && typeof result === 'object' && 'status' in result) {
+        res.status(result.status);
+        if (result.headers) {
+          Object.keys(result.headers).forEach(key => {
+            res.setHeader(key, result.headers[key]);
+          });
+        }
+        return result.data;
+      }
+      return result;
     }
     const result = await this.service.uploadFileInit(body, headers);
     res.status(result.status);
@@ -110,8 +120,9 @@ export class GoogleProxyController {
   async getFile(
     @Param('path') path: string,
     @Headers() headers: any,
+    @Query() query: any,
   ) {
     const url = `https://generativelanguage.googleapis.com/v1beta/${path}`;
-    return this.service.getFileInfo(url, headers);
+    return this.service.getFileInfo(url, headers, query);
   }
 }

src/google/google-proxy.service.ts

@@ -55,9 +55,14 @@ export class GoogleProxyService {
       timeout: 5 * MINUTE,
       maxContentLength: Infinity,
       maxBodyLength: Infinity,
-      validateStatus: (status) => status === 200 || status === 308,
+      validateStatus: (status) => status === 200 || status === 201 || status === 308,
       isBinaryData: true,
+      returnFullResponse: true,
     });
+
+    if (result.status === 201) {
+      return result;
+    }
     return result.data;
   }
 
@@ -82,14 +87,18 @@ export class GoogleProxyService {
       timeout: 5 * MINUTE,
       maxContentLength: Infinity,
       maxBodyLength: Infinity,
-      validateStatus: (status) => status === 200 || status === 308,
+      validateStatus: (status) => status === 200 || status === 201 || status === 308,
       isBinaryData: true,
+      returnFullResponse: true,
     });
+    if (result.status === 201) {
+      return result;
+    }
     return result.data;
   }
 
-  async getFileInfo(url: string, headers: any) {
-    return this.makeRequest(url, headers, null, {}, false, {
+  async getFileInfo(url: string, headers: any, query?: any) {
+    return this.makeRequest(url, headers, null, query || {}, false, {
       method: 'GET',
       timeout: 30000,
     });