Skip to content

Commit a4a0ecc

Browse files
committed
LLMs: reset user params
1 parent 686ad2e commit a4a0ecc

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

src/common/stores/llms/store-llms.ts

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ interface LlmsRootActions {
3939
updateLLM: (id: DLLMId, partial: Partial<DLLM>) => void;
4040
updateLLMUserParameters: (id: DLLMId, partial: Partial<DModelParameterValues>) => void;
4141
deleteLLMUserParameter: (id: DLLMId, parameterId: DModelParameterId) => void;
42+
resetLLMUserParameters: (id: DLLMId) => void;
4243

4344
createModelsService: (vendor: IModelVendor) => DModelsService;
4445
removeService: (id: DModelsServiceId) => void;
@@ -155,6 +156,15 @@ export const useModelsStore = create<LlmsStore>()(persist(
155156
),
156157
})),
157158

159+
resetLLMUserParameters: (id: DLLMId) =>
160+
set(({ llms }) => ({
161+
llms: llms.map((llm: DLLM): DLLM =>
162+
llm.id === id
163+
? { ...llm, userParameters: {} }
164+
: llm,
165+
),
166+
})),
167+
158168
createModelsService: (vendor: IModelVendor): DModelsService => {
159169

160170
function _locallyUniqueServiceId(vendorId: ModelVendorId, existingServices: DModelsService[]): DModelsServiceId {

0 commit comments

Comments
 (0)