Skip to content

Commit 4d32b23

Browse files
brainlessclaude
andcommitted
fix: add qwen to no-temperature models in document labeler
The document labeler was sending temperature=0.1 for Qwen models, while agent.rs already excluded qwen from temperature. Align both to use temperature=None for qwen to match the model's recommended settings. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
1 parent cd3cbce commit 4d32b23

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

dwata-agents/src/kg_email_extractor/document_labeler.rs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,7 +127,10 @@ impl TemplateDocumentLabelerAgent {
127127
max_tokens: 512,
128128
model: self.model.clone(),
129129
system: Some(system_prompt.clone()),
130-
temperature: if self.model.contains("nano") || self.model.contains("mini") {
130+
temperature: if self.model.contains("nano")
131+
|| self.model.contains("mini")
132+
|| self.model.contains("qwen")
133+
{
131134
None
132135
} else {
133136
Some(0.1)

0 commit comments

Comments
 (0)