Update code.gs

Hi Florent,

Thanks a lot for the comment, it's been corrected normally, I can't wait to think about the next steps and new functions, it's really a great project 🤗

https://cdn-uploads.huggingface.co/production/uploads/6459fa0f5b3111fbe83286e1/e9q5ueTV3gOTiF9J47AU5.qt

code.gs

@@ -1,8 +1,8 @@
 // Default system prompt used if none is provided
 const DEFAULT_SYSTEM_PROMPT = 'You are a helpful and honest assistant. Please, respond concisely and truthfully.';
 
-// Base URL for Hugging Face API
-const API_URL = "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3";
+// Base model for Hugging Face API
+const DEFAULT_MODEL = "mistralai/Mistral-7B-Instruct-v0.3";
 
 /**
  * Adds a custom menu to Google Sheets to allow users to input their Hugging Face API key.
@@ -35,12 +35,12 @@ function showApiKeyPrompt() {
  * Sends a request to the Hugging Face API with the specified prompt and model.
  * 
  * @param {string} prompt - The input prompt to be sent to the model.
- * @param {string} model - The model ID to query (e.g., 'mistralai/Mistral-7B-Instruct-v0.3').
+ * @param {string} [model=DEFAULT_MODEL] - The model ID to query (e.g., 'mistralai/Mistral-7B-Instruct-v0.3').
  * @param {string} [systemPrompt=DEFAULT_SYSTEM_PROMPT] - The system prompt to customize the assistant's behavior.
  * @returns {object} JSON response from the Hugging Face API.
  * @throws {Error} If the API key is not set or if the API request fails.
  */
-function queryHuggingFace(prompt, model, systemPrompt = DEFAULT_SYSTEM_PROMPT) {
+function queryHuggingFace(prompt, model = DEFAULT_MODEL, systemPrompt = DEFAULT_SYSTEM_PROMPT) {
   const apiKey = PropertiesService.getScriptProperties().getProperty('HF_API_KEY');
   if (!apiKey) {
     throw new Error('Please enter your Hugging Face API key using the menu.');
@@ -53,7 +53,7 @@ function queryHuggingFace(prompt, model, systemPrompt = DEFAULT_SYSTEM_PROMPT) {
   };
 
 
-  const formattedPrompt = <s> [INST] ${systemPrompt} [/INST] ${prompt} </s>`;
+  const formattedPrompt = `<s> [INST] ${systemPrompt} [/INST] ${prompt} </s>`;
   const payload = {
     "inputs": formattedPrompt
   };
@@ -76,14 +76,14 @@ function queryHuggingFace(prompt, model, systemPrompt = DEFAULT_SYSTEM_PROMPT) {
 
 /**
  * Custom Google Sheets formula to query the Hugging Face API and return the generated text.
- * Function to create the custom formula =HF(prompt, model, [systemPrompt])
+ * Function to create the custom formula =HF(prompt; [model]; [systemPrompt])
  * 
  * @param {string} prompt - The input prompt to be sent to the model.
- * @param {string} model - The model ID to query (e.g., 'mistralai/Mistral-7B-Instruct-v0.3').
- * @param {string} [systemPrompt] - The system prompt to customize the assistant's behavior. Defaults to DEFAULT_SYSTEM_PROMPT.
+ * @param {string} [model=DEFAULT_MODEL] - The model ID to query (e.g., 'mistralai/Mistral-7B-Instruct-v0.3').
+ * @param {string} [systemPrompt=DEFAULT_SYSTEM_PROMPT] - The system prompt to customize the assistant's behavior. Defaults to DEFAULT_SYSTEM_PROMPT.
  * @returns {string} The generated output text from the Hugging Face API, or an error message if the request fails.
  */
-function HF(prompt, model, systemPrompt) {
+function HF(prompt, model = DEFAULT_MODEL, systemPrompt = DEFAULT_SYSTEM_PROMPT) {
   try {
     const response = queryHuggingFace(prompt, model, systemPrompt);
     if (response && response.length > 0 && response[0].generated_text) {
@@ -111,4 +111,4 @@ function onInstall(e) {
     .setFunction('HF')
     .build();
   SpreadsheetApp.installUserDefinedFunction(formula);
-}
+}