### 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:
@@ -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,
|
||||
};
|
||||
};
|
||||
|
||||
|
||||
Reference in New Issue
Block a user