adding support for the google palm api (#524)

* adding support for palm api

* docs: :bug: update modelData to new format

---------

Co-authored-by: Nate Sesti <sestinj@gmail.com>

3 files changed, 38 insertions, 0 deletions

diff --git a/extension/react-app/public/logos/google-palm.png b/extension/react-app/public/logos/google-palm.png
new file mode 100644
index 00000000..0f9dc345
--- /dev/null
+++ b/extension/react-app/public/logos/google-palm.png
diff --git a/extension/react-app/src/components/ModelSelect.tsx b/extension/react-app/src/components/ModelSelect.tsx
index 9dd3f489..6856a2cf 100644
--- a/extension/react-app/src/components/ModelSelect.tsx
+++ b/extension/react-app/src/components/ModelSelect.tsx
@@ -69,6 +69,14 @@ const MODEL_INFO: { title: string; class: string; args: any }[] = [
     },
   },
   {
+    title: "Google PaLM API",
+    class: "GooglePaLMAPI",
+    args: {
+      model: "chat-bison-001",
+      api_key: "<MAKERSUITE_API_KEY>",
+    },
+  },
+  {
     title: "LM Studio",
     class: "GGML",
     args: {
diff --git a/extension/react-app/src/util/modelData.ts b/extension/react-app/src/util/modelData.ts
index 615cbb79..91259446 100644
--- a/extension/react-app/src/util/modelData.ts
+++ b/extension/react-app/src/util/modelData.ts
@@ -387,6 +387,36 @@ After it's up and running, you can start using Continue.`,
     packages: [llama2FamilyPackage],
     collectInputFor: [contextLengthInput],
   },
+  palm: {
+    title: "Google PaLM API",
+    class: "GooglePaLMAPI",
+    description:
+      "Try out the Google PaLM API, which is currently in public preview, using an API key from Google Makersuite",
+    longDescription: `To get started with Google Makersuite, obtain your API key from [here](https://developers.generativeai.google/products/makersuite) and paste it below.
+> Note: Google's PaLM language models do not support streaming, so the response will appear all at once after a few seconds.`,
+    icon: "google-palm.png",
+    tags: [ModelProviderTag["Requires API Key"]],
+    collectInputFor: [
+      {
+        inputType: CollectInputType.text,
+        key: "api_key",
+        label: "API Key",
+        placeholder: "Enter your MakerSpace API key",
+        required: true,
+      },
+    ],
+    packages: [
+      {
+        title: "chat-bison-001",
+        description:
+          "Google PaLM's chat-bison-001 model, fine-tuned for chatting about code",
+        params: {
+          model: "chat-bison-001",
+          context_length: 8000,
+        },
+      },
+    ],
+  },
   hftgi: {
     title: "HuggingFace TGI",
     class: "HuggingFaceTGI",