7486d561ec
- Update the RunPod example to use a new API key and - reduce resource allocation for pods. - Added stop pod functionality to the RunPod client and example. - Updated the RunPod client to use new API endpoints. - Updated the base URL for the RunPod client. - Added authorization header to HTTP client. Co-authored-by: mariobassem12 <mariobassem12@gmail.com>
92 lines
4.0 KiB
V
92 lines
4.0 KiB
V
module runpod
|
|
|
|
// #### Internally method doing a network call to create a new on-demand pod.
|
|
// - Build the required query based pn the input sent by the user and send the request.
|
|
// - Decode the response received from the API into two objects `Data` and `Error`.
|
|
// - The data field should contains the pod details same as `PodResult` struct.
|
|
// - The error field should contain the error message.
|
|
fn (mut rp RunPod) create_on_demand_pod_request(request PodFindAndDeployOnDemandRequest) !PodResult {
|
|
gql := build_query(
|
|
query_type: .mutation
|
|
method_name: 'podFindAndDeployOnDemand'
|
|
request_model: request
|
|
response_model: PodResult{}
|
|
)
|
|
response := rp.make_request[GqlResponse[PodResult]](.post, '/graphql', gql)!
|
|
return response.data['podFindAndDeployOnDemand'] or {
|
|
return error('Could not find "podFindAndDeployOnDemand" in response data: ${response.data}')
|
|
}
|
|
}
|
|
|
|
// #### Internally method doing a network call to create a new spot pod.
|
|
// - Build the required query based pn the input sent by the user and send the request.
|
|
// - Decode the response received from the API into two objects `Data` and `Error`.
|
|
// - The data field should contains the pod details same as `PodResult` struct.
|
|
// - The error field should contain the error message.
|
|
fn (mut rp RunPod) create_spot_pod_request(input PodRentInterruptableInput) !PodResult {
|
|
gql := build_query(
|
|
query_type: .mutation
|
|
method_name: 'podRentInterruptable'
|
|
request_model: input
|
|
response_model: PodResult{}
|
|
)
|
|
response := rp.make_request[GqlResponse[PodResult]](.post, '/graphql', gql)!
|
|
return response.data['podRentInterruptable'] or {
|
|
return error('Could not find "podRentInterruptable" in response data: ${response.data}')
|
|
}
|
|
}
|
|
|
|
// #### Internally method doing a network call to start on demand pod.
|
|
// - Build the required query based pn the input sent by the user and send the request.
|
|
// - Decode the response received from the API into two objects `Data` and `Error`.
|
|
// - The data field should contains the pod details same as `PodResult` struct.
|
|
// - The error field should contain the error message.
|
|
fn (mut rp RunPod) start_on_demand_pod_request(input PodResume) !PodResult {
|
|
gql := build_query(
|
|
query_type: .mutation
|
|
method_name: 'podResume'
|
|
request_model: input
|
|
response_model: PodResult{}
|
|
)
|
|
response := rp.make_request[GqlResponse[PodResult]](.post, '/graphql', gql)!
|
|
return response.data['podResume'] or {
|
|
return error('Could not find "podResume" in response data: ${response.data}')
|
|
}
|
|
}
|
|
|
|
// #### Internally method doing a network call to start spot pod.
|
|
// - Build the required query based pn the input sent by the user and send the request.
|
|
// - Decode the response received from the API into two objects `Data` and `Error`.
|
|
// - The data field should contains the pod details same as `PodResult` struct.
|
|
// - The error field should contain the error message.
|
|
fn (mut rp RunPod) start_spot_pod_request(input PodBidResume) !PodResult {
|
|
gql := build_query(
|
|
query_type: .mutation
|
|
method_name: 'podBidResume'
|
|
request_model: input
|
|
response_model: PodResult{}
|
|
)
|
|
response := rp.make_request[GqlResponse[PodResult]](.post, '/graphql', gql)!
|
|
return response.data['podBidResume'] or {
|
|
return error('Could not find "podBidResume" in response data: ${response.data}')
|
|
}
|
|
}
|
|
|
|
// #### Internally method doing a network call to stop a pod.
|
|
// - Build the required query based pn the input sent by the user and send the request.
|
|
// - Decode the response received from the API into two objects `Data` and `Error`.
|
|
// - The data field should contains the pod details same as `PodResult` struct.
|
|
// - The error field should contain the error message.
|
|
fn (mut rp RunPod) stop_pod_request(input PodResume) !PodResult {
|
|
gql := build_query(
|
|
query_type: .mutation
|
|
method_name: 'podStop'
|
|
request_model: input
|
|
response_model: PodResult{}
|
|
)
|
|
response := rp.make_request[GqlResponse[PodResult]](.post, '/graphql', gql)!
|
|
return response.data['podStop'] or {
|
|
return error('Could not find "podStop" in response data: ${response.data}')
|
|
}
|
|
}
|