diff options
Diffstat (limited to 'extension')
-rw-r--r-- | extension/react-app/public/logos/google-palm.png | bin | 0 -> 2608 bytes | |||
-rw-r--r-- | extension/react-app/src/components/ModelSelect.tsx | 8 | ||||
-rw-r--r-- | extension/react-app/src/util/modelData.ts | 30 |
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 Binary files differnew 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", |