v1.1.0 updates

CHANGELOG.md

@@ -1,3 +1,7 @@
+# 1.1.0 - December 7, 2023.
+
+- Updated model options
+
 # 1.0.1 - November 26, 2023.
 
 - Open-source release

README.md

@@ -43,10 +43,10 @@ The endpoint option specifies the API endpoint to call. `playwright-ai` will bro
 
 ### --model (-m) :: `String`
 
-_Default_: `gpt4`.
+_Default_: `gpt-4`.
 
 The LLM model to use to generate the tests.
-Available options: `gpt3`, `gpt4`, `claude`.
+Available options: `gpt-3`, `gpt-4`, `claude`.
 
 ### --tests (-t) :: `Number`
 
@@ -68,6 +68,11 @@ Shows CLI version.
 
 See the [CHANGELOG.md](CHANGELOG.md).
 
+## TODO
+
+- Support for other models
+- More complex test cases
+
 ## License
 
 [MIT](LICENSE)

lib/prompts.js

@@ -69,20 +69,21 @@ async function initialPrompt() {
     moreComponents: `This is great, let us proceed with an interactive test with several assertions for component that you have not generated a test for already. For example, components could be the navigation bar, the search bar, footer links, the login form. Include a short name of the component in a few words in the response wrapped in "$$" characters, use snake case format inside of test with a JavaScript comment. The HTML is the same as before for a website hosted at ${process.endpoint}:`,
   };
   let model = process.model;
-  if (model === "gpt3") {
+  if (model === "gpt-3") {
     model = "gpt-3.5-turbo-0613";
   }
 
-  api = new ChatGPTAPI({
-    apiKey: process.env["OPENAI_API_KEY"],
-    completionParams: {
-      model: model,
-    },
-  });
   if (model === "claude") {
     api = new Claude({
       apiKey: process.env["ANTHROPIC_API_KEY"],
     });
+  } else {
+    api = new ChatGPTAPI({
+      apiKey: process.env["OPENAI_API_KEY"],
+      completionParams: {
+        model: model || "gpt-4",
+      },
+    });
   }
 
   await askAi(PROMPTS.start, true);