feat: add user_id to new_llm_configs and image_generation_configs for user association

This commit is contained in:
Anish Sarkar 2026-02-09 18:30:52 +05:30
parent 7cede99d29
commit 4b60068e8b
7 changed files with 214 additions and 4 deletions

View file

@ -62,6 +62,7 @@ export const newLLMConfig = z.object({
// Metadata
created_at: z.string(),
search_space_id: z.number(),
user_id: z.string(),
});
/**
@ -75,6 +76,7 @@ export const newLLMConfigPublic = newLLMConfig.omit({ api_key: true });
export const createNewLLMConfigRequest = newLLMConfig.omit({
id: true,
created_at: true,
user_id: true,
});
export const createNewLLMConfigResponse = newLLMConfig;
@ -109,6 +111,7 @@ export const updateNewLLMConfigRequest = z.object({
id: true,
created_at: true,
search_space_id: true,
user_id: true,
})
.partial(),
});
@ -200,11 +203,13 @@ export const imageGenerationConfig = z.object({
litellm_params: z.record(z.string(), z.any()).nullable().optional(),
created_at: z.string(),
search_space_id: z.number(),
user_id: z.string(),
});
export const createImageGenConfigRequest = imageGenerationConfig.omit({
id: true,
created_at: true,
user_id: true,
});
export const createImageGenConfigResponse = imageGenerationConfig;
@ -213,7 +218,7 @@ export const getImageGenConfigsResponse = z.array(imageGenerationConfig);
export const updateImageGenConfigRequest = z.object({
id: z.number(),
data: imageGenerationConfig.omit({ id: true, created_at: true, search_space_id: true }).partial(),
data: imageGenerationConfig.omit({ id: true, created_at: true, search_space_id: true, user_id: true }).partial(),
});
export const updateImageGenConfigResponse = imageGenerationConfig;