sqgen-risk 0.1.0

This is a template repository for Python projects that use uv for their dependency management.

sqgen_risk

This is a template repository for Python projects that use uv for their dependency management.

• Github repository: https://github.com/SquareGen-LLC/saas-sdk/
• Documentation https://github.com/SquareGen-LLC/saas-sdk/blob/main/README.md

Installation

pip install sqgen-risk

With uv:

uv add sqgen-risk

Basic Usage Example

Below is a basic example of how to use the SDK to authenticate, list models, submit an inference, and retrieve results:

from os import getenv
from sqgen_risk.client import Client

# Initialize the client
client = Client()

# Authenticate using credentials from environment variables
email = getenv("API_EMAIL")
password = getenv("API_PASSWORD")
if not email or not password:
    raise RuntimeError("Missing required environment variables")

token = client.authenticate(email=email, password=password)
print(f"Authenticated successfully. Token: {token}")

# List available models
models = client.list_models()
print("\nAvailable models:")
for model in models:
    print(f"- {model.name} (ID: {model.model_id})")

# Submit an inference request
inference = client.create_inference(model_id="model-123", file_path="data.csv")
print(f"\nCreated inference request: {inference.inference_request_id}")

# Check inference status and get results
result = client.get_inference_result(inference.inference_request_id)
if result is None:
    print("Inference is still processing...")
else:
    print("Inference completed. Saving results...")
    with open("result.csv", "wb") as f:
        f.write(result)

# List all inferences
inferences = client.list_inferences()
print("\nAll inference requests:")
for inf in inferences:
    print(f"- ID: {inf.inference_request_id}, Model: {inf.model_id}, Created: {inf.created_at}")

Set the following environment variables before running the example:

export API_EMAIL="[e-mail]"
export API_PASSWORD="your-password"

Replace model-123 and data.csv with your actual model ID and CSV file path.