Nick: fixed /extract without a schema

This commit is contained in:
Nicolas 2024-12-03 12:08:15 -03:00
parent e485ea7e40
commit eb2e51e50b

View File

@ -162,7 +162,7 @@ export async function generateOpenAICompletions(logger: Logger, options: Extract
extract = JSON.parse(jsonCompletion.choices[0].message.content); extract = JSON.parse(jsonCompletion.choices[0].message.content);
} else { } else {
const extractData = JSON.parse(jsonCompletion.choices[0].message.content); const extractData = JSON.parse(jsonCompletion.choices[0].message.content);
extract = extractData.data.extract; extract = options.schema ? extractData.data.extract : extractData;
} }
} catch (e) { } catch (e) {
logger.error("Failed to parse returned JSON, no schema specified.", { error: e }); logger.error("Failed to parse returned JSON, no schema specified.", { error: e });