praisonaiagents 0.0.141__py3-none-any.whl → 0.0.142__py3-none-any.whl

praisonaiagents/agent/agent.py

@@ -354,6 +354,8 @@ class Agent:
         self.instructions = instructions
         # Check for model name in environment variable if not provided
         self._using_custom_llm = False
+        # Flag to track if final result has been displayed to prevent duplicates
+        self._final_display_shown = False
         
         # Store OpenAI client parameters for lazy initialization
         self._openai_api_key = api_key
@@ -1173,18 +1175,23 @@ Your Goal: {self.goal}"""
                 task_description=task_description, 
                 task_id=task_id
             )
-        # Only display interaction if not using custom LLM (to avoid double output) and verbose is True
-        if self.verbose and not self._using_custom_llm:
+        # Always display final interaction when verbose is True to ensure consistent formatting
+        # This ensures both OpenAI and custom LLM providers (like Gemini) show formatted output
+        if self.verbose and not self._final_display_shown:
             display_interaction(prompt, response, markdown=self.markdown, 
                               generation_time=generation_time, console=self.console,
                               agent_name=self.name,
                               agent_role=self.role,
                               agent_tools=[t.__name__ for t in self.tools] if self.tools else None,
-                              task_name=task_name,
-                              task_description=task_description,
-                              task_id=task_id)
+                              task_name=None,  # Not available in this context
+                              task_description=None,  # Not available in this context
+                              task_id=None)  # Not available in this context
+            self._final_display_shown = True
 
     def chat(self, prompt, temperature=0.2, tools=None, output_json=None, output_pydantic=None, reasoning_steps=False, stream=True, task_name=None, task_description=None, task_id=None):
+        # Reset the final display flag for each new conversation
+        self._final_display_shown = False
+        
         # Log all parameter values when in debug mode
         if logging.getLogger().getEffectiveLevel() == logging.DEBUG:
             param_info = {
@@ -1533,6 +1540,9 @@ Output MUST be JSON with 'reflection' and 'satisfactory'.
 
     async def achat(self, prompt: str, temperature=0.2, tools=None, output_json=None, output_pydantic=None, reasoning_steps=False, task_name=None, task_description=None, task_id=None):
         """Async version of chat method with self-reflection support.""" 
+        # Reset the final display flag for each new conversation
+        self._final_display_shown = False
+        
         # Log all parameter values when in debug mode
         if logging.getLogger().getEffectiveLevel() == logging.DEBUG:
             param_info = {

praisonaiagents/llm/llm.py

@@ -329,16 +329,29 @@ class LLM:
         # For Ollama, always generate summary when we have tool results
         # This prevents infinite loops caused by empty/minimal responses
             
-        # Build tool summary efficiently
-        summary_lines = ["Based on the tool execution results:"]
-        for i, result in enumerate(tool_results):
+        # Build tool summary more naturally to match OpenAI-style responses
+        if len(tool_results) == 1:
+            # Single tool result - create natural response
+            result = tool_results[0]
             if isinstance(result, dict) and 'result' in result:
-                function_name = result.get('function_name', 'Tool')
-                summary_lines.append(f"- {function_name}: {result['result']}")
+                return str(result['result'])
             else:
-                summary_lines.append(f"- Tool {i+1}: {result}")
-        
-        return "\n".join(summary_lines)
+                return str(result)
+        else:
+            # Multiple tool results - create coherent summary
+            summary_lines = []
+            for i, result in enumerate(tool_results):
+                if isinstance(result, dict) and 'result' in result:
+                    function_name = result.get('function_name', 'Tool')
+                    summary_lines.append(f"{function_name}: {result['result']}")
+                else:
+                    summary_lines.append(f"Tool {i+1}: {result}")
+            
+            # Create more natural summary text
+            if len(summary_lines) == 2:
+                return f"{summary_lines[0]}. {summary_lines[1]}."
+            else:
+                return "Based on the tool execution: " + ". ".join(summary_lines) + "."
 
     def _format_ollama_tool_result_message(self, function_name: str, tool_result: Any) -> Dict[str, str]:
         """
@@ -1173,6 +1186,7 @@ class LLM:
                             final_response_text = response_text.strip()
                             break
                         
+                        
                         # Special handling for Ollama to prevent infinite loops
                         # Only generate summary after multiple iterations to allow sequential execution
                         should_break, tool_summary_text, iteration_count = self._handle_ollama_sequential_logic(
@@ -1198,6 +1212,17 @@ class LLM:
                         continue
                     else:
                         # No tool calls, we're done with this iteration
+                        
+                        # Special early stopping logic for Ollama when tool results are available
+                        # Ollama often provides empty responses after successful tool execution
+                        if (self._is_ollama_provider() and accumulated_tool_results and iteration_count >= 1 and 
+                            (not response_text or response_text.strip() == "")):
+                            # Generate coherent response from tool results
+                            tool_summary = self._generate_ollama_tool_summary(accumulated_tool_results, response_text)
+                            if tool_summary:
+                                final_response_text = tool_summary
+                                break
+                        
                         # If we've executed tools in previous iterations, this response contains the final answer
                         if iteration_count > 0 and not final_response_text:
                             final_response_text = response_text.strip() if response_text else ""
@@ -1956,6 +1981,7 @@ Output MUST be JSON with 'reflection' and 'satisfactory'.
                         final_response_text = response_text.strip()
                         break
                     
+                    
                     # Special handling for Ollama to prevent infinite loops
                     # Only generate summary after multiple iterations to allow sequential execution
                     should_break, tool_summary_text, iteration_count = self._handle_ollama_sequential_logic(
@@ -1981,6 +2007,17 @@ Output MUST be JSON with 'reflection' and 'satisfactory'.
                     continue
                 else:
                     # No tool calls, we're done with this iteration
+                    
+                    # Special early stopping logic for Ollama when tool results are available
+                    # Ollama often provides empty responses after successful tool execution
+                    if (self._is_ollama_provider() and accumulated_tool_results and iteration_count >= 1 and 
+                        (not response_text or response_text.strip() == "")):
+                        # Generate coherent response from tool results
+                        tool_summary = self._generate_ollama_tool_summary(accumulated_tool_results, response_text)
+                        if tool_summary:
+                            final_response_text = tool_summary
+                            break
+                    
                     # If we've executed tools in previous iterations, this response contains the final answer
                     if iteration_count > 0 and not final_response_text:
                         final_response_text = response_text.strip()

praisonaiagents/telemetry/__init__.py

@@ -65,8 +65,8 @@ def _ensure_atexit():
     ])
     
     if not telemetry_disabled:
-        # Register atexit handler to flush telemetry on exit
-        atexit.register(lambda: get_telemetry().flush())
+        # Register atexit handler to properly shutdown telemetry on exit
+        atexit.register(lambda: get_telemetry().shutdown())
         _atexit_registered = True
 
 def _initialize_telemetry():

praisonaiagents/telemetry/telemetry.py

@@ -354,8 +354,8 @@ class TelemetryCollector:
         pass
         
     def stop(self):
-        """Stop telemetry collection and flush data."""
-        self.telemetry.flush()
+        """Stop telemetry collection and properly shutdown."""
+        self.telemetry.shutdown()
     
     def trace_agent_execution(self, agent_name: str, **attributes):
         """Compatibility method for agent execution tracking."""

{praisonaiagents-0.0.141.dist-info → praisonaiagents-0.0.142.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonaiagents
-Version: 0.0.141
+Version: 0.0.142
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Python: >=3.10

{praisonaiagents-0.0.141.dist-info → praisonaiagents-0.0.142.dist-info}/RECORD

@@ -3,7 +3,7 @@ praisonaiagents/approval.py,sha256=UJ4OhfihpFGR5CAaMphqpSvqdZCHi5w2MGw1MByZ1FQ,9
 praisonaiagents/main.py,sha256=b5dKlkf6NMeumSzixreHB9ui90f8YMAi5r1fCbTpQVw,17225
 praisonaiagents/session.py,sha256=FHWButPBaFGA4x1U_2gImroQChHnFy231_aAa_n5KOQ,20364
 praisonaiagents/agent/__init__.py,sha256=FkjW6f3EU8heQ9tvctfLbOWV9_dOXmS1PcFNgcStns8,403
-praisonaiagents/agent/agent.py,sha256=mtI8Y_OWHQ_zLpIq5LWuuqYhYr5ipzEUsWSlSm2hN00,125235
+praisonaiagents/agent/agent.py,sha256=1wI2Ohp9evUza7qlZt3yIu4goroR8Jm8EIenXaOpDso,125805
 praisonaiagents/agent/handoff.py,sha256=Saq0chqfvC6Zf5UbXvmctybbehqnotrXn72JsS-76Q0,13099
 praisonaiagents/agent/image_agent.py,sha256=Bbwg_h3qhjhG7gMH8sdcQXhcOFgE_wSvcdhtqH5f2UM,9145
 praisonaiagents/agent/router_agent.py,sha256=a_b6w5Ti05gvK80uKGMIcT14fiCTKv8rCQPCWAUfIiE,12713
@@ -17,7 +17,7 @@ praisonaiagents/knowledge/__init__.py,sha256=xL1Eh-a3xsHyIcU4foOWF-JdWYIYBALJH9b
 praisonaiagents/knowledge/chunking.py,sha256=G6wyHa7_8V0_7VpnrrUXbEmUmptlT16ISJYaxmkSgmU,7678
 praisonaiagents/knowledge/knowledge.py,sha256=-di_h9HxXQfAhTMMerhK16tfw8DtUndp44TGkBOzkZs,15539
 praisonaiagents/llm/__init__.py,sha256=tHvWq5mv4K4MhWr0s6rqox8UnJ5RK0kXhYuD40WkZQA,1747
-praisonaiagents/llm/llm.py,sha256=JJ-MT09nC9h5NSYeBTG0LoLmeODNiyRGq09-VgLW1Ok,134482
+praisonaiagents/llm/llm.py,sha256=vBw810jpgjZyVAvHcGAG0-QpdbL5e2DBwjn8qgE5NXc,136663
 praisonaiagents/llm/model_capabilities.py,sha256=cxOvZcjZ_PIEpUYKn3S2FMyypfOSfbGpx4vmV7Y5vhI,3967
 praisonaiagents/llm/model_router.py,sha256=Jy2pShlkLxqXF3quz-MRB3-6L9vaUSgUrf2YJs_Tsg0,13995
 praisonaiagents/llm/openai_client.py,sha256=EgWjkDjVpnLKCp1gBFjccDGyqR1anOcSYJYCo45fuEI,46046
@@ -31,9 +31,9 @@ praisonaiagents/process/__init__.py,sha256=lkYbL7Hn5a0ldvJtkdH23vfIIZLIcanK-65C0
 praisonaiagents/process/process.py,sha256=wXKZ2Z26vB9osmVbD5xqkUlUQRvWEpvL8j9hiuiHrQ0,78246
 praisonaiagents/task/__init__.py,sha256=VL5hXVmyGjINb34AalxpBMl-YW9m5EDcRkMTKkSSl7c,80
 praisonaiagents/task/task.py,sha256=-EXxw3czWZdAK1WWI6Dvga5CujsItgk9RWYD_CdW47w,24075
-praisonaiagents/telemetry/__init__.py,sha256=5iAOrj_N_cKMmh2ltWGYs3PfOYt_jcwUoElW8fTAIsc,3062
+praisonaiagents/telemetry/__init__.py,sha256=x66unQefO4Rt_kAC1CyuRtq2txbJ5pqEXvoaCncha90,3077
 praisonaiagents/telemetry/integration.py,sha256=8h8TDlPFTbsBmU5rIYNOibJbwEEEWmzS1ENE9uPTvvg,8696
-praisonaiagents/telemetry/telemetry.py,sha256=w15OMQwL2rGQG7CIpwABmVFsAQKSAVmi9BYbmR3BdDQ,13459
+praisonaiagents/telemetry/telemetry.py,sha256=jVtUjXDRvRLIZa9TiAHdTNec08pqpYJLx26_CiiDXSc,13469
 praisonaiagents/tools/README.md,sha256=am9mlHp46sC1U9HfyXtX-E_cckxpazprl4tuVFYHP_0,4905
 praisonaiagents/tools/__init__.py,sha256=9NYh9anzJZlaLtrRINdM1uD6JfNSuOzZAFMaarO6yAU,9321
 praisonaiagents/tools/arxiv_tools.py,sha256=1stb31zTjLTon4jCnpZG5de9rKc9QWgC0leLegvPXWo,10528
@@ -57,7 +57,7 @@ praisonaiagents/tools/xml_tools.py,sha256=iYTMBEk5l3L3ryQ1fkUnNVYK-Nnua2Kx2S0dxN
 praisonaiagents/tools/yaml_tools.py,sha256=uogAZrhXV9O7xvspAtcTfpKSQYL2nlOTvCQXN94-G9A,14215
 praisonaiagents/tools/yfinance_tools.py,sha256=s2PBj_1v7oQnOobo2fDbQBACEHl61ftG4beG6Z979ZE,8529
 praisonaiagents/tools/train/data/generatecot.py,sha256=H6bNh-E2hqL5MW6kX3hqZ05g9ETKN2-kudSjiuU_SD8,19403
-praisonaiagents-0.0.141.dist-info/METADATA,sha256=QTjOqWcXdAooalUe3_TtFp2NyMXuj_cof83ZiOxFPtA,1673
-praisonaiagents-0.0.141.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-praisonaiagents-0.0.141.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
-praisonaiagents-0.0.141.dist-info/RECORD,,
+praisonaiagents-0.0.142.dist-info/METADATA,sha256=LlCoVIYbTPQrfhx1mkLuxt1-PGYAdWLo-DRMjm8xiMk,1673
+praisonaiagents-0.0.142.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+praisonaiagents-0.0.142.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
+praisonaiagents-0.0.142.dist-info/RECORD,,