changes

waynehamadi · waynehamadi · commit 79e11f0adb41 · 2023-11-29T10:30:19.000-05:00
diff --git a/.gitignore b/.gitignore
@@ -6,4 +6,5 @@ out
 stackRegistry.json
 dist
 .env
-.vscode/launch.json
+.vscode/launch.json
+test-workspace/.env
diff --git a/preload-dotenv.js b/preload-dotenv.js
@@ -0,0 +1 @@
+require('dotenv').config();
diff --git a/src/stack/getCaviat.ts b/src/stack/getCaviat.ts
@@ -0,0 +1,24 @@
+export function getCaviat(brief: string) {
+  // brief contains openai or gpt-4 or gpt-3.5-turbo or gpt4 or gpt3.5turbo
+  //convert brief to lowercase
+  brief = brief.toLowerCase();
+  if (brief.includes('openai') || brief.includes('gpt-4') || brief.includes('gpt4') || brief.includes('gpt-3.5-turbo') || brief.includes('gpt3.5turbo')) {
+    return `Examples:
+import OpenAI from 'openai';
+
+const openai = new OpenAI({
+  apiKey: process.env.OPENAI_API_KEY
+});
+
+async function main() {
+  const chatCompletion = await openai.chat.completions.create({
+    messages: [{ role: 'user', content: 'Say this is a test' }],
+    model: 'gpt-3.5-turbo',
+  });
+}
+
+main();
+`;
+  }
+  return ``;
+}
diff --git a/src/stack/integrations/generic/generateFunction.ts b/src/stack/integrations/generic/generateFunction.ts
@@ -3,6 +3,7 @@ import { BoilerplateMetadata, Message } from '../lib/types';
 import { processBoilerplate, readFunctionToString } from '../lib/utils';
 import { combineSkeleton } from '../../createSkeleton';
 import { openai } from '../openai/construct';
+import { getCaviat } from '../../getCaviat';
 
 export default async function generateFunction(
   briefSkeleton: string,
@@ -44,6 +45,7 @@ ${startingBoilerplate}`,
 ${startingBoilerplate}`,
     });
   }
+  const caveats = getCaviat(brief);
 
   try {
     const response = await openai.chat.completions.create({
@@ -58,7 +60,11 @@ Ensure that you keep the return type, function name, and parameters the same. Yo
         },
         {
           role: 'user',
-          content: `Here is the user brief: ${brief}. 
+          content: ``,
+        },
+        {
+          role: 'user',
+          content: `${caveats}Here is the user brief: ${brief}. 
 Respond with just what would go in the function file and nothing else. No explanation or words, just the contents of the file. Make sure that the code is runnable if I were to execute it directly.`,
         },
       ],
diff --git a/test-workspace/fixtures/callOpenAI/input.ts b/test-workspace/fixtures/callOpenAI/input.ts
@@ -1,4 +1,12 @@
+import answerQuestionUsingOpenAI from '../../stacks/answerQuestionUsingOpenAI';
 
 const question = 'what is the meaning of life?';
 
-stack('answer my  question using OpenAI', {in: question, out: '42'});
+async function main() {
+  // console.log(await answerQuestionUsingOpenAI("question"));
+  const response = stack('answer my  question using OpenAI', {in: question, out: '42'});;
+  console.log;
+}
+
+main();
+
diff --git a/test-workspace/package-lock.json b/test-workspace/package-lock.json
diff --git a/test-workspace/package.json b/test-workspace/package.json
@@ -9,6 +9,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "dotenv": "^16.3.1",
     "robotjs": "^0.6.0",
     "ts-node": "^10.9.1",
     "typescript": "^5.3.2"
diff --git a/test-workspace/stacks/answerQuestionUsingOpenAI.ts b/test-workspace/stacks/answerQuestionUsingOpenAI.ts
@@ -0,0 +1,18 @@
+import OpenAI from 'openai';
+require('dotenv').config();
+
+const openai = new OpenAI({
+  apiKey: process.env.OPENAI_API_KEY
+});
+
+/**
+ * Brief: answer my  question using OpenAI
+ */
+export default async function answerQuestionUsingOpenAI(input: string): Promise<string> {
+  const response = await openai.chat.completions.create({
+    messages: [{ role: 'user', content: input }],
+    model: 'gpt-3.5-turbo',
+  });
+  
+  return response.choices[0].message.content;
+}
