/opt/canhelp/node_modules/@anthropic-ai/sdk/src/lib
Edit: /opt/canhelp/node_modules/@anthropic-ai/sdk/src/lib/parser.ts (3874B)
import type { Logger } from '../client';
import { AnthropicError } from '../core/error';
import {
ContentBlock,
JSONOutputFormat,
Message,
OutputConfig,
TextBlock,
MessageCreateParams,
} from '../resources/messages/messages';
// vendored from typefest just to make things look a bit nicer on hover
type Simplify
= { [KeyType in keyof T]: T[KeyType] } & {};
type AutoParseableOutputConfig = Omit & {
format?: JSONOutputFormat | AutoParseableOutputFormat | null;
};
export type ParseableMessageCreateParams = Simplify<
Omit & {
output_config?: AutoParseableOutputConfig | null;
}
>;
export type ExtractParsedContentFromParams =
Params['output_config'] extends { format: AutoParseableOutputFormat } ? P : null;
export type AutoParseableOutputFormat = JSONOutputFormat & {
parse(content: string): ParsedT;
};
export type ParsedMessage = Message & {
content: Array>;
parsed_output: ParsedT | null;
};
export type ParsedContentBlock =
| (TextBlock & { parsed_output: ParsedT | null })
| Exclude;
function getOutputFormat(
params: ParseableMessageCreateParams | null,
): JSONOutputFormat | AutoParseableOutputFormat | null | undefined {
return params?.output_config?.format;
}
export function maybeParseMessage(
message: Message,
params: Params,
opts: { logger: Logger },
): ParsedMessage>> {
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 ParsedContentBlock>>;
return parsedBlock;
}
return block;
}),
parsed_output: null,
} as ParsedMessage>>;
}
return parseMessage(message, params, opts);
}
export function parseMessage(
message: Message,
params: Params,
opts: { logger: Logger },
): ParsedMessage> {
let firstParsedOutput: ReturnType> | null = null;
const content: Array>> = message.content.map(
(block) => {
if (block.type === 'text') {
const parsedOutput = parseOutputFormat(params, block.text);
if (firstParsedOutput === null) {
firstParsedOutput = parsedOutput;
}
const parsedBlock = Object.defineProperty({ ...block }, 'parsed_output', {
value: parsedOutput,
enumerable: false,
}) as ParsedContentBlock>;
return parsedBlock;
}
return block;
},
);
return {
...message,
content,
parsed_output: firstParsedOutput,
} as ParsedMessage>;
}
function parseOutputFormat(
params: Params,
content: string,
): ExtractParsedContentFromParams | 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}`);
}
}