@purpleschool/ai-proxy 0.0.2 → 0.1.4

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/{errors → constants/errors}/errors.js

@@ -129,4 +129,99 @@ 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,
+    },
 };

package/build/constants/index.js

@@ -0,0 +1,18 @@
+"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("./errors"), exports);
+__exportStar(require("./enums"), exports);

package/build/index.js

@@ -14,4 +14,4 @@ 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("./errors"), exports);
+__exportStar(require("./constants"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@purpleschool/ai-proxy",
-  "version": "0.0.2",
+  "version": "0.1.4",
   "description": "",
   "main": "build/index.js",
   "types": "build/index.d.ts",
@@ -17,5 +17,8 @@
   "license": "ISC",
   "dependencies": {
     "zod": "^3.22.4"
+  },
+  "devDependencies": {
+    "@types/node": "^24.3.0"
   }
 }

package/proto/ai-proxy.proto

@@ -0,0 +1,250 @@
+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);
+}
+
+// Text Generation (stream)
+message TextGenerationRequest {
+  string model = 1;
+  repeated Message messages = 2;
+  repeated string features = 3;
+  WebSearchParams webSearchParams = 4;
+}
+
+message Message {
+  optional string uuid = 1;
+  string role = 2;
+  string text = 3;
+  repeated File files = 4;
+  optional string status = 5;
+  optional string externalId = 6;
+  google.protobuf.Timestamp createdAt = 7;
+  google.protobuf.Timestamp updatedAt = 8;
+}
+
+message WebSearchParams {
+  UserLocation userLocation = 1;
+}
+
+message UserLocation {
+  string country = 1;
+  string city = 2;
+  string region = 3;
+}
+
+message File {
+  string url = 1;
+  string type = 2;
+  string mimeType = 3;
+}
+
+message TextGenerationResponse {
+  optional string delta = 1;
+  optional int32 inputTokens = 2;
+  optional int32 outputTokens = 3;
+  bool isFinish = 4;
+}
+
+// TextGenerationUnary 
+message TextGenerationUnaryRequest {
+  string model = 1;
+  repeated Message messages = 2;
+  repeated string features = 3;
+  WebSearchParams webSearchParams = 4;
+  string responseSchema = 5;
+}
+
+message TextGenerationUnaryResponse {
+  string text = 1;
+  int32 inputTokens = 2;
+  int32 outputTokens = 3;
+}
+
+// Image Generation
+message ImageGenerationRequest {
+  string model = 1;
+  string prompt = 2;
+  int32 width = 3;
+  int32 height = 4;
+  repeated Message messages = 5;
+}
+
+message ImageGenerationResponse { 
+  string model = 1;
+  ImageGenerationData data = 2;
+}
+
+message ImageGenerationData {
+  optional string url = 1;
+  optional string base64 = 2;
+  optional string externalId = 3;
+  string type = 4;
+}
+
+// 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
+message GetAllAiModelsRequest {
+}
+
+message GetAllAiModelsResponse {
+  repeated AiModel data = 1;
+}
+
+message AiModel {
+  string model = 1;
+  string contentType = 2;
+  repeated string features = 3;
+}
+
+// GetAiModel
+message GetAiModelRequest {
+  string model = 1;
+}
+
+message GetAiModelResponse {
+  AiModel data = 1;
+}
+
+// GenerateEmbeddings
+message EmbeddingsGenerationRequest {
+  repeated string queries = 1;
+}
+
+message Embedding {
+  repeated double embedding = 1;
+}
+
+message EmbeddingsGenerationResponse {
+  repeated Embedding data = 1;
+}

package/proto/ai-api.proto

@@ -1,92 +0,0 @@
-syntax = "proto3";
-
-package AiApi;
-
-service AiApiService {
-  rpc ChatCompletions (ChatCompletionsRequest) returns (stream ChatCompletionsResponse);
-  rpc ImageGeneration (ImageGenerationRequest) returns (ImageGenerationResponse);
-  rpc GetAllAiModels(GetAllAiModelsRequest) returns (GetAllAiModelsResponse);
-  rpc GetAiModel(GetAiModelRequest) returns (GetAiModelResponse);
-}
-
-// Chat Completions
-message ChatCompletionsRequest {
-  string model = 1;
-  repeated Message messages = 2;
-  repeated string features = 3;
-  UserLocation userLocation = 4;
-}
-
-message Message {
-  string role = 1;
-  string text = 2;
-  repeated File files = 3;
-}
-
-message UserLocation {
-  string country = 1;
-  string city = 2;
-  string region = 3;
-}
-
-message File {
-  string url = 1;
-  string type = 2;
-}
-
-message ChatCompletionsResponse {
-  optional string delta = 1;
-  bool isFinish = 2;
-}
-
-// Image Generation
-message ImageGenerationRequest {
-  string model = 1;
-  string prompt = 2;
-  int32 width = 3;
-  int32 height = 4;
-}
-
-message ImageGenerationResponse { 
-  string model = 1;
-  ImageGenerationData data = 2;
-}
-
-message ImageGenerationData {
-  string url = 1;
-}
-
-// GetAllAiModels
-message GetAllAiModelsRequest {
-}
-
-message GetAllAiModelsResponse {
-  repeated AiModel data = 1;
-}
-
-message AiModel {
-  string model = 1;
-  string contentType = 2;
-  repeated string features = 3;
-  int32 maxInputLength = 4;
-  int32 maxInputTokens = 5;
-  AiModelConfig config = 6;
-}
-
-message AiModelConfig {
-  repeated ImageSizeOption supportedResolutions = 1;
-}
-
-message ImageSizeOption {
-  int32 width = 1;
-  int32 height = 2;
-}
-
-// GetAiModel
-message GetAiModelRequest {
-  string model = 1;
-}
-
-message GetAiModelResponse {
-  AiModel data = 1;
-}