syntax = "proto3";

package grpc_sam3_img;

service TaskService {
    rpc ProcessTask (TaskRequest) returns (TaskResponse);
}

// 添加健康检查服务
service HealthCheck {
    rpc Check (HealthCheckRequest) returns (HealthCheckResponse);
}

message HealthCheckRequest {
    string service = 1;
}

message HealthCheckResponse {
    enum ServingStatus {
        UNKNOWN = 0;
        SERVING = 1;
        NOT_SERVING = 2;
        SERVICE_UNKNOWN = 3;
    }
    ServingStatus status = 1;
}

message TaskRequest {
    string task_id = 1;
    string sn = 2;
    ContentBody content_body = 3;
}

message ContentBody {
    string img_url = 1;
    string prompt = 2;
    float confidence =3;
    string mqtt_ip = 4;
    int32 mqtt_port = 5;
    string mqtt_topic = 6;

}

message TaskResponse {
    string task_id = 1;
    bool success = 2;
    string message = 3;
}