import json
import os

from parable_types_web_api import RunArtifactSchemaInferenceInput
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 = RunArtifactSchemaInferenceInput.model_validate(
    json.loads("{\"artifactId\":\"550e8400-e29b-41d4-a716-446655440000\"}"),
)
result = sdk.artifacts.run_artifact_schema_inference(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.RunArtifactSchemaInferenceInput
if err := json.Unmarshal([]byte(`{"artifactId":"550e8400-e29b-41d4-a716-446655440000"}`), &input); err != nil {
	return err
}
result, err := sdk.ArtifactsNamespace.RunArtifactSchemaInference(ctx, input)

{
  "data": {
    "artifact": {
      "createdAt": "2023-11-07T05:31:56Z",
      "createdBy": "<string>",
      "fileName": "<string>",
      "fileSize": 4503599627370495,
      "fileType": "csv",
      "id": "<string>",
      "schemaInferenceStatus": "not_requested",
      "updatedAt": "2023-11-07T05:31:56Z",
      "updatedBy": "<string>",
      "uploadStatus": "pending",
      "uploadedBy": {
        "email": "jsmith@example.com",
        "id": "<string>",
        "name": "<string>"
      },
      "analysisWork": {
        "artifactId": "<string>",
        "state": "PENDING",
        "workId": "<string>",
        "errorCode": "INVALID_SOURCE",
        "errorMessage": "<string>",
        "tapId": "<string>"
      },
      "attributedTapId": "<string>",
      "dataSchema": {},
      "path": "<string>",
      "publishWork": {
        "artifactId": "<string>",
        "state": "PENDING",
        "workId": "<string>",
        "errorCode": "INVALID_SOURCE",
        "errorMessage": "<string>",
        "tapId": "<string>"
      }
    },
    "sampleRows": [
      {}
    ],
    "inference": {
      "columnMetadata": [
        {
          "confidence": "low",
          "demoted": true,
          "fieldPath": "<string>",
          "sampleFailures": 0,
          "sampleSeen": 0,
          "firstSampleError": "<string>"
        }
      ],
      "model": "<string>",
      "schema": {},
      "tokenInputCount": 0,
      "tokenOutputCount": 0,
      "processor": "VERTEX"
    },
    "inferenceError": {
      "code": "WA-AF-001",
      "message": "<string>"
    },
    "inferenceRefusal": "ENVIRONMENT_POLICY_MISSING"
  },
  "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>"
}

Gateway Timeout