tni1dZddlmZmZ d#dededeed ed ed ed eed eedeedeedeedeedeedeedeedeedeedeedeedeeedeedeedeedef0dZdedefd Z dededefd!Z dedefd"Z dS)$z) Runpod | API Wrapper | Mutations | Pods )ListOptionalNALLTname image_name gpu_type_id cloud_typesupport_public_ip start_sshdata_center_id country_code gpu_count volume_in_gbcontainer_disk_in_gbmin_vcpu_countmin_memory_in_gb docker_argsportsvolume_mount_pathenv template_idnetwork_volume_idallowed_cuda_versions min_download min_upload instance_idreturncg}|d|dd|dd|g|r|d||d|d|dt|||d || |d | | |d | | |d | | |d | d|8dd|D}|d|dn||d|d|pd}||d|dn|d||d|d| |d| |-|d|ddd||d|d|Jd|D}|dd|d||d|d||d|d||d |||d!||rd"nd#}d|}d$|d%|d&S)'a Generates a mutation to deploy a pod on demand. Args: name: Name of the pod image_name: Docker image name gpu_type_id: GPU type ID for GPU pods, None for CPU pods cloud_type: Cloud type (ALL, COMMUNITY, or SECURE) support_public_ip: Whether to support public IP start_ssh: Whether to start SSH service data_center_id: Data center ID country_code: Country code for pod location gpu_count: Number of GPUs (for GPU pods) volume_in_gb: Volume size in GB container_disk_in_gb: Container disk size in GB min_vcpu_count: Minimum vCPU count min_memory_in_gb: Minimum memory in GB docker_args: Docker arguments ports: Port mappings (e.g. "8080/tcp,22/tcp") volume_mount_path: Volume mount path env: Environment variables dict template_id: Template ID network_volume_id: Network volume ID allowed_cuda_versions: List of allowed CUDA versions min_download: Minimum download speed in Mbps min_upload: Minimum upload speed in Mbps instance_id: Instance ID for CPU pods Returns: str: GraphQL mutation string zname: ""z imageName: "z cloudType: zstartSsh: trueNz gpuTypeId: "zsupportPublicIp: z gpuCount: z volumeInGb: zminVcpuCount: zminMemoryInGb: z dockerArgs: "z, c3"K|] }d|dV dS)rN).0vs V/var/www/html/gpu-tools/venv/lib/python3.11/site-packages/runpod/api/mutations/pods.py z3generate_pod_deployment_mutation..^s*%N%N1h!hhh%N%N%N%N%N%NzallowedCudaVersions: []z instanceId: "z runpod-ubuntuzdataCenterId: "zdataCenterId: nullzcountryCode: "zcontainerDiskInGb: zports: " zvolumeMountPath: "c&g|]\}}d|d|dS)z{ key: "z ", value: "z" }r!)r"kr#s r$ z4generate_pod_deployment_mutation..vs0OOO41a666q666OOOr&zenv: [z templateId: "znetworkVolumeId: "z minDownload: z minUpload: podFindAndDeployOnDemand deployCpuPodz mutation { z( input: { z } ) { id imageName env machineId machine { podHostId } } } )extendappendstrlowerjoinreplaceitems)rrrr r r r r rrrrrrrrrrrrrrr input_fields cuda_versions env_items mutation_type input_strings r$ generate_pod_deployment_mutationr; spL$$z$$$"j"" .,---9;999:::P4E0F0F0L0L0N0NPPQQQ     8Y 8 8 9 9 9  #    =| = = > > >  %    A A A B B B  '    D2B D D E E E  "    > > > > ? ? ? , II%N%N8M%N%N%NNNM    I I I I J J J  "    > > > > ? ? ?!4_ !?n???@@@@0111<\<<<==='H2FHHIII @u}}S"'='=@@@AAA$E1BEEEFFF OO399;;OOO M99\**L        r&pod_idcd|dS)z- Generates a mutation to stop a pod. z1 mutation { podStop(input: { podId: "zD" }) { id desiredStatus } } r!r<s r$generate_pod_stop_mutationr?s #)   r&cd|d|dS)z/ Generates a mutation to resume a pod. z3 mutation { podResume(input: { podId: "z ", gpuCount: z }) { id desiredStatus imageName env machineId machine { podHostId } } } r!)r<rs r$generate_pod_resume_mutationrAs+ %+  :C    r&cd|dS)z2 Generates a mutation to terminate a pod. z6 mutation { podTerminate(input: { podId: "z" }) } r!r>s r$generate_pod_terminate_mutationrCs (.   r&)NrTTNNNNNNNNNNNNNNNNN) __doc__typingrrr1boolintdictr;r?rArCr!r&r$rIsQ "!!!!!!! "&"$("&#"&*.$(&*!%'+!%'+15"& $!%/JJ JJ#J J  J  JSMJ3-J}J3-J#3-JSMJsmJ#J C=J  }!J" $#J$#%J& }'J($DI.)J*3-+J, -J.#/J0 1JJJJZ s s    (CCr&