@purpleschool/ai-proxy 0.0.4 → 0.1.5

package/build/constants/enums/index.js

@@ -0,0 +1,17 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./job-status.enum"), exports);

package/build/constants/enums/job-status.enum.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.JOB_STATUS = void 0;
+var JOB_STATUS;
+(function (JOB_STATUS) {
+    JOB_STATUS["PENDING"] = "pending";
+    JOB_STATUS["PROCESSING"] = "processing";
+    JOB_STATUS["COMPLETED"] = "completed";
+    JOB_STATUS["FAILED"] = "failed";
+})(JOB_STATUS || (exports.JOB_STATUS = JOB_STATUS = {}));

package/build/constants/errors/errors.js

@@ -129,4 +129,109 @@ exports.ERRORS = {
         message: 'Вложения файлов и веб-поиск нельзя использовать вместе',
         httpCode: 400,
     },
+    VIDEO_MISSING_PARAMS_ERROR: {
+        code: 'A026',
+        message: 'Не переданы параметры при запросе генерации видео',
+        httpCode: 400,
+    },
+    VIDEO_JOB_SAVE_ERROR: {
+        code: 'A027',
+        message: 'Произошла ошибка при сохранении задания на генерацию видео',
+        httpCode: 500,
+    },
+    VIDEO_JOB_NOT_FOUND: {
+        code: 'A028',
+        message: 'Задание на генерацию видео не найдено',
+        httpCode: 404,
+    },
+    EXTERNAL_PROVIDER_CONTENT_POLICY: {
+        code: 'A029',
+        message: 'Запрос был отклонен, т.к. он не соответствует требованиям политики безопасности поставщика модели',
+        httpCode: 400,
+    },
+    GOOGLE_CONTENT_POLICY: {
+        code: 'A030',
+        message: 'Запрос был отклонен, т.к. он не соответствует требованиям политики безопасности Google.',
+        httpCode: 400,
+    },
+    VIDEO_EDITOR_SAVE_ERROR: {
+        code: 'A031',
+        message: 'Произошла ошибка при сохранении задания на редактирование видео',
+        httpCode: 500,
+    },
+    VIDEO_EDITOR_FIND_ERROR: {
+        code: 'A032',
+        message: 'Произошла ошибка при поиске задания на редактирование видео',
+        httpCode: 500,
+    },
+    STT_JOB_SAVE_ERROR: {
+        code: 'A033',
+        message: 'Произошла ошибка при сохранении задания STT',
+        httpCode: 500,
+    },
+    STT_JOB_FIND_ERROR: {
+        code: 'A034',
+        message: 'Произошла ошибка при поиске задания STT',
+        httpCode: 500,
+    },
+    TTS_JOB_SAVE_ERROR: {
+        code: 'A035',
+        message: 'Произошла ошибка при сохранении задания TTS',
+        httpCode: 500,
+    },
+    TTS_JOB_FIND_ERROR: {
+        code: 'A036',
+        message: 'Произошла ошибка при поиске задания TTS',
+        httpCode: 500,
+    },
+    MUSIC_JOB_SAVE_ERROR: {
+        code: 'A037',
+        message: 'Произошла ошибка при сохранении задания на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_JOB_FIND_ERROR: {
+        code: 'A038',
+        message: 'Произошла ошибка при поиске задания на генерацию музыки',
+        httpCode: 500,
+    },
+    MUSIC_JOB_NOT_FOUND: {
+        code: 'A039',
+        message: 'Задание на генерацию музыки не найдено',
+        httpCode: 404,
+    },
+    SUNO_CONTENT_POLICY: {
+        code: 'A040',
+        message: 'Запрос на создание музыки был отклонен, т.к. он не соответствует требованиям политики безопасности Suno.',
+        httpCode: 400,
+    },
+    GENERATE_LYRICS_ERROR: {
+        code: 'A041',
+        message: 'Произошла ошибка при генерации текста музыки',
+        httpCode: 500,
+    },
+    CONVERT_TO_WAV_ERROR: {
+        code: 'A042',
+        message: 'Произошла ошибка при конвертации mp3 в wav',
+        httpCode: 500,
+    },
+    POLL_TIMEOUT: {
+        code: 'A043',
+        message: 'Превышено время ожидания генерации музыки',
+        httpCode: 500,
+    },
+    MUSIC_JOB_NOT_AN_OWNER: {
+        code: 'A044',
+        message: 'Пользователь не является владельцем задания на генерацию музыки',
+        httpCode: 403,
+    },
+    IMAGE_JOB_SAVE_ERROR: {
+        code: 'A045',
+        message: 'Произошла ошибка при сохранении задания на генерацию изображения',
+        httpCode: 500,
+    },
+    IMAGE_JOB_NOT_FOUND: {
+        code: 'A046',
+        message: 'Задание на генерацию изображения не найдено',
+        httpCode: 404,
+    },
 };

package/build/constants/index.js

@@ -15,3 +15,4 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./errors"), exports);
+__exportStar(require("./enums"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@purpleschool/ai-proxy",
-  "version": "0.0.4",
+  "version": "0.1.5",
   "description": "",
   "main": "build/index.js",
   "types": "build/index.d.ts",

package/proto/ai-proxy.proto

@@ -1,12 +1,26 @@
 syntax = "proto3";
 import "google/protobuf/timestamp.proto";
+import "google/protobuf/struct.proto";
 package AiProxy;
 
 service AiProxyService {
   rpc TextGeneration (TextGenerationRequest) returns (stream TextGenerationResponse);
   rpc TextGenerationUnary (TextGenerationUnaryRequest) returns (TextGenerationUnaryResponse);
+
   rpc ImageGeneration (ImageGenerationRequest) returns (ImageGenerationResponse);
+
+  rpc VideoGeneration (VideoGenerationRequest) returns (VideoGenerationResponse);
+  rpc VideoEditor (VideoEditorRequest) returns (VideoEditorResponse);
+
+  rpc SpeechToText (SpeechToTextRequest) returns (SpeechToTextResponse);
+  rpc TextToSpeech (TextToSpeechRequest) returns (stream TextToSpeechResponse);
+
   rpc EmbeddingsGeneration (EmbeddingsGenerationRequest) returns (EmbeddingsGenerationResponse);
+
+  rpc MusicGeneration (MusicGenerationRequest) returns (MusicGenerationResponse);
+  rpc GenerateLyrics (GenerateLyricsRequest) returns (GenerateLyricsResponse);
+  rpc ConvertToWav (ConvertToWavRequest) returns (ConvertToWavResponse);
+
   rpc GetAllAiModels(GetAllAiModelsRequest) returns (GetAllAiModelsResponse);
   rpc GetAiModel(GetAiModelRequest) returns (GetAiModelResponse);
 }
@@ -75,6 +89,7 @@ message ImageGenerationRequest {
   int32 width = 3;
   int32 height = 4;
   repeated Message messages = 5;
+  optional string callbackUrl = 6;
 }
 
 message ImageGenerationResponse { 
@@ -83,10 +98,119 @@ message ImageGenerationResponse {
 }
 
 message ImageGenerationData {
-  optional string url = 1;
-  optional string base64 = 2;
-  optional string externalId = 3;
-  string type = 4;
+  optional string data = 1;
+  optional string externalId = 2;
+  string type = 3;
+}
+
+// Video Generation
+message VideoGenerationRequest {
+  string model = 1;
+  string prompt = 2;
+  string callbackUrl = 3;
+  VideoParams params = 4;
+}
+
+message VideoParams {
+  repeated string imageUrls = 1;
+  int32 duration = 2;
+  optional string aspectRatio = 3;
+  optional string quality = 4;
+}
+
+message VideoGenerationResponse { 
+  string jobId = 1;
+}
+
+// Video Editor
+message VideoEditorRequest {
+  string model = 1;
+  string prompt = 2;
+  string callbackUrl = 3;
+  string inputVideoUrl = 4;
+  int32 duration = 5;
+}
+
+message VideoEditorResponse { 
+  string jobId = 1;
+}
+
+// Speech To Text
+message SpeechToTextRequest {
+  string model = 1;
+  string fileId = 2;
+  string fileKey = 3;
+  string fileUrl = 4;
+  int32 durationInSeconds = 5;
+  string callbackUrl = 6;
+}
+
+message SpeechToTextResponse { 
+  string jobId = 1;
+  optional SpeechToTextAiResponse aiResponse = 2;
+}
+
+message SpeechToTextAiResponse {
+  google.protobuf.Struct raw = 1;
+  string md = 2;
+}
+
+// Text To Speech
+message TextToSpeechRequest {
+  string model = 1;
+  string voiceId = 2;
+  string input = 3;
+  TextToSpeechParams params = 4;
+}
+
+message TextToSpeechResponse {
+  bytes audioChunk = 1;
+}
+
+message TextToSpeechParams {
+  int32 speed = 1;
+  int32 similarity = 2;
+  int32 stability = 3;
+  int32 style = 4;
+}
+
+// Music Generation
+message MusicGenerationRequest {
+  string model = 1;
+  string callbackUrl = 2;
+  MusicGenerationParams params = 3;
+}
+
+message MusicGenerationResponse {
+  string jobId = 1;
+}
+
+message MusicGenerationParams {
+  string mode = 1;
+  string style = 2;
+  optional string prompt = 3;
+  optional string title = 4;
+  optional string vocalGender = 5;
+}
+
+// Generate Lyrics
+message GenerateLyricsRequest {
+  string prompt = 1;
+}
+
+message GenerateLyricsResponse {
+  string title = 1;
+  string lyrics = 2;
+}
+
+// Convert To Wav
+message ConvertToWavRequest {
+  string taskId = 1;
+  string audioId = 2;
+}
+
+message ConvertToWavResponse {
+  string url = 1;
 }
 
 // GetAllAiModels