fix cohere models

src/lib/JsonEditor/JsonEditor.svelte

@@ -136,7 +136,11 @@
 			if (exampleHelloWorld) {
 				exampleInputs = [
 					...exampleInputs,
-					{ id: 'hello-world', label: 'hello world example', content: exampleHelloWorld }
+					{
+						id: 'hello-world',
+						label: 'hello world example',
+						content: transformInput(exampleHelloWorld, selectedTemplate.template)
+					}
 				];
 			}
 
@@ -144,7 +148,11 @@
 			if (exampleReasoning) {
 				exampleInputs = [
 					...exampleInputs,
-					{ id: 'reasoning', label: 'reasoning example', content: exampleReasoning }
+					{
+						id: 'reasoning',
+						label: 'reasoning example',
+						content: transformInput(exampleReasoning, selectedTemplate.template)
+					}
 				];
 			}
 
@@ -152,7 +160,11 @@
 			if (exampleToolUsage) {
 				exampleInputs = [
 					...exampleInputs,
-					{ id: 'tool-usage', label: 'tool usage example', content: exampleToolUsage }
+					{
+						id: 'tool-usage',
+						label: 'tool usage example',
+						content: transformInput(exampleToolUsage, selectedTemplate.template)
+					}
 				];
 			}
 
@@ -170,10 +182,6 @@
 				}
 			}
 
-			exampleInputs = exampleInputs.map((exampleInput) =>
-				transformInput(exampleInput, selectedTemplate.template)
-			);
-
 			content = exampleInputs.at(-1)?.content ?? {};
 			selectedExampleInputId = exampleInputs.at(-1)?.id ?? '';
 			value = JSON5.stringify(content, null, 2);

src/lib/example-inputs/helloWorld.ts

@@ -1,4 +1,5 @@
 import { Template } from '@huggingface/jinja';
+import { transformInput } from '$lib/utils/transformInput';
 
 const variations = {
 	variation1_with_system_prompt: {
@@ -50,7 +51,9 @@ const variations = {
 export function getExampleHelloWorld(templateStr: string): Record<string, unknown> | undefined {
 	const template = new Template(templateStr);
 	const variationSystemPrompt = variations.variation1_with_system_prompt.example;
-	const variationSystemPromptRendered = template.render(variationSystemPrompt);
+	const variationSystemPromptRendered = template.render(
+		transformInput(variationSystemPrompt, templateStr)
+	);
 	if (variationSystemPromptRendered.includes('You are a helpful assistant.')) {
 		return variations.variation1_with_system_prompt.example;
 	}

src/lib/example-inputs/reasoning.ts

@@ -1,3 +1,4 @@
+import { transformInput } from '$lib/utils/transformInput';
 import { Template } from '@huggingface/jinja';
 
 const variations = {
@@ -55,7 +56,9 @@ export function getExampleReasoning(templateStr: string): Record<string, unknown
 	}
 	const template = new Template(templateStr);
 	const variationSystemPrompt = variations.variation1_with_system_prompt.example;
-	const variationSystemPromptRendered = template.render(variationSystemPrompt);
+	const variationSystemPromptRendered = template.render(
+		transformInput(variationSystemPrompt, templateStr)
+	);
 	if (variationSystemPromptRendered.includes('You are a helpful assistant.')) {
 		return variations.variation1_with_system_prompt.example;
 	}

src/lib/example-inputs/toolUsage.ts

@@ -1,3 +1,4 @@
+import { transformInput } from '$lib/utils/transformInput';
 import { Template } from '@huggingface/jinja';
 
 const variations = {
@@ -383,7 +384,7 @@ export function getExampleToolUsage(templateStr: string): Record<string, unknown
 	const template = new Template(templateStr);
 	for (const variation of Object.values(variations)) {
 		try {
-			const variationRendered = template.render(variation.example);
+			const variationRendered = template.render(transformInput(variation.example, templateStr));
 			if (variationRendered.includes('get_weather')) {
 				return variation.example;
 			}

src/lib/utils/transformInput.ts

@@ -1,10 +1,7 @@
-export function transformInput(
-	input: { id: string; label: string; content: Record<string, unknown> },
-	templateStr: string
-) {
+export function transformInput(input: Record<string, unknown>, templateStr: string) {
 	// handle cohere special case
 	if (templateStr.includes('safety_mode')) {
-		input.content.safety_mode = '';
+		input.safety_mode = '';
 	}
 	return input;
 }