94 lines
2.0 KiB
Protocol Buffer
94 lines
2.0 KiB
Protocol Buffer
syntax = "proto3";
|
|
|
|
package topfans.ai_chat;
|
|
|
|
option go_package = "github.com/topfans/backend/pkg/proto/ai_chat;ai_chat";
|
|
|
|
import "google/api/annotations.proto";
|
|
|
|
// ==================== 对话消息 ====================
|
|
|
|
message Message {
|
|
string role = 1; // "user" / "assistant"
|
|
string content = 2;
|
|
}
|
|
|
|
// ==================== 对话相关 ====================
|
|
|
|
message InitSessionRequest {
|
|
string session_id = 1;
|
|
int64 user_id = 2;
|
|
}
|
|
|
|
message InitSessionResponse {
|
|
string welcome_message = 1;
|
|
string session_id = 2;
|
|
}
|
|
|
|
message ChatMessageRequest {
|
|
string session_id = 1;
|
|
string message = 2;
|
|
string persona_id = 3; // 可选，空则用默认人设
|
|
int64 user_id = 4;
|
|
}
|
|
|
|
message ChatMessageResponse {
|
|
string content = 1;
|
|
string session_id = 2;
|
|
bool is_end = 3;
|
|
string error = 4;
|
|
}
|
|
|
|
message ChatHistoryRequest {
|
|
string session_id = 1;
|
|
int32 limit = 2; // 默认 20
|
|
}
|
|
|
|
message ChatHistoryResponse {
|
|
repeated Message history = 1;
|
|
}
|
|
|
|
// ==================== 人设管理 ====================
|
|
|
|
message GetPersonasRequest {
|
|
int64 user_id = 1;
|
|
}
|
|
|
|
message PersonaListResponse {
|
|
repeated PersonaInfo personas = 1;
|
|
}
|
|
|
|
message PersonaInfo {
|
|
string id = 1;
|
|
string name = 2;
|
|
string description = 3;
|
|
string avatar_url = 4;
|
|
string talk_style = 5;
|
|
bool is_default = 6;
|
|
int64 created_at = 7;
|
|
int64 updated_at = 8;
|
|
}
|
|
|
|
// ==================== AI Chat Service ====================
|
|
|
|
service AIChatService {
|
|
// 初始化会话，获取欢迎消息
|
|
rpc InitSession(InitSessionRequest) returns (InitSessionResponse);
|
|
|
|
// 发送消息，流式返回
|
|
rpc SendMessage(ChatMessageRequest) returns (stream ChatMessageResponse);
|
|
|
|
// 获取对话历史
|
|
rpc GetHistory(ChatHistoryRequest) returns (ChatHistoryResponse) {
|
|
option (google.api.http) = {
|
|
get: "/api/v1/ai-chat/history/{session_id}"
|
|
};
|
|
}
|
|
|
|
// 获取用户的所有人设
|
|
rpc GetPersonas(GetPersonasRequest) returns (PersonaListResponse) {
|
|
option (google.api.http) = {
|
|
get: "/api/v1/ai-chat/personas"
|
|
};
|
|
}
|
|
} |