From 83edeb40481478a21bfeb0b26ebf04e441ceb203 Mon Sep 17 00:00:00 2001 From: Henry Fontanier Date: Thu, 25 Jul 2024 11:14:52 +0200 Subject: [PATCH] enh(viz): use claude 3.5 sonnet unless forbidden (#6485) Co-authored-by: Henry Fontanier --- front/lib/api/assistant/actions/visualization.ts | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/front/lib/api/assistant/actions/visualization.ts b/front/lib/api/assistant/actions/visualization.ts index 1ac26cec0aa0..30e313652ea0 100644 --- a/front/lib/api/assistant/actions/visualization.ts +++ b/front/lib/api/assistant/actions/visualization.ts @@ -14,8 +14,10 @@ import type { } from "@dust-tt/types"; import { BaseAction, + CLAUDE_3_5_SONNET_DEFAULT_MODEL_CONFIG, cloneBaseConfig, DustProdActionRegistry, + isProviderWhitelisted, Ok, VisualizationActionOutputSchema, } from "@dust-tt/types"; @@ -245,10 +247,19 @@ export class VisualizationConfigurationServerRunner extends BaseActionConfigurat const config = cloneBaseConfig( DustProdActionRegistry["assistant-v2-visualization"].config ); - const model = agentConfiguration.model; + + // If we can use Sonnet 3.5, we use it. + // Otherwise, we use the model from the agent configuration. + const model = + auth.isUpgraded() && isProviderWhitelisted(owner, "anthropic") + ? CLAUDE_3_5_SONNET_DEFAULT_MODEL_CONFIG + : agentConfiguration.model; + config.MODEL.provider_id = model.providerId; config.MODEL.model_id = model.modelId; - config.MODEL.temperature = model.temperature; + + // Preserve the temperature from the agent configuration. + config.MODEL.temperature = agentConfiguration.model.temperature; // Execute the Vizualization Dust App. const visualizationRes = await runActionStreamed(