Skip to content

Commit

Permalink
Fix missing attributes in dotprompt serialization
Browse files Browse the repository at this point in the history
  • Loading branch information
MichaelDoyle committed May 10, 2024
1 parent 8d871f0 commit efec3cf
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion js/plugins/dotprompt/src/metadata.ts
Original file line number Diff line number Diff line change
Expand Up @@ -128,6 +128,7 @@ export function toMetadata(attributes: unknown): Partial<PromptMetadata> {
});
return stripUndefined({
name: fm.name,
variant: fm.variant,
model: fm.model,
config: fm.config,
input: fm.input
Expand All @@ -145,6 +146,7 @@ export function toMetadata(attributes: unknown): Partial<PromptMetadata> {
export function toFrontmatter(md: PromptMetadata): PromptFrontmatter {
return stripUndefined({
name: md.name,
variant: md.variant,
model: typeof md.model === 'string' ? md.model : md.model?.name,
config: md.config,
input: md.input
Expand All @@ -169,6 +171,6 @@ export function toFrontmatter(md: PromptMetadata): PromptFrontmatter {
tools: md.tools?.map((t) =>
typeof t === 'string' ? t : (t as any).__action?.name || (t as any).name
),
variant: md.variant,
candidates: md.candidates,
});
}

0 comments on commit efec3cf

Please sign in to comment.