feat: add thinking mode functionality to AI widget (#9)

* feat: add thinking mode functionality to AI widget

* style: remove debug logs from js files

Author: cmgzn

docs/sphinx_doc/source/_static/ask-ai-modules/ask-ai-api.js

@@ -157,12 +157,15 @@ export class AskAIApi {
    * @param {Function} onComplete - Callback when complete with verified messages (userMessage, assistantMessage)
    * @param {Function} onError - Callback for errors (error)
    * @param {Function} onToolComplete - Callback when tool execution completes (callId)
+   * @param {Object} modelConfig - Optional model configuration (e.g., { enable_thinking: true })
+   * @param {Function} onThinkingUpdate - Callback for thinking content updates (thinkingText)
    */
-  async getAIResponseStream(message, onContentUpdate, onToolUse, onComplete, onError, onToolComplete) {
+  async getAIResponseStream(message, onContentUpdate, onToolUse, onComplete, onError, onToolComplete, modelConfig = null, onThinkingUpdate = null) {
     let currentStreamContent = '';
     let streamMessageId = `temp_${Date.now()}_${Math.random().toString(36).substr(2, 9)}`;
     let hasReceivedContent = false;
     let streamCompletedSuccessfully = false;
+    let isInReasoningPhase = false;
 
     try {
       const requestBody = {
@@ -181,6 +184,10 @@ export class AskAIApi {
         user_id: this.sessionId,
       };
 
+      if (modelConfig && typeof modelConfig === 'object') {
+        requestBody.model_params = modelConfig;
+      }
+
       console.log('Sending streaming request to:', `${this.getApiBaseUrl()}/process`);
 
       const response = await fetch(`${this.getApiBaseUrl()}/process`, {
@@ -271,28 +278,70 @@ export class AskAIApi {
               }
             }
 
-            // Handle incremental text content
+            // Handle reasoning phase: object="message", type="reasoning"
+            if (data.object === "message" && data.type === "reasoning") {
+              if (data.status === "in_progress") {
+                isInReasoningPhase = true;
+                console.log('Entering reasoning phase, msg_id:', data.id);
+              } else if (data.status === "completed") {
+                // Reasoning phase completed - ignore entirely (skip all further processing)
+                isInReasoningPhase = false;
+                console.log('Reasoning phase completed, msg_id:', data.id);
+                continue;
+              }
+            }
+
+            // Handle normal message phase start: object="message", type="message"
+            // This signals the end of reasoning and start of normal response
+            if (data.object === "message" && data.type === "message" && data.status === "in_progress") {
+              if (isInReasoningPhase) {
+                isInReasoningPhase = false;
+                console.log('Exiting reasoning phase, entering message phase');
+              }
+            }
+
+            // Skip non-delta content events (e.g., completed content summaries from reasoning)
+            if (
+              data.object === "content" &&
+              data.type === "text" &&
+              data.delta !== true &&
+              data.status === "completed"
+            ) {
+              console.log('Skipping completed content summary, sequence:', data.sequence_number);
+              continue;
+            }
+
+            // Handle incremental text content (used for both reasoning and normal text)
             if (
               data.object === "content" &&
               data.type === "text" &&
               data.delta === true &&
               data.text !== undefined
             ) {
-              if (!hasReceivedContent) {
-                currentStreamContent = '';
-                hasReceivedContent = true;
-              }
-              currentStreamContent += data.text;
-              if (onContentUpdate) {
-                onContentUpdate(currentStreamContent);
+              if (isInReasoningPhase) {
+                // Route to thinking callback during reasoning phase
+                if (onThinkingUpdate) {
+                  onThinkingUpdate(data.text);
+                }
+              } else {
+                // Route to normal content callback
+                if (!hasReceivedContent) {
+                  currentStreamContent = '';
+                  hasReceivedContent = true;
+                }
+                currentStreamContent += data.text;
+                if (onContentUpdate) {
+                  onContentUpdate(currentStreamContent);
+                }
               }
             }
 
-            // Final message delivery
+            // Final message delivery (skip reasoning messages to avoid showing thinking as normal text)
             if (
               data.object === "message" &&
               data.status === "completed" &&
               data.role === "assistant" &&
+              data.type !== "reasoning" &&
               Array.isArray(data.content)
             ) {
               const fullText = data.content

docs/sphinx_doc/source/_static/ask-ai-modules/ask-ai-i18n.js

@@ -32,6 +32,9 @@ export const I18N = {
     feedbackSuccess: 'Thank you for your feedback!',
     feedbackError: 'Failed to submit feedback',
     copiedSuccess: 'Copied to clipboard!',
+    thinking: 'Thinking',
+    thinkingTitle: 'Enable/Disable Thinking',
+    thinkingContent: 'Thinking',
     helpSuffix: '\n\n---\n*If you have any questions, please visit [data-juicer issues](https://github.com/datajuicer/data-juicer/issues) or [data-juicer-agents issues](https://github.com/datajuicer/data-juicer-agents/issues)*'
   },
   zh_CN: {
@@ -63,6 +66,9 @@ export const I18N = {
     feedbackSuccess: '感谢您的反馈！',
     feedbackError: '提交反馈失败',
     copiedSuccess: '已复制到剪贴板！',
+    thinking: '思考',
+    thinkingTitle: '开启/关闭思考模式',
+    thinkingContent: '思考',
     helpSuffix: '\n\n---\n*如果您有任何问题，请访问 [data-juicer issues](https://github.com/datajuicer/data-juicer/issues) 或 [data-juicer-agents issues](https://github.com/datajuicer/data-juicer-agents/issues)*'
   }
 };