@purpleschool/ai-proxy 0.0.2 → 0.0.4

package/build/constants/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("./errors"), 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.0.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,126 @@
+syntax = "proto3";
+import "google/protobuf/timestamp.proto";
+package AiProxy;
+
+service AiProxyService {
+  rpc TextGeneration (TextGenerationRequest) returns (stream TextGenerationResponse);
+  rpc TextGenerationUnary (TextGenerationUnaryRequest) returns (TextGenerationUnaryResponse);
+  rpc ImageGeneration (ImageGenerationRequest) returns (ImageGenerationResponse);
+  rpc EmbeddingsGeneration (EmbeddingsGenerationRequest) returns (EmbeddingsGenerationResponse);
+  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;
+}
+
+// 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;
-}