added thinking model support which directly use thinking model's reasoning instead of cot

Author: Li Yin

adalflow/adalflow/components/agent/agent.py

@@ -260,7 +260,7 @@ def __init__(
         # default agent parameters
         answer_data_type: Optional[Type[T]] = str,  # the data type of the final answer
         max_steps: Optional[int] = 10,
-        is_thinking_model: Optional[bool] = False,  # support thinking model in agent
+        is_thinking_model: Optional[bool] = False,  # when thinking model turned on, it disables the CoT field in the output
         # for fully customize the agent
         tool_manager: Optional[
             ToolManager

adalflow/adalflow/components/agent/runner.py

@@ -154,6 +154,9 @@ def __init__(
         self._current_task = None  # Track the current running task
         self._current_streaming_result = None  # Track the current streaming result
 
+        # support thinking model
+        self.is_thinking_model = agent.is_thinking_model if hasattr(agent, 'is_thinking_model')  else False
+
     def _init_permission_manager(self):
         """Initialize the permission manager and register tools that require approval."""
         if self.permission_manager and hasattr(self.agent, "tool_manager"):
@@ -546,8 +549,11 @@ def call(
                             break
 
                         function = output.data
+                        thinking = output.thinking if hasattr(output, 'thinking') else None
                         if function is not None:
-                            function.id = str(uuid.uuid4()) # add function id 
+                            function.id = str(uuid.uuid4()) # add function id
+                            if thinking is not None and self.is_thinking_model:
+                                function.thought = thinking
                         printc(f"function: {function}", color="yellow")
                         if function is None:
                             error_msg = output.error
@@ -832,9 +838,12 @@ async def acall(
                             break
 
                         function = output.data
+                        thinking = output.thinking if hasattr(output, 'thinking') else None
                         if function is not None:
                             # add a function id
                             function.id = str(uuid.uuid4())
+                            if thinking is not None and self.is_thinking_model:
+                                function.thought = thinking
                         printc(f"function: {function}", color="yellow")
 
                         if self._check_last_step(function):
@@ -1175,6 +1184,7 @@ async def impl_astream(
                         # handle function output 
 
                         function = output.data # here are the recoverable errors, should continue to step output
+                        thinking = output.thinking # check the reasoning model response
                         function.id = str(uuid.uuid4()) # add function id 
                         function_result = None
                         function_output_observation = None
@@ -1201,6 +1211,9 @@ async def impl_astream(
                             # for normal function
                             function.id = str(uuid.uuid4())
 
+                            if thinking is not None and self.is_thinking_model:
+                                function.thought = thinking
+
                             # TODO: simplify this
                             tool_call_id = function.id
                             tool_call_name = function.name