Update code.gs
Browse filesFixed payloadString bug: Corrected JSON stringification of payload.
code.gs
CHANGED
|
@@ -1,7 +1,4 @@
|
|
| 1 |
-
// Default system prompt used if none is provided
|
| 2 |
const DEFAULT_SYSTEM_PROMPT = 'You are a helpful and honest assistant. Please, respond concisely and truthfully.';
|
| 3 |
-
|
| 4 |
-
// Base model for Hugging Face API
|
| 5 |
const DEFAULT_MODEL = "mistralai/Mistral-7B-Instruct-v0.3";
|
| 6 |
|
| 7 |
/**
|
|
@@ -35,7 +32,7 @@ function showApiKeyPrompt() {
|
|
| 35 |
* Sends a request to the Hugging Face API with the specified prompt and model.
|
| 36 |
*
|
| 37 |
* @param {string} prompt - The input prompt to be sent to the model.
|
| 38 |
-
* @param {string} [model
|
| 39 |
* @param {string} [systemPrompt=DEFAULT_SYSTEM_PROMPT] - The system prompt to customize the assistant's behavior.
|
| 40 |
* @returns {object} JSON response from the Hugging Face API.
|
| 41 |
* @throws {Error} If the API key is not set or if the API request fails.
|
|
@@ -46,23 +43,21 @@ function queryHuggingFace(prompt, model = DEFAULT_MODEL, systemPrompt = DEFAULT_
|
|
| 46 |
throw new Error('Please enter your Hugging Face API key using the menu.');
|
| 47 |
}
|
| 48 |
|
| 49 |
-
const url = `https://api-inference.huggingface.co/models/${model}`;
|
| 50 |
-
const headers = {
|
| 51 |
-
"Authorization": `Bearer ${apiKey}`,
|
| 52 |
-
"Content-Type": "application/json"
|
| 53 |
-
};
|
| 54 |
-
|
| 55 |
-
|
| 56 |
const formattedPrompt = `<s> [INST] ${systemPrompt} [/INST] ${prompt} </s>`;
|
| 57 |
const payload = {
|
| 58 |
"inputs": formattedPrompt
|
| 59 |
};
|
| 60 |
|
|
|
|
| 61 |
|
|
|
|
| 62 |
const options = {
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
|
|
|
|
|
|
|
|
|
| 66 |
};
|
| 67 |
|
| 68 |
try {
|
|
@@ -76,11 +71,11 @@ function queryHuggingFace(prompt, model = DEFAULT_MODEL, systemPrompt = DEFAULT_
|
|
| 76 |
|
| 77 |
/**
|
| 78 |
* Custom Google Sheets formula to query the Hugging Face API and return the generated text.
|
| 79 |
-
* Function to create the custom formula =HF(prompt
|
| 80 |
*
|
| 81 |
* @param {string} prompt - The input prompt to be sent to the model.
|
| 82 |
-
* @param {string} [model
|
| 83 |
-
* @param {string} [systemPrompt
|
| 84 |
* @returns {string} The generated output text from the Hugging Face API, or an error message if the request fails.
|
| 85 |
*/
|
| 86 |
function HF(prompt, model = DEFAULT_MODEL, systemPrompt = DEFAULT_SYSTEM_PROMPT) {
|
|
|
|
|
|
|
| 1 |
const DEFAULT_SYSTEM_PROMPT = 'You are a helpful and honest assistant. Please, respond concisely and truthfully.';
|
|
|
|
|
|
|
| 2 |
const DEFAULT_MODEL = "mistralai/Mistral-7B-Instruct-v0.3";
|
| 3 |
|
| 4 |
/**
|
|
|
|
| 32 |
* Sends a request to the Hugging Face API with the specified prompt and model.
|
| 33 |
*
|
| 34 |
* @param {string} prompt - The input prompt to be sent to the model.
|
| 35 |
+
* @param {string} [model] - The model ID to query (e.g., 'mistralai/Mistral-7B-Instruct-v0.3').
|
| 36 |
* @param {string} [systemPrompt=DEFAULT_SYSTEM_PROMPT] - The system prompt to customize the assistant's behavior.
|
| 37 |
* @returns {object} JSON response from the Hugging Face API.
|
| 38 |
* @throws {Error} If the API key is not set or if the API request fails.
|
|
|
|
| 43 |
throw new Error('Please enter your Hugging Face API key using the menu.');
|
| 44 |
}
|
| 45 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
const formattedPrompt = `<s> [INST] ${systemPrompt} [/INST] ${prompt} </s>`;
|
| 47 |
const payload = {
|
| 48 |
"inputs": formattedPrompt
|
| 49 |
};
|
| 50 |
|
| 51 |
+
const payloadString = JSON.stringify(payload);
|
| 52 |
|
| 53 |
+
const url = `https://api-inference.huggingface.co/models/${model}`;
|
| 54 |
const options = {
|
| 55 |
+
'method': 'post',
|
| 56 |
+
'contentType': 'application/json',
|
| 57 |
+
'headers': {
|
| 58 |
+
'Authorization': `Bearer ${apiKey}`
|
| 59 |
+
},
|
| 60 |
+
'payload': payloadString
|
| 61 |
};
|
| 62 |
|
| 63 |
try {
|
|
|
|
| 71 |
|
| 72 |
/**
|
| 73 |
* Custom Google Sheets formula to query the Hugging Face API and return the generated text.
|
| 74 |
+
* Function to create the custom formula =HF(prompt, model, [systemPrompt])
|
| 75 |
*
|
| 76 |
* @param {string} prompt - The input prompt to be sent to the model.
|
| 77 |
+
* @param {string} [model] - The model ID to query (e.g., 'mistralai/Mistral-7B-Instruct-v0.3').
|
| 78 |
+
* @param {string} [systemPrompt] - The system prompt to customize the assistant's behavior. Defaults to DEFAULT_SYSTEM_PROMPT.
|
| 79 |
* @returns {string} The generated output text from the Hugging Face API, or an error message if the request fails.
|
| 80 |
*/
|
| 81 |
function HF(prompt, model = DEFAULT_MODEL, systemPrompt = DEFAULT_SYSTEM_PROMPT) {
|