fix undefined keys

huggingface · Oct 22, 2024 · 5bc694b · 5bc694b
1 parent a444cac
commit 5bc694b
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/packages/tasks/src/snippets/curl.ts b/packages/tasks/src/snippets/curl.ts
@@ -54,9 +54,9 @@ export const snippetTextGeneration = (
  ];
 
  const config = {
- temperature: opts?.temperature,
+ ...(opts?.temperature ? { temperature: opts.temperature } : undefined),
  max_tokens: opts?.max_tokens ?? 500,
- top_p: opts?.top_p,
+ ...(opts?.top_p ? { top_p: opts.top_p } : undefined),
  };
  return {
  content: `curl 'https://api-inference.huggingface.co/models/${model.id}/v1/chat/completions' \\

diff --git a/packages/tasks/src/snippets/js.ts b/packages/tasks/src/snippets/js.ts
@@ -60,9 +60,9 @@ export const snippetTextGeneration = (
  const messagesStr = formatGenerationMessages({ messages, sep: ",\n\t\t", start: "[\n\t\t", end: "\n\t]" });
 
  const config = {
- temperature: opts?.temperature,
+ ...(opts?.temperature ? { temperature: opts.temperature } : undefined),
  max_tokens: opts?.max_tokens ?? 500,
- top_p: opts?.top_p,
+ ...(opts?.top_p ? { top_p: opts.top_p } : undefined),
  };
  const configStr = formatGenerationConfig({ config, sep: ",\n\t", start: "", end: "" });
 

diff --git a/packages/tasks/src/snippets/python.ts b/packages/tasks/src/snippets/python.ts
@@ -36,9 +36,9 @@ export const snippetConversational = (
  const messagesStr = formatGenerationMessages({ messages, sep: ",\n\t", start: `[\n\t`, end: `\n]` });
 
  const config = {
- temperature: opts?.temperature,
+ ...(opts?.temperature ? { temperature: opts.temperature } : undefined),
  max_tokens: opts?.max_tokens ?? 500,
- top_p: opts?.top_p,
+ ...(opts?.top_p ? { top_p: opts.top_p } : undefined),
  };
  const configStr = formatGenerationConfig({ config, sep: ",\n\t", start: "", end: "", connector: "=" });