memos/proto/api/v1/ai_service.proto

syntax = "proto3";

package memos.api.v1;

import "google/api/annotations.proto";
import "google/protobuf/empty.proto";

option go_package = "gen/api/v1";

// AIService provides AI-related functionality
service AIService {
  // Get AI provider settings
  rpc GetAISettings(GetAISettingsRequest) returns (AISettings) {
    option (google.api.http) = {
      get: "/api/v1/ai/settings"
    };
  }

  // Update AI provider settings
  rpc UpdateAISettings(UpdateAISettingsRequest) returns (AISettings) {
    option (google.api.http) = {
      patch: "/api/v1/ai/settings"
      body: "settings"
    };
  }

  // Test AI provider connection
  rpc TestAIProvider(TestAIProviderRequest) returns (TestAIProviderResponse) {
    option (google.api.http) = {
      post: "/api/v1/ai/test"
      body: "*"
    };
  }

  // Generate text completion
  rpc GenerateCompletion(GenerateCompletionRequest) returns (GenerateCompletionResponse) {
    option (google.api.http) = {
      post: "/api/v1/ai/completion"
      body: "*"
    };
  }
}

message AISettings {
  GroqSettings groq = 1;
  OllamaSettings ollama = 2;
}

message GroqSettings {
  string api_key = 1;
  string default_model = 2;
  bool enabled = 3;
}

message OllamaSettings {
  string host = 1;
  string default_model = 2;
  bool enabled = 3;
}

message GetAISettingsRequest {}

message UpdateAISettingsRequest {
  AISettings settings = 1;
}

message TestAIProviderRequest {
  ProviderType provider = 1;
}

message TestAIProviderResponse {
  bool success = 1;
  string message = 2;
}

message GenerateCompletionRequest {
  ProviderType provider = 1;
  string prompt = 2;
  string model = 3;
  double temperature = 4;
  int32 max_tokens = 5;
}

message GenerateCompletionResponse {
  string text = 1;
  string model_used = 2;
  int32 prompt_tokens = 3;
  int32 completion_tokens = 4;
  int32 total_tokens = 5;
}

enum ProviderType {
  PROVIDER_TYPE_UNSPECIFIED = 0;
  PROVIDER_TYPE_GROQ = 1;
  PROVIDER_TYPE_OLLAMA = 2;
}