Skip to main content

Official InferenceKey SDK — declare AI workloads in code, ensure they exist, and call the resulting OpenAI-compatible endpoints.

Project description

inferencekey (Python)

Official Python SDK for the InferenceKey platform — a thin, typed layer over the shared Rust core (inferencekey-core).

Declare AI workloads in code, ensure they exist on the platform, and call the resulting OpenAI-compatible endpoints.

from inferencekey import ManagementClient, DataClient, WorkloadSpec, Backend

mgmt = ManagementClient.from_env(project="acme")          # INFERENCEKEY_SDK_TOKEN
ref = mgmt.ensure(WorkloadSpec(
    name="support-bot", slug="support-bot",
    model="meta-llama/Llama-3.1-8B-Instruct", backend=Backend.VLLM,
    command="vllm serve meta-llama/Llama-3.1-8B-Instruct --max-model-len 8192",
))

data = DataClient.from_env(project="acme")
out = data.endpoint(ref.workload_slug, api_key="ik_live_...").generate_text(prompt="Hola")
print(out.text)

Two tokens, two clients (least privilege): ik_sdk_ provisions (control plane); ik_live_ calls inference (data plane, per workload). Full docs at docs.inferencekey.com.

License: Apache-2.0.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

inferencekey-0.1.0-cp39-abi3-manylinux_2_34_x86_64.whl (1.7 MB view details)

Uploaded CPython 3.9+manylinux: glibc 2.34+ x86-64

File details

Details for the file inferencekey-0.1.0-cp39-abi3-manylinux_2_34_x86_64.whl.

File metadata

File hashes

Hashes for inferencekey-0.1.0-cp39-abi3-manylinux_2_34_x86_64.whl
Algorithm Hash digest
SHA256 a5f476199dc2d75a5908b728e3ca8d76d46ab1390a76fd933c2cb4547f718cf7
MD5 0a8e89f531578503ea99dc14e4b711fe
BLAKE2b-256 5ec83b92376b0775ab94b4e71b69e0534ac8444c0b44268a208f61f71ecf6fd5

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page