commit message:

Loz: Update API and model settings
==============================

In this commit, we updated the `api` and `model` settings in the `Loz` class to allow for different OpenAI models to be used. We also added a new method `setAPI(api: string, model?: string): Promise<void>` to set these settings programmatically. Additionally, we updated the `initLLMfromConfig()` method to use the new `config` object and added a new test case for the `Loz.ollama` property.

The `Loz` class now has a `checkAPI()` method that returns the currently set API, which can be used to verify the correctness of the settings. We also added a new test case for the `completeUserPrompt()` method to ensure it returns the correct result given a user prompt.

These changes improve the functionality and reliability of the `Loz` class, making it easier to use and maintain in the long run.

Generated by llama2

src/loz.ts

@@ -40,7 +40,7 @@ export class Loz {
   config: Config = new Config();
   git: Git = new Git();
 
-  constructor(llmAPI?: string) {
+  constructor() {
     this.defaultSettings = {
       model: DEFAULT_OPENAI_MODEL,
       prompt: "",
@@ -283,6 +283,19 @@ export class Loz {
     }
   }
 
+  public async setAPI(api: string, model?: string): Promise<void> {
+    if (api === "ollama" || api === "openai") {
+      this.config.set("api", api);
+    }
+
+    if (model) {
+      this.config.set("model", model);
+    }
+
+    this.config.save();
+    await this.initLLMfromConfig();
+  }
+
   public async handlePrompt(prompt: string): Promise<void> {
     const systemPrompt =
       "Decide if the following prompt can be translated into Linux commands. " +

test/a.loz.test.ts

@@ -46,11 +46,13 @@ describe("Test OpenAI API", () => {
 if (GITHUB_ACTIONS === false) {
   describe("Loz.ollama", () => {
     it("should return true", async () => {
-      let loz = new Loz("ollama");
+      let loz = new Loz();
       await loz.init();
+      await loz.setAPI("ollama", "llama2");
 
+      expect((loz as any).checkAPI()).to.equal("ollama");
       const completion = await loz.completeUserPrompt("1+1=");
-      expect(completion.content).to.equal("2");
+      expect(completion.content).contains("2");
     });
   });
 }