refactor stringifyGenerationConfig

Author: mishig25

packages/tasks/src/snippets/common.ts

@@ -23,25 +23,17 @@ export function stringifyMessages(
 
 type PartialGenerationParameters = Partial<Pick<GenerationParameters, "temperature" | "max_tokens" | "top_p">>;
 
-export interface StringifyGenerationConfigOptions {
-	sep: string;
-	start: string;
-	end: string;
-	attributeValueConnector: string;
-	attributeKeyQuotes?: boolean;
-}
-
 export function stringifyGenerationConfig(
 	config: PartialGenerationParameters,
-	opts: StringifyGenerationConfigOptions
+	opts: {
+		indent: string;
+		attributeValueConnector: string;
+		attributeKeyQuotes?: boolean;
+	}
 ): string {
 	const quote = opts.attributeKeyQuotes ? `"` : "";
 
-	return (
-		opts.start +
-		Object.entries(config)
-			.map(([key, val]) => `${quote}${key}${quote}${opts.attributeValueConnector}${val}`)
-			.join(opts.sep) +
-		opts.end
-	);
+	return Object.entries(config)
+		.map(([key, val]) => `${quote}${key}${quote}${opts.attributeValueConnector}${val}`)
+		.join(`,${opts.indent}`);
 }

packages/tasks/src/snippets/curl.ts

@@ -46,9 +46,7 @@ export const snippetTextGeneration = (
 			customContentEscaper: (str) => str.replace(/'/g, "'\\''"),
 		})},
     ${stringifyGenerationConfig(config, {
-			sep: ",\n    ",
-			start: "",
-			end: "",
+			indent: "\n    ",
 			attributeKeyQuotes: true,
 			attributeValueConnector: ": ",
 		})},

packages/tasks/src/snippets/js.ts

@@ -50,9 +50,7 @@ export const snippetTextGeneration = (
 			...(opts?.top_p ? { top_p: opts.top_p } : undefined),
 		};
 		const configStr = stringifyGenerationConfig(config, {
-			sep: ",\n\t",
-			start: "",
-			end: "",
+			indent: "\n\t",
 			attributeValueConnector: ": ",
 		});
 

packages/tasks/src/snippets/python.ts

@@ -26,9 +26,7 @@ export const snippetConversational = (
 		...(opts?.top_p ? { top_p: opts.top_p } : undefined),
 	};
 	const configStr = stringifyGenerationConfig(config, {
-		sep: ",\n\t",
-		start: "",
-		end: "",
+		indent: "\n\t",
 		attributeValueConnector: "=",
 	});