diff --git a/src/renderer/components/Settings/AIProviderModal.tsx b/src/renderer/components/Settings/AIProviderModal.tsx
new file mode 100644
index 0000000000000000000000000000000000000000..484518c02579f5e317674db410b865c5e649b2d9
--- /dev/null
+++ b/src/renderer/components/Settings/AIProviderModal.tsx
@@ -0,0 +1,165 @@
+import * as React from 'react';
+import {
+ Button,
+ Modal,
+ ModalDialog,
+ FormControl,
+ FormLabel,
+ Input,
+ Box,
+ Typography,
+ IconButton,
+} from '@mui/joy';
+import useSWR from 'swr';
+
+import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
+import { EyeIcon, EyeOffIcon } from 'lucide-react';
+const fetcher = (url: string) => fetch(url).then((res) => res.json());
+
+interface Provider {
+ name: string;
+ keyName: string;
+}
+
+export default function AIProviderModal({
+ dialogOpen,
+ setDialogOpen,
+ selectedProvider,
+}: {
+ dialogOpen: boolean;
+ setDialogOpen: (open: boolean) => void;
+ selectedProvider: Provider | null;
+}) {
+ const [showApiKey, setShowApiKey] = React.useState(false);
+ const { data: apiKey, mutate: mutateApiKey } = useSWR(
+ chatAPI.Endpoints.Config.Get(selectedProvider?.keyName),
+ fetcher,
+ );
+
+ const saveProvider = async (provider: Provider, token: string) => {
+ await fetch(chatAPI.Endpoints.Config.Set(provider.keyName, token));
+ mutateApiKey();
+ };
+
+ const handleSubmit = (e: React.FormEvent) => {
+ e.preventDefault();
+ const data = new FormData(e.target as HTMLFormElement);
+ const apiKey = data.get('apiKey') as string;
+ alert(JSON.stringify(Object.fromEntries(data.entries())));
+ saveProvider(selectedProvider!, apiKey);
+ setDialogOpen(false);
+ };
+
+ if (!selectedProvider) {
+ return null;
+ }
+
+ return (
+ <Modal open={dialogOpen} onClose={() => setDialogOpen(false)}>
+ <ModalDialog
+ aria-labelledby="connect-dialog-title"
+ sx={{
+ position: 'absolute',
+ top: '50%',
+ left: '50%',
+ transform: 'translate(-50%, -50%)',
+ maxWidth: 400,
+ width: '90%',
+ }}
+ >
+ <Typography id="connect-dialog-title" component="h2">
+ Connect to {selectedProvider?.name}
+ </Typography>
+ <form onSubmit={handleSubmit}>
+ {selectedProvider?.name === 'Custom API' ? (
+ <>
+ <FormControl sx={{ mt: 2 }}>
+ <FormLabel>API Name</FormLabel>
+ <Input name="customApiName" />
+ </FormControl>
+ <FormControl sx={{ mt: 2 }}>
+ <FormLabel>Base URL</FormLabel>
+ <Input name="customBaseURL" />
+ </FormControl>
+ <FormControl sx={{ mt: 2 }}>
+ <FormLabel>API Key</FormLabel>
+ <Input name="customApiKey" type="password" />
+ </FormControl>
+ <FormControl sx={{ mt: 2 }}>
+ <FormLabel>Model Name</FormLabel>
+ <Input name="customModelName" />
+ </FormControl>
+ </>
+ ) : (
+ <FormControl sx={{ mt: 2 }}>
+ <FormLabel>{selectedProvider?.name} API Key</FormLabel>
+ <Input
+ endDecorator={
+ <IconButton
+ onClick={() => {
+ setShowApiKey(!showApiKey);
+ }}
+ >
+ {showApiKey ? <EyeOffIcon /> : <EyeIcon />}
+ </IconButton>
+ }
+ name="apiKey"
+ type={showApiKey ? 'text' : 'password'}
+ defaultValue={apiKey}
+ />
+ </FormControl>
+ )}
+ {/* Conditional help steps */}
+ {selectedProvider?.name === 'OpenAI' && (
+ <>
+ <Typography level="title-md" mt={2}>
+ Steps to get an OpenAI API Key:
+ </Typography>
+ <ol>
+ <li>
+ Visit{' '}
+ <a
+ href="https://platform.openai.com/account/api-keys"
+ target="_blank"
+ rel="noreferrer"
+ >
+ OpenAI API website
+ </a>
+ </li>
+ <li>Log in to your OpenAI account.</li>
+ <li>Create a new API key and copy it.</li>
+ </ol>
+ </>
+ )}
+ {selectedProvider?.name === 'Anthropic' && (
+ <>
+ <Typography level="title-md" mt={2}>
+ Steps to get a Anthropic API Key:
+ </Typography>
+ <ol>
+ <li>
+ Visit{' '}
+ <a
+ href="https://console.anthropic.com/settings/keys"
+ target="_blank"
+ rel="noreferrer"
+ >
+ Anthropic API Keys Console
+ </a>
+ </li>
+ <li>Log in/Create an account.</li>
+ <li>Follow instructions to generate an API key.</li>
+ </ol>
+ </>
+ )}
+ <Box
+ sx={{ display: 'flex', justifyContent: 'flex-end', gap: 1, mt: 2 }}
+ >
+ <Button onClick={() => setDialogOpen(false)}>Cancel</Button>
+ <Button type="submit">Save</Button>
+ </Box>
+ </form>
+ </ModalDialog>
+ </Modal>
+ );
+}
diff --git a/src/renderer/components/Settings/AIProvidersSettings.tsx b/src/renderer/components/Settings/AIProvidersSettings.tsx
index f52d351a960ae4abbfac5881f2f32fa0a0590961..3e4d00c453eaa7c4274d4b2550696161bbc41d19 100644
--- a/src/renderer/components/Settings/AIProvidersSettings.tsx
+++ b/src/renderer/components/Settings/AIProvidersSettings.tsx
@@ -1,305 +1,92 @@
import * as React from 'react';
-import {
- Button,
- Modal,
- ModalDialog,
- FormControl,
- FormLabel,
- Input,
- List,
- ListItem,
- Box,
- Typography
-} from '@mui/joy';
-import * as chatAPI from 'renderer/lib/transformerlab-api-sdk';
-import useSWR from 'swr';
+import { Button, List, ListItem, Box, Typography, Chip } from '@mui/joy';
+import { ChevronLeftIcon } from 'lucide-react';
+import AIProviderModal from './AIProviderModal';
-const fetcher = (url: string) => fetch(url).then((res) => res.json());
-
-interface Provider {
- name: string;
- keyName: string;
- setKeyEndpoint: () => string;
- checkKeyEndpoint: () => string;
-}
-
-const providers: Provider[] = [
- {
- name: 'OpenAI',
- keyName: 'OPENAI_API_KEY',
- setKeyEndpoint: () => chatAPI.Endpoints.Models.SetOpenAIKey(),
- checkKeyEndpoint: () => chatAPI.Endpoints.Models.CheckOpenAIAPIKey(),
- },
- {
- name: 'Anthropic',
- keyName: 'ANTHROPIC_API_KEY',
- setKeyEndpoint: () => chatAPI.Endpoints.Models.SetAnthropicKey(),
- checkKeyEndpoint: () => chatAPI.Endpoints.Models.CheckAnthropicAPIKey(),
- },
- {
- name: 'Custom API',
- keyName: 'CUSTOM_MODEL_API_KEY',
- setKeyEndpoint: () => chatAPI.Endpoints.Models.SetCustomAPIKey(),
- checkKeyEndpoint: () => chatAPI.Endpoints.Models.CheckCustomAPIKey(),
- }
+const providers = [
+ {
+ name: 'OpenAI',
+ keyName: 'OPENAI_API_KEY',
+ },
+ {
+ name: 'Anthropic',
+ keyName: 'ANTHROPIC_API_KEY',
+ },
+ {
+ name: 'Custom API',
+ keyName: 'CUSTOM_MODEL_API_KEY',
+ },
];
-interface AIProvidersSettingsProps {
- onBack?: () => void;
-}
-
-export default function AIProvidersSettings({ onBack }: AIProvidersSettingsProps) {
- const { data: openaiApiKey, mutate: mutateOpenAI } = useSWR(
- chatAPI.Endpoints.Config.Get('OPENAI_API_KEY'),
- fetcher
- );
- const { data: claudeApiKey, mutate: mutateClaude } = useSWR(
- chatAPI.Endpoints.Config.Get('ANTHROPIC_API_KEY'),
- fetcher
- );
- const { data: customAPIStatus, mutate: mutateCustom } = useSWR(
- chatAPI.Endpoints.Config.Get('CUSTOM_MODEL_API_KEY'),
- fetcher
- );
-
- const getProviderStatus = (provider: Provider) => {
- if (provider.name === 'OpenAI') return openaiApiKey;
- if (provider.name === 'Anthropic') return claudeApiKey;
- if (provider.name === 'Custom API') return customAPIStatus;
- return null;
- };
-
- const setProviderStatus = async (provider: Provider, token: string) => {
- await fetch(chatAPI.Endpoints.Config.Set(provider.keyName, token));
- await fetch(provider.setKeyEndpoint());
- const response = await fetch(provider.checkKeyEndpoint());
- const result = await response.json();
- return result.message === 'OK';
- };
-
- const [dialogOpen, setDialogOpen] = React.useState(false);
- const [selectedProvider, setSelectedProvider] = React.useState<Provider | null>(null);
- const [apiKey, setApiKey] = React.useState('');
- const [hoveredProvider, setHoveredProvider] = React.useState<string | null>(null);
- // States for custom API additional fields
- const [customApiName, setCustomApiName] = React.useState('');
- const [customBaseURL, setCustomBaseURL] = React.useState('');
- const [customApiKey, setCustomApiKey] = React.useState('');
- const [customModelName, setCustomModelName] = React.useState('');
-
- const handleConnectClick = (provider: Provider) => {
- setSelectedProvider(provider);
- if (provider.name === 'Custom API') {
- setCustomApiName('');
- setCustomBaseURL('');
- setCustomApiKey('');
- setCustomModelName('');
- } else {
- setApiKey('');
- }
- setDialogOpen(true);
- };
+export default function AIProvidersSettings({
+ onBack,
+}: {
+ onBack?: () => void;
+}) {
+ const [dialogOpen, setDialogOpen] = React.useState(false);
+ const [selectedProvider, setSelectedProvider] = React.useState(null);
- const handleDisconnectClick = async (provider: Provider) => {
- await fetch(chatAPI.Endpoints.Config.Set(provider.keyName, ''));
- if (provider.name === 'OpenAI') {
- mutateOpenAI();
- } else if (provider.name === 'Anthropic') {
- mutateClaude();
- } else if (provider.name === 'Custom API') {
- mutateCustom();
- }
- };
+ const handleConnectClick = (provider) => {
+ setSelectedProvider(provider);
+ setDialogOpen(true);
+ };
- const handleSave = async () => {
- if (selectedProvider) {
- let token = '';
- if (selectedProvider.name === 'Custom API') {
- const customObj = {
- apiName: customApiName,
- baseURL: customBaseURL,
- apiKey: customApiKey,
- modelName: customModelName
- };
- token = JSON.stringify(customObj);
- } else if (apiKey) {
- token = apiKey;
- }
- if (token) {
- const success = await setProviderStatus(selectedProvider, token);
- if (success) {
- alert(`Successfully connected to ${selectedProvider.name}`);
- if (selectedProvider.name === 'OpenAI') {
- mutateOpenAI();
- } else if (selectedProvider.name === 'Anthropic') {
- mutateClaude();
- } else if (selectedProvider.name === 'Custom API') {
- mutateCustom();
- }
- } else {
- alert(`Failed to connect to ${selectedProvider.name}`);
- }
- }
- }
- setDialogOpen(false);
- };
+ return (
+ <Box sx={{ p: 2 }}>
+ <Button
+ onClick={onBack}
+ startDecorator={<ChevronLeftIcon />}
+ variant="plain"
+ >
+ Back to Settings
+ </Button>
+ <Typography level="h3" mb={2} mt={2}>
+ AI Providers
+ </Typography>
+ <List sx={{ gap: 1 }}>
+ {providers.map((provider) => {
+ const status = false;
+ const isConnected = status && status !== '';
+ return (
+ <ListItem
+ variant="soft"
+ key={provider.name}
+ sx={{
+ display: 'flex',
+ justifyContent: 'space-between',
+ alignItems: 'center',
+ p: 2,
+ paddingLeft: 3,
+ borderRadius: '8px',
+ }}
+ >
+ <Typography level="title-md">{provider.name}</Typography>
+ <Box>
+ {isConnected && (
+ <>
+ <Chip variant="solid" color="success">
+ Connected
+ </Chip>
+ </>
+ )}
- return (
- <Box sx={{ p: 2 }}>
- <Button onClick={onBack}>Back to Settings</Button>
- <Typography level="h3" mb={2}>
- AI Providers
- </Typography>
- <List sx={{ gap: 1 }}>
- {providers.map((provider) => {
- const status = getProviderStatus(provider);
- const isConnected = status && status !== '';
- return (
- <ListItem
- key={provider.name}
- sx={{
- display: 'flex',
- justifyContent: 'space-between',
- alignItems: 'center',
- p: 1,
- borderRadius: '8px',
- bgcolor: 'neutral.softBg'
- }}
- >
- <Typography>{provider.name}</Typography>
- {isConnected ? (
- <Box
- onMouseEnter={() => setHoveredProvider(provider.name)}
- onMouseLeave={() => setHoveredProvider(null)}
- onClick={() => handleDisconnectClick(provider)}
- sx={{
- cursor: 'pointer',
- border: '1px solid',
- borderColor: 'neutral.outlinedBorder',
- borderRadius: '4px',
- px: 1,
- py: 0.5,
- fontSize: '0.875rem',
- color: 'success.600'
- }}
- >
- {hoveredProvider === provider.name ? 'Disconnect' : 'Connected'}
- </Box>
- ) : (
- <Button variant="soft" onClick={() => handleConnectClick(provider)}>
- Connect
- </Button>
- )}
- </ListItem>
- );
- })}
- </List>
- {/* API Key Modal */}
- <Modal open={dialogOpen} onClose={() => setDialogOpen(false)}>
- <ModalDialog
- layout="stack"
- aria-labelledby="connect-dialog-title"
- sx={{
- position: 'absolute',
- top: '50%',
- left: '50%',
- transform: 'translate(-50%, -50%)',
- maxWidth: 400,
- width: '90%'
- }}
+ <Button
+ variant="outlined"
+ onClick={() => handleConnectClick(provider)}
+ sx={{ ml: 1 }}
>
- <Typography id="connect-dialog-title" component="h2">
- Connect to {selectedProvider?.name}
- </Typography>
- {selectedProvider?.name === 'Custom API' ? (
- <>
- <FormControl sx={{ mt: 2 }}>
- <FormLabel>API Name</FormLabel>
- <Input
- value={customApiName}
- onChange={(e) => setCustomApiName(e.target.value)}
- />
- </FormControl>
- <FormControl sx={{ mt: 2 }}>
- <FormLabel>Base URL</FormLabel>
- <Input
- value={customBaseURL}
- onChange={(e) => setCustomBaseURL(e.target.value)}
- />
- </FormControl>
- <FormControl sx={{ mt: 2 }}>
- <FormLabel>API Key</FormLabel>
- <Input
- type="password"
- value={customApiKey}
- onChange={(e) => setCustomApiKey(e.target.value)}
- />
- </FormControl>
- <FormControl sx={{ mt: 2 }}>
- <FormLabel>Model Name</FormLabel>
- <Input
- value={customModelName}
- onChange={(e) => setCustomModelName(e.target.value)}
- />
- </FormControl>
- </>
- ) : (
- <FormControl sx={{ mt: 2 }}>
- <FormLabel>{selectedProvider?.name} API Key</FormLabel>
- <Input
- type="password"
- value={apiKey}
- onChange={(e) => setApiKey(e.target.value)}
- />
- </FormControl>
- )}
- {/* Conditional help steps */}
- {selectedProvider?.name === 'OpenAI' && (
- <Box sx={{ mt: 2 }}>
- <Typography level="body2">
- Steps to get an OpenAI API Key:
- </Typography>
- <ol>
- <li>
- Visit{' '}
- <a
- href="https://platform.openai.com/account/api-keys"
- target="_blank"
- rel="noreferrer"
- >
- OpenAI API website
- </a>
- </li>
- <li>Log in to your OpenAI account.</li>
- <li>Create a new API key and copy it.</li>
- </ol>
- </Box>
- )}
- {selectedProvider?.name === 'Anthropic' && (
- <Box sx={{ mt: 2 }}>
- <Typography level="body2">
- Steps to get a Anthropic API Key:
- </Typography>
- <ol>
- <li>Visit{' '}
- <a
- href="https://console.anthropic.com/settings/keys"
- target="_blank"
- rel="noreferrer"
- >
- Anthropic API Keys Console
- </a></li>
- <li>Log in/Create an account.</li>
- <li>Follow instructions to generate an API key.</li>
- </ol>
- </Box>
- )}
- <Box sx={{ display: 'flex', justifyContent: 'flex-end', gap: 1, mt: 2 }}>
- <Button onClick={() => setDialogOpen(false)}>Cancel</Button>
- <Button onClick={handleSave}>Save</Button>
- </Box>
- </ModalDialog>
- </Modal>
- </Box>
- );
+ Set API Key
+ </Button>
+ </Box>
+ </ListItem>
+ );
+ })}
+ </List>
+ <AIProviderModal
+ dialogOpen={dialogOpen}
+ setDialogOpen={setDialogOpen}
+ selectedProvider={selectedProvider}
+ />
+ </Box>
+ );
}
diff --git a/src/renderer/lib/transformerlab-api-sdk.ts b/src/renderer/lib/transformerlab-api-sdk.ts
index 819a33154002cbc8cfc59cf243ff344c369616f6..488601f514caa4ce551be4865f07afc641131323 100644
--- a/src/renderer/lib/transformerlab-api-sdk.ts
+++ b/src/renderer/lib/transformerlab-api-sdk.ts
@@ -53,7 +53,7 @@ export async function sendAndReceive(
temperature: number,
maxTokens: number,
topP: number,
- systemMessage: string
+ systemMessage: string,
) {
const shortModelName = currentModel.split('/').slice(-1)[0];
@@ -82,7 +82,7 @@ export async function sendAndReceive(
accept: 'application/json',
},
body: JSON.stringify(data),
- }
+ },
);
result = await response.json();
} catch (error) {
@@ -112,7 +112,7 @@ export async function sendAndReceiveStreaming(
freqencyPenalty: number,
systemMessage: string,
stopString = null,
- image?: string
+ image?: string,
) {
let shortModelName = currentModel.split('/').slice(-1)[0];
@@ -231,7 +231,7 @@ export async function sendAndReceiveStreaming(
document.getElementById('resultText').innerText = finalResult;
setTimeout(
() => document.getElementById('endofchat')?.scrollIntoView(),
- 100
+ 100,
);
}
}
@@ -270,7 +270,7 @@ export async function sendCompletion(
useLongModelName = true,
stopString = null,
targetElementForStreaming,
- logprobs = false
+ logprobs = false,
) {
console.log('sent completion request');
let model = '';
@@ -395,7 +395,7 @@ export async function sendCompletion(
// document.getElementById('completion-textarea').value = finalResult;
setTimeout(
() => document.getElementById('endofchat')?.scrollIntoView(),
- 100
+ 100,
);
}
}
@@ -435,7 +435,7 @@ export async function sendCompletionReactWay(
useLongModelName = true,
stopString = null,
updateFunction,
- logprobs = false
+ logprobs = false,
) {
console.log('sent completion request');
let model = '';
@@ -568,7 +568,7 @@ export async function sendCompletionReactWay(
// document.getElementById('completion-textarea').value = finalResult;
setTimeout(
() => document.getElementById('endofchat')?.scrollIntoView(),
- 100
+ 100,
);
}
}
@@ -607,7 +607,7 @@ export async function sendBatchedCompletion(
topP: number = 1.0,
useLongModelName = true,
stopString = null,
- repeatTimes = 1
+ repeatTimes = 1,
) {
let model = '';
if (useLongModelName) {
@@ -668,7 +668,7 @@ export async function sendBatchedChat(
maxTokens: number = 256,
topP: number = 1.0,
useLongModelName = true,
- stopString = null
+ stopString = null,
) {
let model = '';
if (useLongModelName) {
@@ -752,7 +752,7 @@ export async function sendBatchedChat(
export async function callTool(
function_name: String,
- function_args: Object = {}
+ function_args: Object = {},
) {
const arg_string = JSON.stringify(function_args);
console.log(`Calling Function: ${function_name}`);
@@ -772,12 +772,12 @@ export async function getAvailableModels() {
export async function downloadModelFromHuggingFace(
modelName: string,
- job_id = null
+ job_id = null,
) {
console.log(encodeURIComponent(modelName));
let requestString = `${API_URL()}model/download_from_huggingface?model=${encodeURIComponent(
- modelName
+ modelName,
)}`;
if (job_id) {
requestString += `&job_id=${job_id}`;
@@ -801,12 +801,12 @@ export async function downloadModelFromHuggingFace(
export async function downloadModelFromGallery(
galleryID: string,
- job_id = null
+ job_id = null,
) {
console.log(encodeURIComponent(galleryID));
let requestString = `${API_URL()}model/download_model_from_gallery?gallery_id=${encodeURIComponent(
- galleryID
+ galleryID,
)}`;
if (job_id) {
requestString += `&job_id=${job_id}`;
@@ -1010,7 +1010,7 @@ function convertSlashInUrl(url: string) {
Endpoints.Workflows = {
List: () => API_URL() + 'workflows/list',
-}
+};
Endpoints.Dataset = {
Gallery: () => API_URL() + 'data/gallery',
@@ -1019,7 +1019,7 @@ Endpoints.Dataset = {
datasetId: string,
split: string = '',
offset: number = 0,
- limit: number = 10
+ limit: number = 10,
) =>
API_URL() +
'data/preview?dataset_id=' +
@@ -1034,7 +1034,7 @@ Endpoints.Dataset = {
datasetId: string,
template: string,
offset: number,
- limit: number
+ limit: number,
) =>
API_URL() +
'data/preview_with_template?dataset_id=' +
@@ -1050,7 +1050,8 @@ Endpoints.Dataset = {
Create: (datasetId: string) => API_URL() + 'data/new?dataset_id=' + datasetId,
Download: (datasetId: string) =>
API_URL() + 'data/download?dataset_id=' + datasetId,
- LocalList: (generated: boolean = true) => API_URL() + 'data/list?generated=' + generated,
+ LocalList: (generated: boolean = true) =>
+ API_URL() + 'data/list?generated=' + generated,
GeneratedList: () => API_URL() + 'data/generated_datasets_list',
FileUpload: (datasetId: string) =>
API_URL() + 'data/fileupload?dataset_id=' + datasetId,
@@ -1065,7 +1066,7 @@ Endpoints.Models = {
modelId: string,
modelName: string,
organizationName?: string,
- model_card_data?: object
+ model_card_data?: object,
) =>
API_URL() +
'model/upload_to_huggingface?model_id=' +
@@ -1101,12 +1102,6 @@ Endpoints.Models = {
Delete: (modelId: string) => API_URL() + 'model/delete?model_id=' + modelId,
wandbLogin: () => API_URL() + 'model/login_to_wandb',
testWandbLogin: () => API_URL() + 'model/test_wandb_login',
- SetOpenAIKey: () => API_URL() + 'model/set_openai_api_key',
- SetAnthropicKey: () => API_URL() + 'model/set_anthropic_api_key',
- CheckOpenAIAPIKey: () => API_URL() + 'model/check_openai_api_key',
- CheckAnthropicAPIKey: () => API_URL() + 'model/check_anthropic_api_key',
- SetCustomAPIKey: () => API_URL() + 'model/set_custom_api_key',
- CheckCustomAPIKey: () => API_URL() + 'model/check_custom_api_key',
};
Endpoints.Plugins = {
@@ -1125,7 +1120,11 @@ Endpoints.Config = {
Endpoints.Documents = {
List: (experimentId: string, currentFolder: string = '') =>
- API_URL() + 'experiment/' + experimentId + '/documents/list?folder=' + currentFolder,
+ API_URL() +
+ 'experiment/' +
+ experimentId +
+ '/documents/list?folder=' +
+ currentFolder,
Open: (experimentId: string, document_name: string, folder: string) =>
API_URL() +
'experiment/' +
@@ -1135,7 +1134,11 @@ Endpoints.Documents = {
'?folder=' +
folder,
Upload: (experimentId: string, currentFolder: string = '') =>
- API_URL() + 'experiment/' + experimentId + '/documents/upload?folder=' + currentFolder,
+ API_URL() +
+ 'experiment/' +
+ experimentId +
+ '/documents/upload?folder=' +
+ currentFolder,
Delete: (experimentId: string, document_name: string) =>
API_URL() +
'experiment/' +
@@ -1156,11 +1159,11 @@ Endpoints.Rag = {
model_name: string,
query: string,
settings: string,
- ragFolder: string = 'rag'
+ ragFolder: string = 'rag',
) =>
API_URL() +
`experiment/${experimentId}/rag/query?model=${model_name}&query=${query}&settings=${settings}&rag_folder=${ragFolder}`,
- ReIndex: (experimentId: string, folderName: string = "rag") =>
+ ReIndex: (experimentId: string, folderName: string = 'rag') =>
API_URL() + `experiment/${experimentId}/rag/reindex&folder=${folderName}`,
};
@@ -1173,8 +1176,7 @@ Endpoints.Prompts = {
Endpoints.BatchedPrompts = {
List: () => API_URL() + 'batch/list',
New: () => API_URL() + 'batch/new',
- Delete: (promptId: string) =>
- API_URL() + 'batch/delete/' + promptId,
+ Delete: (promptId: string) => API_URL() + 'batch/delete/' + promptId,
};
Endpoints.Tools = {
@@ -1201,7 +1203,7 @@ Endpoints.ServerInfo = {
Endpoints.Charts = {
CompareEvals: (jobIds: string) =>
API_URL() + 'evals/compare_evals?job_list=' + jobIds,
-}
+};
export function GET_TRAINING_TEMPLATE_URL() {
return API_URL() + 'train/templates';
@@ -1209,7 +1211,7 @@ export function GET_TRAINING_TEMPLATE_URL() {
export function CREATE_TRAINING_JOB_URL(
template_id: string,
- experiment_id: string
+ experiment_id: string,
) {
return (
API_URL() +
@@ -1288,12 +1290,12 @@ Endpoints.Experiment = {
'?eval_name=' +
evalName,
DeleteGeneration: (experimentId: string, evalName: string) =>
- API_URL() +
- 'experiment/' +
- experimentId +
- '/generations/delete' +
- '?generation_name=' +
- evalName,
+ API_URL() +
+ 'experiment/' +
+ experimentId +
+ '/generations/delete' +
+ '?generation_name=' +
+ evalName,
GetEvalOutput: (experimentId: string, eval_name: string) =>
API_URL() +
'experiment/' +
@@ -1302,17 +1304,17 @@ Endpoints.Experiment = {
'?eval_name=' +
eval_name,
GetGenerationOutput: (experimentId: string, eval_name: string) =>
- API_URL() +
- 'experiment/' +
- experimentId +
- '/generations/get_output' +
- '?eval_name=' +
- eval_name,
+ API_URL() +
+ 'experiment/' +
+ experimentId +
+ '/generations/get_output' +
+ '?eval_name=' +
+ eval_name,
RunExport: (
id: string,
pluginName: string,
pluginArchitecture: string,
- pluginParams: string
+ pluginParams: string,
) => {
return (
API_URL() +
@@ -1341,9 +1343,9 @@ Endpoints.Experiment = {
DeleteConversation: (experimentId: string, conversationId: string) =>
FULL_PATH(
'experiment/' +
- experimentId +
- '/conversations/delete?conversation_id=' +
- conversationId
+ experimentId +
+ '/conversations/delete?conversation_id=' +
+ conversationId,
),
InstallPlugin: (experimentId: string, pluginId: string) =>
API_URL() +
@@ -1364,14 +1366,14 @@ Endpoints.Experiment = {
ListScriptsOfType: (
experimentId: string,
type: string,
- filter: string | null = null
+ filter: string | null = null,
) =>
FULL_PATH(
'experiment/' +
- experimentId +
- '/plugins/list?type=' +
- type +
- (filter ? '&filter=' + filter : '')
+ experimentId +
+ '/plugins/list?type=' +
+ type +
+ (filter ? '&filter=' + filter : ''),
),
ScriptListFiles: (experimentId: string, id: string) =>
API_URL() + 'experiment/' + experimentId + '/plugins/' + id + '/list_files',
@@ -1394,7 +1396,7 @@ Endpoints.Experiment = {
ScriptDeleteFile: (
experimentId: string,
pluginId: string,
- filename: string
+ filename: string,
) =>
API_URL() +
'experiment/' +
@@ -1429,13 +1431,13 @@ Endpoints.Experiment = {
StreamOutputFromJob: (jobId: string) =>
API_URL() + `jobs/${jobId}/stream_output`,
StreamDetailedJSONReportFromJob: (jobId: string, fileName: string) =>
- API_URL() + `jobs/${jobId}/stream_detailed_json_report?file_name=${fileName}`,
- GetAdditionalDetails : (jobId: string, task: string = "view") =>
+ API_URL() +
+ `jobs/${jobId}/stream_detailed_json_report?file_name=${fileName}`,
+ GetAdditionalDetails: (jobId: string, task: string = 'view') =>
API_URL() + `jobs/${jobId}/get_additional_details?task=${task}`,
- GetGeneratedDataset : (jobId: string) =>
+ GetGeneratedDataset: (jobId: string) =>
API_URL() + `jobs/${jobId}/get_generated_dataset`,
- GetPlotJSON: (jobId: string) =>
- API_URL() + `jobs/${jobId}/get_figure_json`,
+ GetPlotJSON: (jobId: string) => API_URL() + `jobs/${jobId}/get_figure_json`,
};
Endpoints.Jobs = {
@@ -1445,7 +1447,7 @@ Endpoints.Jobs = {
experimentId?: string,
type?: string,
status?: string,
- data?: string //Should be JSON
+ data?: string, //Should be JSON
) =>
API_URL() +
'jobs/create' +
@@ -1464,7 +1466,7 @@ Endpoints.Jobs = {
name: string,
description: string,
type: string,
- config: Object
+ config: Object,
) =>
API_URL() +
'jobs/template/update' +
@@ -1495,7 +1497,7 @@ export function GET_EXPERIMENTS_URL() {
export function GET_EXPERIMENT_UPDATE_CONFIG_URL(
id: string,
key: string,
- value: string | undefined
+ value: string | undefined,
) {
if (value === undefined) {
value = '';
@@ -1516,7 +1518,7 @@ export async function EXPERIMENT_ADD_EVALUATION(
id: string,
name: string,
pluginId: string,
- scriptParameters: any
+ scriptParameters: any,
) {
const newPlugin = {
name: name,
@@ -1539,7 +1541,7 @@ export async function EXPERIMENT_ADD_EVALUATION(
export async function EXPERIMENT_EDIT_EVALUATION(
id: string,
evalName: string,
- scriptParameters: any
+ scriptParameters: any,
) {
const newPlugin = {
evalName: evalName,
@@ -1562,7 +1564,7 @@ export async function EXPERIMENT_ADD_GENERATION(
id: string,
name: string,
pluginId: string,
- scriptParameters: any
+ scriptParameters: any,
) {
const newPlugin = {
name: name,
@@ -1570,14 +1572,17 @@ export async function EXPERIMENT_ADD_GENERATION(
script_parameters: scriptParameters,
};
- const response = await fetch(API_URL() + 'experiment/' + id + '/generations/add', {
- method: 'POST',
- headers: {
- 'Content-Type': 'application/json',
- accept: 'application/json',
+ const response = await fetch(
+ API_URL() + 'experiment/' + id + '/generations/add',
+ {
+ method: 'POST',
+ headers: {
+ 'Content-Type': 'application/json',
+ accept: 'application/json',
+ },
+ body: JSON.stringify(newPlugin),
},
- body: JSON.stringify(newPlugin),
- });
+ );
const result = await response.json();
return result;
}
@@ -1585,26 +1590,28 @@ export async function EXPERIMENT_ADD_GENERATION(
export async function EXPERIMENT_EDIT_GENERATION(
id: string,
evalName: string,
- scriptParameters: any
+ scriptParameters: any,
) {
const newPlugin = {
evalName: evalName,
script_parameters: scriptParameters,
};
- const response = await fetch(API_URL() + 'experiment/' + id + '/generations/edit', {
- method: 'POST',
- headers: {
- 'Content-Type': 'application/json',
- accept: 'application/json',
+ const response = await fetch(
+ API_URL() + 'experiment/' + id + '/generations/edit',
+ {
+ method: 'POST',
+ headers: {
+ 'Content-Type': 'application/json',
+ accept: 'application/json',
+ },
+ body: JSON.stringify(newPlugin),
},
- body: JSON.stringify(newPlugin),
- });
+ );
const result = await response.json();
return result;
}
-
export function CREATE_EXPERIMENT_URL(name: string) {
return API_URL() + 'experiment/create?name=' + name;
}
@@ -1706,7 +1713,7 @@ export async function activateWorker(
adaptorName: string = '',
engine: string | null = 'default',
parameters: object = {},
- experimentId: string = ''
+ experimentId: string = '',
) {
let response;
@@ -1724,16 +1731,16 @@ export async function activateWorker(
try {
response = await fetch(
API_URL() +
- 'server/worker_start?model_name=' +
- model +
- '&adaptor=' +
- adaptorName +
- '&engine=' +
- engine +
- '&experiment_id=' +
- experimentId +
- '¶meters=' +
- paramsJSON
+ 'server/worker_start?model_name=' +
+ model +
+ '&adaptor=' +
+ adaptorName +
+ '&engine=' +
+ engine +
+ '&experiment_id=' +
+ experimentId +
+ '¶meters=' +
+ paramsJSON,
);
const result = await response.json();
return result;
@@ -1762,7 +1769,7 @@ export function activateTransformerLabAPI(): void {
export async function startFinetune(
modelName: string,
adaptorName: string,
- trainingData: string
+ trainingData: string,
) {
const response = await fetch(
`${API_URL()}train/finetune_lora?model=${modelName}&adaptor_name=${adaptorName}`,
@@ -1773,7 +1780,7 @@ export async function startFinetune(
accept: 'application/json',
},
body: JSON.stringify(trainingData),
- }
+ },
);
const result = await response.json();
return result;
@@ -1802,7 +1809,7 @@ export async function saveTrainingTemplate(
name: string,
description: string,
type: string,
- config: string
+ config: string,
) {
// template_id: str, description: str, type: str, datasets: str, config: str
@@ -1820,7 +1827,7 @@ export async function saveTrainingTemplate(
accept: 'application/json',
},
body: JSON.stringify(configBody),
- }
+ },
);
const result = await response.json();
return result;
@@ -1880,14 +1887,14 @@ export function usePluginStatus(experimentInfo: any) {
experimentInfo
? Endpoints.Experiment.ListScripts(experimentInfo?.id)
: null,
- fetcher
+ fetcher,
);
let outdatedPluginsCount = null;
if (data) {
outdatedPluginsCount = data.filter(
(plugin: any) =>
- plugin?.gallery_version && plugin?.version != plugin?.gallery_version
+ plugin?.gallery_version && plugin?.version != plugin?.gallery_version,
).length;
}
@@ -1913,7 +1920,7 @@ export function useServerStats() {
export async function downloadPlugin(pluginId: string) {
const response = await fetch(
- API_URL() + 'plugins/download?plugin_slug=' + pluginId
+ API_URL() + 'plugins/download?plugin_slug=' + pluginId,
);
const result = await response.json();
return result;