/opt/canhelp/node_modules/@anthropic-ai/sdk/src/lib
Edit: /opt/canhelp/node_modules/@anthropic-ai/sdk/src/lib/beta-parser.ts (5106B)
import type { Logger } from '../client';
import { AnthropicError } from '../core/error';
import {
BetaContentBlock,
BetaJSONOutputFormat,
BetaMessage,
BetaOutputConfig,
BetaTextBlock,
MessageCreateParams,
} from '../resources/beta/messages/messages';
// vendored from typefest just to make things look a bit nicer on hover
type Simplify
= { [KeyType in keyof T]: T[KeyType] } & {};
type AutoParseableBetaOutputConfig = Omit & {
format?: BetaJSONOutputFormat | AutoParseableBetaOutputFormat | null;
};
export type BetaParseableMessageCreateParams = Simplify<
Omit & {
/**
* @deprecated Use `output_config.format` instead. This parameter will be removed in a future
* release.
*/
output_format?: BetaJSONOutputFormat | AutoParseableBetaOutputFormat | null;
output_config?: AutoParseableBetaOutputConfig | null;
}
>;
export type ExtractParsedContentFromBetaParams =
Params['output_format'] extends AutoParseableBetaOutputFormat ? P
: Params['output_config'] extends { format: AutoParseableBetaOutputFormat } ? P
: null;
export type AutoParseableBetaOutputFormat = BetaJSONOutputFormat & {
parse(content: string): ParsedT;
};
export type ParsedBetaMessage = BetaMessage & {
content: Array>;
parsed_output: ParsedT | null;
};
export type ParsedBetaContentBlock =
| (BetaTextBlock & { parsed_output: ParsedT | null })
| Exclude;
function getOutputFormat(
params: BetaParseableMessageCreateParams | null,
): BetaJSONOutputFormat | AutoParseableBetaOutputFormat | null | undefined {
// Prefer output_format (deprecated) over output_config.format for backward compatibility
return params?.output_format ?? params?.output_config?.format;
}
export function maybeParseBetaMessage(
message: BetaMessage,
params: Params,
opts: { logger: Logger },
): ParsedBetaMessage>> {
const outputFormat = getOutputFormat(params);
if (!params || !('parse' in (outputFormat ?? {}))) {
return {
...message,
content: message.content.map((block) => {
if (block.type === 'text') {
const parsedBlock = Object.defineProperty({ ...block }, 'parsed_output', {
value: null,
enumerable: false,
}) as ParsedBetaContentBlock>>;
return Object.defineProperty(parsedBlock, 'parsed', {
get() {
opts.logger.warn(
'The `parsed` property on `text` blocks is deprecated, please use `parsed_output` instead.',
);
return null;
},
enumerable: false,
});
}
return block;
}),
parsed_output: null,
} as ParsedBetaMessage>>;
}
return parseBetaMessage(message, params, opts);
}
export function parseBetaMessage(
message: BetaMessage,
params: Params,
opts: { logger: Logger },
): ParsedBetaMessage> {
let firstParsedOutput: ReturnType> | null = null;
const content: Array>> =
message.content.map((block) => {
if (block.type === 'text') {
const parsedOutput = parseBetaOutputFormat(params, block.text);
if (firstParsedOutput === null) {
firstParsedOutput = parsedOutput;
}
const parsedBlock = Object.defineProperty({ ...block }, 'parsed_output', {
value: parsedOutput,
enumerable: false,
}) as ParsedBetaContentBlock>;
return Object.defineProperty(parsedBlock, 'parsed', {
get() {
opts.logger.warn(
'The `parsed` property on `text` blocks is deprecated, please use `parsed_output` instead.',
);
return parsedOutput;
},
enumerable: false,
});
}
return block;
});
return {
...message,
content,
parsed_output: firstParsedOutput,
} as ParsedBetaMessage>;
}
function parseBetaOutputFormat(
params: Params,
content: string,
): ExtractParsedContentFromBetaParams | null {
const outputFormat = getOutputFormat(params);
if (outputFormat?.type !== 'json_schema') {
return null;
}
try {
if ('parse' in outputFormat) {
return outputFormat.parse(content);
}
return JSON.parse(content);
} catch (error) {
throw new AnthropicError(`Failed to parse structured output: ${error}`);
}
}