@purpleschool/ai-proxy 0.1.17 → 0.2.18

package/build/constants/errors/errors.js

@@ -234,4 +234,19 @@ exports.ERRORS = {
         message: 'Задание на генерацию изображения не найдено',
         httpCode: 404,
     },
+    IMAGE_EDITOR_JOB_SAVE_ERROR: {
+        code: 'A047',
+        message: 'Произошла ошибка при сохранении задания на редактирование изображения',
+        httpCode: 500,
+    },
+    IMAGE_EDITOR_JOB_NOT_FOUND: {
+        code: 'A048',
+        message: 'Задание на редактирование изображения не найдено',
+        httpCode: 404,
+    },
+    IMAGE_EDITOR_JOB_FIND_ERROR: {
+        code: 'A049',
+        message: 'Ошибка при нахождении задание на редактирование изображения',
+        httpCode: 500,
+    },
 };

package/build/image-generation/enums/image-generation-response-type.enum.js

@@ -6,4 +6,5 @@ var IMAGE_GENERATION_RESPONSE_TYPE;
     IMAGE_GENERATION_RESPONSE_TYPE["URL"] = "url";
     IMAGE_GENERATION_RESPONSE_TYPE["BASE64"] = "base64";
     IMAGE_GENERATION_RESPONSE_TYPE["JOB"] = "job";
+    IMAGE_GENERATION_RESPONSE_TYPE["ERROR"] = "error";
 })(IMAGE_GENERATION_RESPONSE_TYPE || (exports.IMAGE_GENERATION_RESPONSE_TYPE = IMAGE_GENERATION_RESPONSE_TYPE = {}));

package/constants/errors/errors.ts

@@ -235,4 +235,19 @@ export const ERRORS = {
         message: 'Задание на генерацию изображения не найдено',
         httpCode: 404,
     },
+    IMAGE_EDITOR_JOB_SAVE_ERROR: {
+        code: 'A047',
+        message: 'Произошла ошибка при сохранении задания на редактирование изображения',
+        httpCode: 500,
+    },
+    IMAGE_EDITOR_JOB_NOT_FOUND: {
+        code: 'A048',
+        message: 'Задание на редактирование изображения не найдено',
+        httpCode: 404,
+    },
+    IMAGE_EDITOR_JOB_FIND_ERROR: {
+        code: 'A049',
+        message: 'Ошибка при нахождении задание на редактирование изображения',
+        httpCode: 500,
+    },
 };

package/image-generation/enums/image-generation-response-type.enum.ts

@@ -2,4 +2,5 @@ export enum IMAGE_GENERATION_RESPONSE_TYPE {
     URL = 'url',
     BASE64 = 'base64',
     JOB = 'job',
+    ERROR = 'error',
 }

package/package.json

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

package/proto/ai-proxy.proto

@@ -1,5 +1,4 @@
 syntax = "proto3";
-import "google/protobuf/timestamp.proto";
 import "google/protobuf/struct.proto";
 package AiProxy;
 
@@ -8,6 +7,7 @@ service AiProxyService {
   rpc TextGenerationUnary (TextGenerationUnaryRequest) returns (TextGenerationUnaryResponse);
 
   rpc ImageGeneration (ImageGenerationRequest) returns (ImageGenerationResponse);
+  rpc ImageEditor (ImageEditorRequest) returns (ImageEditorResponse);
 
   rpc VideoGeneration (VideoGenerationRequest) returns (VideoGenerationResponse);
   rpc VideoEditor (VideoEditorRequest) returns (VideoEditorResponse);
@@ -34,14 +34,9 @@ message TextGenerationRequest {
 }
 
 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;
+  string role = 1;
+  string content = 2;
+  repeated File files = 3;
 }
 
 message WebSearchParams {
@@ -56,8 +51,7 @@ message UserLocation {
 
 message File {
   string url = 1;
-  string type = 2;
-  string mimeType = 3;
+  string mimeType = 2;
 }
 
 message TextGenerationResponse {
@@ -87,23 +81,37 @@ message TextGenerationUnaryResponse {
 message ImageGenerationRequest {
   string model = 1;
   string prompt = 2;
-  optional int32 width = 3;
-  optional int32 height = 4;
-  repeated Message messages = 5;
+  string aspectRatio = 3;
+  optional string resolution = 4;
+  repeated string attachedImages = 5;
   optional string callbackUrl = 6;
 }
 
 message ImageGenerationResponse { 
   string model = 1;
-  ImageGenerationData data = 2;
+  ImageGenerationData response = 2;
 }
 
 message ImageGenerationData {
-  optional string data = 1;
-  optional string externalId = 2;
+  repeated string data = 1;
+  string externalId = 2;
   string type = 3;
+  optional string errorMsg = 4;
 }
 
+// Image Editor
+message ImageEditorRequest {
+  string model = 1;
+  string prompt = 2;
+  repeated string imageUrls = 3;
+  string callbackUrl = 4;
+}
+
+message ImageEditorResponse {
+  string jobId = 1;
+  optional string imageUrl = 2;
+  optional string error = 3;
+}
 // Video Generation
 message VideoGenerationRequest {
   string model = 1;