AI Integration

from gravixlayer import GravixLayer


def run_python_task(user_prompt: str) -> str:
    client = GravixLayer()  # defaults to cloud="aws", region="us-east-1"
    sandbox = None

    try:
        sandbox = client.runtime.create()  # defaults to template="base-small"

        code = f"""
prompt = {user_prompt!r}
print('Received prompt:', prompt)
print('Computation complete')
"""
        response = sandbox.run_code(code=code)

        if response.success:
            return response.text
        return f"Execution error: {response.error.value if response.error else 'unknown'}"

    finally:
        if sandbox is not None:
            sandbox.kill()

import { GravixLayer } from 'gravixlayer';

async function runPythonTask(userPrompt: string): Promise<string> {
  const client = new GravixLayer(); // defaults to cloud="aws", region="us-east-1"
  const sandbox = await client.runtime.create(); // defaults to template="base-small"

  try {
    const code = `
prompt = ${JSON.stringify(userPrompt)}
print('Received prompt:', prompt)
print('Computation complete')
`;
    const response = await sandbox.runCode(code);
    if (response.success) return response.text;
    const err = response.error;
    const detail = typeof err === 'string' ? err : err?.value ?? 'unknown';
    return `Execution error: ${detail}`;
  } finally {
    await sandbox.kill();
  }
}

# Requires: GRAVIXLAYER_API_KEY, jq
# CLI has no inline run_code — write generated code to a temp file.

RT=$(gravixlayer runtime create --wait --output json | jq -r '.runtime_id')

cat > /tmp/task.py <<'EOF'
prompt = "summarize sales"
print("Received prompt:", prompt)
print("Computation complete")
EOF

gravixlayer runtime run "$RT" /tmp/task.py
gravixlayer runtime kill "$RT" -y

# Requires: GRAVIXLAYER_API_KEY, jq
API="https://api.gravixlayer.ai/v1"
AUTH=(-H "Authorization: Bearer $GRAVIXLAYER_API_KEY" -H "Content-Type: application/json")

RT=$(curl -sS -X POST "$API/agents/runtime" "${AUTH[@]}" \
  -d '{"template":"base-small","cloud":"aws","region":"us-east-1"}' | jq -r '.runtime_id')

for i in $(seq 1 60); do
  STATUS=$(curl -sS "$API/agents/runtime/$RT" -H "Authorization: Bearer $GRAVIXLAYER_API_KEY" | jq -r '.status')
  [ "$STATUS" = "running" ] && break
  sleep 2
done

curl -sS -X POST "$API/agents/runtime/$RT/code/run" "${AUTH[@]}" \
  -d '{"code":"prompt = \"summarize sales\"\nprint(\"Received prompt:\", prompt)\nprint(\"Computation complete\")","language":"python"}'
echo

curl -sS -X DELETE "$API/agents/runtime/$RT" -H "Authorization: Bearer $GRAVIXLAYER_API_KEY"