diff --git a/src/config/index.ts b/src/config/index.ts index 32add43..63b94c3 100644 --- a/src/config/index.ts +++ b/src/config/index.ts @@ -113,6 +113,11 @@ export class Config implements ConfigInterface { "model", this.get("github-copilot.model")?.value || DEFAULT_GITHUB_COPILOT_MODEL, ); + else if (value === "mock") + this.setInternal( + "model", + this.get("ollama.model")?.value || DEFAULT_OLLAMA_MODEL, + ); else { console.log("Invalid API"); return false; diff --git a/src/llm/index.ts b/src/llm/index.ts index 1fcf10a..bddb5f2 100644 --- a/src/llm/index.ts +++ b/src/llm/index.ts @@ -340,7 +340,7 @@ export class MockLLMAPI extends LLMService { } else if (userPrompt.includes("apache2")) { return '{ "commands": ["systemctl status apache2"] }'; } else if (userPrompt.includes("Find sfsdfef")) { - return '{ "commands": ["grep \'sfsdfef\' *"] }'; + return '{ "commands": ["grep \'__LOZ_TEST_NO_OUTPUT_9f8b7c__\' README.md"] }'; } else { // Default: return a safe echo command without user input return '{ "commands": ["echo Mock LLM - command not recognized"] }';