76 lines
2.7 KiB
Protocol Buffer
76 lines
2.7 KiB
Protocol Buffer
syntax = "proto3";
|
|
option go_package = "ai-chat-service/proto";
|
|
package ai_chat_service.zvoice.com;
|
|
|
|
message ChatCompletionRequest {
|
|
string message = 1[json_name = "message"];
|
|
string id = 2[json_name = "id"];
|
|
string pid = 3[json_name = "p_id"];
|
|
bool enableContext = 4[json_name = "enable_context"];
|
|
ChatParam chatParam = 5[json_name = "chat_param"];
|
|
}
|
|
|
|
message ChatParam {
|
|
string model = 1[json_name = "model"];
|
|
int32 maxTokens = 2[json_name = "max_tokens"];
|
|
float temperature = 3[json_name = "temperature"];
|
|
float topP = 4[json_name = "top_p"];
|
|
float presencePenalty = 5[json_name = "presence_penalty"];
|
|
float frequencyPenalty = 6[json_name = "frequency_penalty"];
|
|
string botDesc =7[json_name = "bot_desc"];
|
|
int32 minResponseTokens = 8[json_name = "min_response_tokens"];
|
|
int32 contextTTL = 9[json_name = "context_ttl"];
|
|
int32 contextLen = 10[json_name = "context_len"];
|
|
}
|
|
|
|
// 服务响应消息，非流式响应
|
|
message ChatCompletionResponse {
|
|
string id = 1 [json_name = "id"];
|
|
string object = 2 [json_name = "object"];
|
|
int64 created = 3 [json_name = "created"];
|
|
string model = 4 [json_name = "model"];
|
|
repeated ChatCompletionChoice choices = 5 [json_name = "choices"];
|
|
Usage usage = 6[json_name = "usage"];
|
|
string source = 7 [json_name = "source"];
|
|
}
|
|
message ChatCompletionChoice {
|
|
int32 index = 1[json_name = "index"];
|
|
ChatCompletionMessage message = 2 [json_name = "message"];
|
|
string finishReason = 3[json_name = "finish_reason"];
|
|
}
|
|
message ChatCompletionMessage {
|
|
string role=1[json_name = "role"];
|
|
string content=2[json_name = "content"];
|
|
string name=3[json_name = "name"];
|
|
}
|
|
message Usage {
|
|
int32 promptTokens = 1 [json_name = "prompt_tokens"];
|
|
int32 completionTokens =2 [json_name = "completion_tokens"];
|
|
int32 totalTokens = 3 [json_name = "total_tokens"];
|
|
}
|
|
|
|
// 服务响应消息，流式响应
|
|
message ChatCompletionStreamResponse {
|
|
string id = 1 [json_name = "id"];
|
|
string object = 2 [json_name = "object"];
|
|
int64 created = 3 [json_name = "created"];
|
|
string model = 4 [json_name = "model"];
|
|
repeated ChatCompletionStreamChoice choices = 5 [json_name = "choices"];
|
|
Usage usage = 6[json_name = "usage"];
|
|
string source = 7 [json_name = "source"];
|
|
}
|
|
message ChatCompletionStreamChoice {
|
|
int32 index = 1[json_name = "index"];
|
|
ChatCompletionStreamChoiceDelta delta =2 [json_name = "delta"];
|
|
string finishReason = 3[json_name="finish_reason"];
|
|
}
|
|
message ChatCompletionStreamChoiceDelta {
|
|
string content = 1 [json_name = "content"];
|
|
string role = 2 [json_name = "role"];
|
|
}
|
|
|
|
service Chat {
|
|
rpc ChatCompletion(ChatCompletionRequest) returns (ChatCompletionResponse);
|
|
rpc ChatCompletionStream(ChatCompletionRequest) returns (stream ChatCompletionStreamResponse);
|
|
}
|