proto

ai.proto

@@ -0,0 +1,36 @@
+syntax = "proto3";
+
+package ai;
+
+option go_package = "ai_service/proto/ai";
+
+// AIRequest содержит запрос к AI модели
+message AIRequest {
+  string text = 1;           // Текст запроса
+  string bot_username = 2;   // Username бота для упоминаний
+}
+
+// AIResponse содержит ответ от AI модели
+message AIResponse {
+  string content = 1;        // Текст ответа
+  string error = 2;          // Текст ошибки, если произошла
+  double duration_ms = 3;    // Время выполнения в миллисекундах
+}
+
+// HealthCheckRequest для проверки здоровья сервиса
+message HealthCheckRequest {}
+
+// HealthCheckResponse для ответа на проверку здоровья
+message HealthCheckResponse {
+  bool serving = 1;
+  string version = 2;
+}
+
+// AIService - основной сервис для работы с AI
+service AIService {
+  // GenerateAnswer генерирует ответ на запрос
+  rpc GenerateAnswer(AIRequest) returns (AIResponse);
+  
+  // HealthCheck проверяет здоровье сервиса
+  rpc HealthCheck(HealthCheckRequest) returns (HealthCheckResponse);
+}