From b8a7fb936f922e6eeece88213d7406c2e4017bd5 Mon Sep 17 00:00:00 2001
From: ItzCrazyKns <95534749+ItzCrazyKns@users.noreply.github.com>
Date: Tue, 2 Dec 2025 11:50:26 +0530
Subject: [PATCH] feat(classifier): add `showCalculationWidget`

---
 src/lib/agents/search/classifier.ts | 3 +++
 src/lib/agents/search/types.ts      | 1 +
 2 files changed, 4 insertions(+)

diff --git a/src/lib/agents/search/classifier.ts b/src/lib/agents/search/classifier.ts
index b5137a6..5909e35 100644
--- a/src/lib/agents/search/classifier.ts
+++ b/src/lib/agents/search/classifier.ts
@@ -23,6 +23,9 @@ const schema = z.object({
     showStockWidget: z
       .boolean()
       .describe('Indicates whether to show the stock widget.'),
+    showCalculationWidget: z
+      .boolean()
+      .describe('Indicates whether to show the calculation widget.'),
   }),
   standaloneFollowUp: z
     .string()
diff --git a/src/lib/agents/search/types.ts b/src/lib/agents/search/types.ts
index a0b27b4..c30d4f7 100644
--- a/src/lib/agents/search/types.ts
+++ b/src/lib/agents/search/types.ts
@@ -62,6 +62,7 @@ export type ClassifierOutput = {
     discussionSearch: boolean;
     showWeatherWidget: boolean;
     showStockWidget: boolean;
+    showCalculationWidget: boolean;
   };
   standaloneFollowUp: string;
 };