diff --git a/src/funcs/generalPartition.ts b/src/funcs/generalPartition.ts index fb40638..a6a6dc6 100644 --- a/src/funcs/generalPartition.ts +++ b/src/funcs/generalPartition.ts @@ -245,8 +245,19 @@ export async function generalPartition( const path = pathToFunc("/general/v0/general")(); + let acceptHeader = "application/json"; + if (payload.partition_parameters.output_format) { + switch (payload.partition_parameters.output_format) { + case "text/csv": + acceptHeader = "text/csv"; + break; + default: + acceptHeader = "application/json"; + } + } + const headers = new Headers({ - Accept: "application/json", + Accept: acceptHeader, "unstructured-api-key": encodeSimple( "unstructured-api-key", payload["unstructured-api-key"],