feat: support AWS Bedrock #308 (#1617)

### What problem does this PR solve?

feat: support AWS Bedrock #308
### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2024-07-19 18:36:49 +08:00
committed by GitHub
parent 58df013722
commit 657019a5a9
8 changed files with 258 additions and 52 deletions

View File

@@ -168,7 +168,6 @@ export const useSubmitOllama = () => {
export const useSubmitVolcEngine = () => {
const loading = useOneNamespaceEffectsLoading('settingModel', ['add_llm']);
const [selectedVolcFactory, setSelectedVolcFactory] = useState<string>('');
const addLlm = useAddLlm();
const {
visible: volcAddingVisible,
@@ -186,18 +185,40 @@ export const useSubmitVolcEngine = () => {
[hideVolcAddingModal, addLlm],
);
const handleShowVolcAddingModal = (llmFactory: string) => {
setSelectedVolcFactory(llmFactory);
showVolcAddingModal();
};
return {
volcAddingLoading: loading,
onVolcAddingOk,
volcAddingVisible,
hideVolcAddingModal,
showVolcAddingModal: handleShowVolcAddingModal,
selectedVolcFactory,
showVolcAddingModal,
};
};
export const useSubmitBedrock = () => {
const loading = useOneNamespaceEffectsLoading('settingModel', ['add_llm']);
const addLlm = useAddLlm();
const {
visible: bedrockAddingVisible,
hideModal: hideBedrockAddingModal,
showModal: showBedrockAddingModal,
} = useSetModalState();
const onBedrockAddingOk = useCallback(
async (payload: IAddLlmRequestBody) => {
const ret = await addLlm(payload);
if (ret === 0) {
hideBedrockAddingModal();
}
},
[hideBedrockAddingModal, addLlm],
);
return {
bedrockAddingLoading: loading,
onBedrockAddingOk,
bedrockAddingVisible,
hideBedrockAddingModal,
showBedrockAddingModal,
};
};