import json
import os

from parable_types_web_api import ImmediatePlotQueryInput
from parable_web_api_sdk import ClientConfig, WebApiSDK

sdk = WebApiSDK(
    ClientConfig(
        base_url=os.environ["PARABLE_API_URL"],
        auth_token=os.environ["PARABLE_API_TOKEN"],
        default_headers={"X-Tenant": os.environ["PARABLE_WORKSPACE"]},
    )
)

input_data = ImmediatePlotQueryInput.model_validate(
    json.loads("{\"sql\":\"example\"}"),
)
result = sdk.parables.estimate_immediate_plot_query(input_data)

sdk, err := webapisdk.New(webapisdk.SDKConfig{
	BaseURL: os.Getenv("PARABLE_API_URL"),
	Auth:    &webapisdk.AuthConfig{Token: os.Getenv("PARABLE_API_TOKEN")},
	RequestInterceptor: func(ctx context.Context, req *http.Request) error {
		req.Header.Set("X-Tenant", os.Getenv("PARABLE_WORKSPACE"))
		return nil
	},
})
if err != nil {
	return err
}

var input webapitypes.ImmediatePlotQueryInput
if err := json.Unmarshal([]byte(`{"sql":"example"}`), &input); err != nil {
	return err
}
result, err := sdk.ParablesNamespace.EstimateImmediatePlotQuery(ctx, input)

{
  "data": {
    "aiCallSites": [
      {
        "functionNames": [
          "<string>"
        ],
        "maxCalls": 1
      }
    ],
    "computeCost": {
      "byBytesRead": 123,
      "centsPerCpuHour": 123,
      "centsPerGibRead": 123,
      "byCpuTime": {
        "high": 123,
        "low": 123,
        "modelVersion": "<string>"
      }
    },
    "input": {
      "inputBytes": 0,
      "inputFileCount": 0,
      "unmeasuredLeaves": 0,
      "inputRows": 1
    },
    "aiCallLimits": {
      "maxCalls": 4503599627370495,
      "maxCostCents": 4503599627370495
    },
    "elapsedMs": {
      "high": 123,
      "low": 123,
      "modelVersion": "<string>"
    }
  },
  "meta": {
    "requestId": "<string>",
    "totalCount": 123
  },
  "links": {
    "first": "<string>",
    "last": "<string>",
    "next": "<string>",
    "prev": "<string>"
  }
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

Too Many Requests

{
  "detail": "<string>",
  "status": 400,
  "title": "Bad Request",
  "type": "about:blank",
  "code": "WA-VL-001",
  "details": "<unknown>",
  "errors": "<unknown>",
  "requestId": "<string>"
}

Gateway Timeout