Skip to content

SocAIty/fastSDK

Repository files navigation

fastSDK. Any service. One typed client.

PyPI version Python versions GitHub License

Call any AI / web service like a native Python function

fastSDK turns any hosted service — OpenAPI/FastAPI, APIPod, RunPod, Replicate, Cog — into a Python client that feels like a local library: typed methods, file upload/download, async job handling and parallel execution included.

Point it at a service. Call it like a function. That's the whole idea.

import fastsdk

client = fastsdk.connect("http://localhost:8009")
job = client.submit_job("/text2voice", text="hello world")
audio = job.get_result()
audio.save("hello.mp3")

Why fastSDK?

Calling a web service from Python sounds trivial until you actually do it in production: you wait synchronously on long-running ML jobs, you hand-write request code for every endpoint, you fight with file uploads (try sending a 1 GB video through requests), you poll job status loops, and you reinvent threading to run requests in parallel.

fastSDK solves exactly that, and nothing else:

  • One call = one job. Every call returns a job object immediately. Get the result when you need it, run hundreds of jobs in parallel meanwhile.
  • Files just work. Images, audio, video are handled by media-toolkit — local paths, URLs, bytes or numpy arrays in; media objects out. Large files are uploaded via cloud storage (S3, Azure) when configured.
  • Job-based providers are normalized. Replicate, RunPod serverless, APIPod and Socaity all expose "submit, then poll" APIs with different wire formats. fastSDK handles submission, polling, progress and cancellation uniformly.
  • Codegen when you want it, not when you don't. Use connect() for instant access, or generate_stub() to get a typed .py client with one method per endpoint - autocomplete and docstrings included.

Installation

pip install fastsdk           # core
pip install fastsdk[replicate]  # + Replicate model support

Get started

Option A: connect — use a service right now

No files, no codegen. Works with a URL, an openapi.json path, or a Replicate model reference.

import fastsdk

client = fastsdk.connect("http://localhost:8009")
job = client.submit_job("/text2voice", text="hello world")
result = job.get_result()

Option B: generate_stub — typed clients for real projects

Generates a .py file with one typed method per endpoint. This is your SDK.

import fastsdk

stub = fastsdk.generate_stub("http://localhost:8009", save_path="clients/")
print(stub.path, stub.class_name)

# use it immediately ...
client = stub.client()
job = client.text2voice(text="hello world")

# ... or import it in your next run like any other module
# from clients.speechcraft import speechcraft
# client = speechcraft()

Re-running generate_stub is safe: the file is overwritten and the service registration is updated, not duplicated.

Replicate models

Official models (called via /v1/models/{owner}/{name}/predictions) and community models (called via /v1/predictions with a version) are resolved automatically — you just name the model:

import fastsdk  # requires: pip install fastsdk[replicate] and REPLICATE_API_KEY

stub = fastsdk.generate_stub("replicate:black-forest-labs/flux-schnell", save_path="clients/")
flux = stub.client()
job = flux(prompt="a t-rex on a skateboard")
image = job.get_result()

Working with jobs

job = client.swap_img_to_img(source_img="face1.jpg", target_img="face2.jpg")
job.get_result()          # block until done and return the result
job.cancel()              # cancel locally and remotely (provider permitting)

# run many jobs in parallel - this is where fastSDK shines
jobs = [client.text2voice(text=t) for t in hundred_texts]
results = fastsdk.gather_results(jobs)

API keys

Pass api_key=... to connect(), generate_stub() or the client constructor — or set environment variables: REPLICATE_API_KEY, RUNPOD_API_KEY, SOCAITY_API_KEY, or <SERVICE_ID>_API_KEY for your own services.

The four concepts

Concept What it is
Service definition The parsed, normalized description of a service: endpoints, parameters, address, provider type. Get one with fastsdk.inspect_service(source) — it has no side effects.
Registry An in-process directory of service definitions, shared by all clients. register_service() adds to it; generated stubs look their service up in it by ID.
Client The runtime object you call (FastClient). It submits jobs to the service. connect() gives you a generic one instantly.
Stub A generated .py file containing a client subclass with one typed method per endpoint. Made by generate_stub(); it's plain code — read it, version it, ship it.

CLI

Everything above also works from the terminal — same verbs, same behavior:

# What can this service do?
fastsdk inspect http://localhost:8009

# Generate a typed client stub
fastsdk generate http://localhost:8009 -o clients/ --name SpeechCraft
fastsdk generate replicate:black-forest-labs/flux-schnell --api-key r8_...

# Call an endpoint without writing any code (curl for AI services)
fastsdk call http://localhost:8009 /text2voice --text "hello world" -o hello.mp3

# Keep services around by name (persisted in ~/.fastsdk/registry)
fastsdk registry add http://localhost:8009 --name speechcraft
fastsdk registry list
fastsdk call speechcraft /text2voice --text "hi again"

Service compatibility

Works out of the box with:

fastSDK + APIPod

APIPod builds and deploys the services; fastSDK consumes them. Two beating hearts 💕 for service ↔ client interaction.

Contribute

We at SocAIty want to provide the best tools to bring generative AI to the cloud. Report bugs, ideas and feature requests in the issues section. fastSDK is MIT-licensed and free to use. Leave a star to support us!


Made with ❤️ by SocAIty

About

Connect to and convert any web endpoint into an python SDK. Built-in threading, async jobs and fastapi/runpod/FastTaskAPI endpoint support.

Resources

License

Stars

7 stars

Watchers

2 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors