diff --git a/surfsense_web/contracts/types/llm-config.types.ts b/surfsense_web/contracts/types/llm-config.types.ts index b862d84b3..40c7574be 100644 --- a/surfsense_web/contracts/types/llm-config.types.ts +++ b/surfsense_web/contracts/types/llm-config.types.ts @@ -107,6 +107,27 @@ export const getLLMConfigRequest = llmConfig.pick({ id: true }); export const getLLMConfigResponse = llmConfig; +/** + * Update LLM config + */ +export const updateLLMConfigRequest = z.object({ + id: z.number(), + data: llmConfig + .pick({ + name: true, + provider: true, + custom_provider: true, + model_name: true, + api_key: true, + api_base: true, + language: true, + litellm_params: true, + }) + .partial(), +}); + +export const updateLLMConfigResponse = llmConfig; + export type LLMConfig = z.infer; export type LiteLLMProvider = z.infer; export type GlobalLLMConfig = z.infer; @@ -117,3 +138,5 @@ export type GetLLMConfigsRequest = z.infer; export type GetLLMConfigsResponse = z.infer; export type GetLLMConfigRequest = z.infer; export type GetLLMConfigResponse = z.infer; +export type UpdateLLMConfigRequest = z.infer; +export type UpdateLLMConfigResponse = z.infer;