abi · December 30, 2024 18:54 · Dec 5, 2022 · Nov 9, 2022 · Nov 9, 2022 · Nov 8, 2022
diff --git a/gpt3-in-your-sheets.js b/gpt3-in-your-sheets.js
@@ -8,11 +8,11 @@ const MAX_TOKENS = 200;
  *
  * @param {string} prompt Prompt
  * @param {number} temperature (Optional) Temperature. 1 is super creative while 0 is very exact and precise. Defaults to 0.4.
- * @param {string} model (Optional) GPT-3 Model to use. Defaults to "text-davinci-002".
+ * @param {string} model (Optional) GPT-3 Model to use. Defaults to "text-davinci-003".
  * @return Completion returned by GPT-3
  * @customfunction
  */
-function AI(prompt, temperature = 0.4, model = "text-davinci-002") {
+function AI(prompt, temperature = 0.4, model = "text-davinci-003") {
   const url = "https://api.openai.com/v1/completions";
   const payload = {
     model: model,
@@ -40,7 +40,7 @@ function AI(prompt, temperature = 0.4, model = "text-davinci-002") {
 function CATEGORIZE(categories, input, rules=[]) {
   const prompt = "The available categories are " + categories.map((c) => `"${c}"`).join(", ") + ". " + rules.join(". ") + "The category for '" + input + "' is ";
   console.log(prompt);
-  const completion = AI(prompt, 0, "text-davinci-002");
+  const completion = AI(prompt, 0, "text-davinci-003");
   // Replace "s and .s at the start and end of the string
   return completion.replace(/^"/g, '').replace(/["|.]{0,2}$/, '');
 }

diff --git a/gpt3-in-your-sheets.js b/gpt3-in-your-sheets.js
@@ -38,7 +38,6 @@ function AI(prompt, temperature = 0.4, model = "text-davinci-002") {
  * @customfunction
  */
 function CATEGORIZE(categories, input, rules=[]) {
-  // map((r) => r ? r.replace(/.$/, '') : '')
   const prompt = "The available categories are " + categories.map((c) => `"${c}"`).join(", ") + ". " + rules.join(". ") + "The category for '" + input + "' is ";
   console.log(prompt);
   const completion = AI(prompt, 0, "text-davinci-002");

diff --git a/gpt3-in-your-sheets.js b/gpt3-in-your-sheets.js
@@ -7,8 +7,8 @@ const MAX_TOKENS = 200;
  * Completes your prompt with GPT-3
  *
  * @param {string} prompt Prompt
- * @param {number} temperature Temperature: 1 is super creative while 0 is very exact and precise. Defaults to 0.4.
- * @param {string} model GPT-3 Model to use. Defaults to "text-davinci-002".
+ * @param {number} temperature (Optional) Temperature. 1 is super creative while 0 is very exact and precise. Defaults to 0.4.
+ * @param {string} model (Optional) GPT-3 Model to use. Defaults to "text-davinci-002".
  * @return Completion returned by GPT-3
  * @customfunction
  */
@@ -31,10 +31,14 @@ function AI(prompt, temperature = 0.4, model = "text-davinci-002") {
 
 /**
  * Classifies an item into a fixed set of categories
- *
+ * @param {range} categories Set of categories
+ * @param {string} item Item to classify
+ * @param {range} rules (Optional) Set of rules written in plain text
+ * @return Completion returned by GPT-3
  * @customfunction
  */
 function CATEGORIZE(categories, input, rules=[]) {
+  // map((r) => r ? r.replace(/.$/, '') : '')
   const prompt = "The available categories are " + categories.map((c) => `"${c}"`).join(", ") + ". " + rules.join(". ") + "The category for '" + input + "' is ";
   console.log(prompt);
   const completion = AI(prompt, 0, "text-davinci-002");

diff --git a/gpt3-in-your-sheets.js b/gpt3-in-your-sheets.js
@@ -40,4 +40,6 @@ function CATEGORIZE(categories, input, rules=[]) {
   const completion = AI(prompt, 0, "text-davinci-002");
   // Replace "s and .s at the start and end of the string
   return completion.replace(/^"/g, '').replace(/["|.]{0,2}$/, '');
-}
+}
+
+// For more cool AI snippets and demos, follow me on Twitter: https://twitter.com/_abi_
diff --git a/gpt3-in-your-sheets.js b/gpt3-in-your-sheets.js
@@ -1,6 +1,8 @@
 const SECRET_KEY = ENTER YOUR SECRET KEY HERE;
 const MAX_TOKENS = 200;
 
+// For more cool AI snippets and demos, follow me on Twitter: https://twitter.com/_abi_
+
 /**
  * Completes your prompt with GPT-3
  *

diff --git a/gpt3-in-your-sheets.js b/gpt3-in-your-sheets.js
@@ -33,8 +33,9 @@ function AI(prompt, temperature = 0.4, model = "text-davinci-002") {
  * @customfunction
  */
 function CATEGORIZE(categories, input, rules=[]) {
-  const prompt = "The available categories are " + categories.map((c) => `"${c}"`).join(", ") + "." + rules.join(".") + ". The category for '" + input + "' is ";
+  const prompt = "The available categories are " + categories.map((c) => `"${c}"`).join(", ") + ". " + rules.join(". ") + "The category for '" + input + "' is ";
+  console.log(prompt);
   const completion = AI(prompt, 0, "text-davinci-002");
   // Replace "s and .s at the start and end of the string
   return completion.replace(/^"/g, '').replace(/["|.]{0,2}$/, '');
-}
+}
diff --git a/gpt3-in-your-sheets → gpt3-in-your-sheets.js b/gpt3-in-your-sheets → gpt3-in-your-sheets.js
diff --git a/gpt3-in-your-sheets b/gpt3-in-your-sheets
@@ -0,0 +1,40 @@
+const SECRET_KEY = ENTER YOUR SECRET KEY HERE;
+const MAX_TOKENS = 200;
+
+/**
+ * Completes your prompt with GPT-3
+ *
+ * @param {string} prompt Prompt
+ * @param {number} temperature Temperature: 1 is super creative while 0 is very exact and precise. Defaults to 0.4.
+ * @param {string} model GPT-3 Model to use. Defaults to "text-davinci-002".
+ * @return Completion returned by GPT-3
+ * @customfunction
+ */
+function AI(prompt, temperature = 0.4, model = "text-davinci-002") {
+  const url = "https://api.openai.com/v1/completions";
+  const payload = {
+    model: model,
+    prompt: prompt,
+    temperature: temperature,
+    max_tokens: MAX_TOKENS,
+  };
+  const options = {
+    contentType: "application/json",
+    headers: { Authorization: "Bearer " + SECRET_KEY },
+    payload: JSON.stringify(payload),
+  };
+  const res = JSON.parse(UrlFetchApp.fetch(url, options).getContentText());
+  return res.choices[0].text.trim();
+}
+
+/**
+ * Classifies an item into a fixed set of categories
+ *
+ * @customfunction
+ */
+function CATEGORIZE(categories, input, rules=[]) {
+  const prompt = "The available categories are " + categories.map((c) => `"${c}"`).join(", ") + "." + rules.join(".") + ". The category for '" + input + "' is ";
+  const completion = AI(prompt, 0, "text-davinci-002");
+  // Replace "s and .s at the start and end of the string
+  return completion.replace(/^"/g, '').replace(/["|.]{0,2}$/, '');
+}