From f09150617ed2454f3074bcf93f53aae5ae637d40 Mon Sep 17 00:00:00 2001
From: Nate Sesti <33237525+sestinj@users.noreply.github.com>
Date: Mon, 9 Oct 2023 18:37:27 -0700
Subject: Preview (#541)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* Strong typing (#533)

* refactor: :recycle: get rid of continuedev.src.continuedev structure

* refactor: :recycle: switching back to server folder

* feat: :sparkles: make config.py imports shorter

* feat: :bookmark: publish as pre-release vscode extension

* refactor: :recycle: refactor and add more completion params to ui

* build: :building_construction: download from preview S3

* fix: :bug: fix paths

* fix: :green_heart: package:pre-release

* ci: :green_heart: more time for tests

* fix: :green_heart: fix build scripts

* fix: :bug: fix import in run.py

* fix: :bookmark: update version to try again

* ci: 💚 Update package.json version [skip ci]

* refactor: :fire: don't check for old extensions version

* fix: :bug: small bug fixes

* fix: :bug: fix config.py import paths

* ci: 💚 Update package.json version [skip ci]

* ci: :green_heart: platform-specific builds test #1

* feat: :green_heart: ship with binary

* fix: :green_heart: fix copy statement to include.exe for windows

* fix: :green_heart: cd extension before packaging

* chore: :loud_sound: count tokens generated

* fix: :green_heart: remove npm_config_arch

* fix: :green_heart: publish as pre-release!

* chore: :bookmark: update version

* perf: :green_heart: hardcode distro paths

* fix: :bug: fix yaml syntax error

* chore: :bookmark: update version

* fix: :green_heart: update permissions and version

* feat: :bug: kill old server if needed

* feat: :lipstick: update marketplace icon for pre-release

* ci: 💚 Update package.json version [skip ci]

* feat: :sparkles: auto-reload for config.py

* feat: :wrench: update default config.py imports

* feat: :sparkles: codelens in config.py

* feat: :sparkles: select model param count from UI

* ci: 💚 Update package.json version [skip ci]

* feat: :sparkles: more model options, ollama error handling

* perf: :zap: don't show server loading immediately

* fix: :bug: fixing small UI details

* ci: 💚 Update package.json version [skip ci]

* feat: :rocket: headers param on LLM class

* fix: :bug: fix headers for openai.;y

* feat: :sparkles: highlight code on cmd+shift+L

* ci: 💚 Update package.json version [skip ci]

* feat: :lipstick: sticky top bar in gui.tsx

* fix: :loud_sound: websocket logging and horizontal scrollbar

* ci: 💚 Update package.json version [skip ci]

* feat: :sparkles: allow AzureOpenAI Service through GGML

* ci: 💚 Update package.json version [skip ci]

* fix: :bug: fix automigration

* ci: 💚 Update package.json version [skip ci]

* ci: :green_heart: upload binaries in ci, download apple silicon

* chore: :fire: remove notes

* fix: :green_heart: use curl to download binary

* fix: :green_heart: set permissions on apple silicon binary

* fix: :green_heart: testing

* fix: :green_heart: cleanup file

* fix: :green_heart: fix preview.yaml

* fix: :green_heart: only upload once per binary

* fix: :green_heart: install rosetta

* ci: :green_heart: download binary after tests

* ci: 💚 Update package.json version [skip ci]

* ci: :green_heart: prepare ci for merge to main

---------

Co-authored-by: GitHub Action <action@github.com>
---
 extension/react-app/src/pages/modelconfig.tsx | 63 ++++++++++++++++++---------
 1 file changed, 42 insertions(+), 21 deletions(-)

(limited to 'extension/react-app/src/pages/modelconfig.tsx')

diff --git a/extension/react-app/src/pages/modelconfig.tsx b/extension/react-app/src/pages/modelconfig.tsx
index 97e2d76c..00d9d9bf 100644
--- a/extension/react-app/src/pages/modelconfig.tsx
+++ b/extension/react-app/src/pages/modelconfig.tsx
@@ -3,7 +3,7 @@ import ModelCard from "../components/ModelCard";
 import styled from "styled-components";
 import { ArrowLeftIcon } from "@heroicons/react/24/outline";
 import {
-  TextInput,
+  Input,
   defaultBorderRadius,
   lightGray,
   vscBackground,
@@ -22,6 +22,7 @@ import { RootStore } from "../redux/store";
 import StyledMarkdownPreview from "../components/StyledMarkdownPreview";
 import { getFontSize } from "../util";
 import { FormProvider, useForm } from "react-hook-form";
+import _ from "lodash";
 
 const GridDiv = styled.div`
   display: grid;
@@ -151,22 +152,28 @@ function ModelConfig() {
             <>
               <h3 className="mb-2">Enter required parameters</h3>
 
-              {modelInfo?.collectInputFor?.map((d) => {
-                return (
-                  <div>
-                    <label htmlFor={d.key}>{d.key}</label>
-                    <TextInput
-                      id={d.key}
-                      className="border-2 border-gray-200 rounded-md p-2 m-2"
-                      placeholder={d.key}
-                      defaultValue={d.defaultValue}
-                      {...formMethods.register(d.key, {
-                        required: true,
-                      })}
-                    />
-                  </div>
-                );
-              })}
+              {modelInfo?.collectInputFor
+                ?.filter((d) => d.required)
+                .map((d) => {
+                  return (
+                    <div>
+                      <label htmlFor={d.key}>{d.key}</label>
+                      <Input
+                        type={d.inputType}
+                        id={d.key}
+                        className="border-2 border-gray-200 rounded-md p-2 m-2"
+                        placeholder={d.key}
+                        defaultValue={d.defaultValue}
+                        min={d.min}
+                        max={d.max}
+                        step={d.step}
+                        {...formMethods.register(d.key, {
+                          required: true,
+                        })}
+                      />
+                    </div>
+                  );
+                })}
             </>
           )}
 
@@ -182,11 +189,15 @@ function ModelConfig() {
                 return (
                   <div>
                     <label htmlFor={d.key}>{d.key}</label>
-                    <TextInput
+                    <Input
+                      type={d.inputType}
                       id={d.key}
                       className="border-2 border-gray-200 rounded-md p-2 m-2"
                       placeholder={d.key}
                       defaultValue={d.defaultValue}
+                      min={d.min}
+                      max={d.max}
+                      step={d.step}
                       {...formMethods.register(d.key, {
                         required: false,
                       })}
@@ -209,19 +220,29 @@ function ModelConfig() {
                 tags={pkg.tags}
                 refUrl={pkg.refUrl}
                 icon={pkg.icon || modelInfo.icon}
-                onClick={(e) => {
+                dimensions={pkg.dimensions}
+                onClick={(e, dimensionChoices) => {
                   if (disableModelCards()) return;
                   const formParams: any = {};
                   for (const d of modelInfo.collectInputFor || []) {
                     formParams[d.key] =
                       d.inputType === "text"
                         ? formMethods.watch(d.key)
-                        : parseInt(formMethods.watch(d.key));
+                        : parseFloat(formMethods.watch(d.key));
                   }
 
                   client?.addModelForRole("*", modelInfo.class, {
                     ...pkg.params,
                     ...modelInfo.params,
+                    ..._.merge(
+                      {},
+                      ...(pkg.dimensions?.map((dimension, i) => {
+                        if (!dimensionChoices?.[i]) return {};
+                        return {
+                          ...dimension.options[dimensionChoices[i]],
+                        };
+                      }) || [])
+                    ),
                     ...formParams,
                   });
                   navigate("/");
@@ -239,7 +260,7 @@ function ModelConfig() {
                 formParams[d.key] =
                   d.inputType === "text"
                     ? formMethods.watch(d.key)
-                    : parseInt(formMethods.watch(d.key));
+                    : parseFloat(formMethods.watch(d.key));
               }
 
               client?.addModelForRole("*", modelInfo.class, {
-- 
cgit v1.2.3-70-g09d2

