From 5eec484dc79bb56dabf9a56af0dbe6bc95227d39 Mon Sep 17 00:00:00 2001
From: Nate Sesti <33237525+sestinj@users.noreply.github.com>
Date: Tue, 22 Aug 2023 13:12:58 -0700
Subject: Config UI (#399)

* feat: :sparkles: UI for config!

* feat: :sparkles: (latent) edit models in settings
---
 .../react-app/src/hooks/AbstractContinueGUIClientProtocol.ts | 10 ++++++++++
 extension/react-app/src/hooks/ContinueGUIClientProtocol.ts   | 12 ++++++++++++
 2 files changed, 22 insertions(+)

(limited to 'extension/react-app/src/hooks')

diff --git a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
index c9e7def2..f8c11527 100644
--- a/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/AbstractContinueGUIClientProtocol.ts
@@ -37,6 +37,16 @@ abstract class AbstractContinueGUIClientProtocol {
 
   abstract editStepAtIndex(userInput: string, index: number): void;
 
+  abstract setSystemMessage(message: string): void;
+
+  abstract setTemperature(temperature: number): void;
+
+  abstract setModelForRole(
+    role: string,
+    model_class: string,
+    model: string
+  ): void;
+
   abstract saveContextGroup(title: string, contextItems: ContextItem[]): void;
 
   abstract selectContextGroup(id: string): void;
diff --git a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
index b3ac2570..ce9b2a0a 100644
--- a/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
+++ b/extension/react-app/src/hooks/ContinueGUIClientProtocol.ts
@@ -133,6 +133,18 @@ class ContinueGUIClientProtocol extends AbstractContinueGUIClientProtocol {
     });
   }
 
+  setSystemMessage(message: string): void {
+    this.messenger?.send("set_system_message", { message });
+  }
+
+  setTemperature(temperature: number): void {
+    this.messenger?.send("set_temperature", { temperature });
+  }
+
+  setModelForRole(role: string, model_class: string, model: any): void {
+    this.messenger?.send("set_model_for_role", { role, model, model_class });
+  }
+
   saveContextGroup(title: string, contextItems: ContextItem[]): void {
     this.messenger?.send("save_context_group", {
       context_items: contextItems,
-- 
cgit v1.2.3-70-g09d2