praisonaiagents 0.0.136__tar.gz → 0.0.138__tar.gz

{praisonaiagents-0.0.136 → praisonaiagents-0.0.138}/PKG-INFO

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

{praisonaiagents-0.0.136 → praisonaiagents-0.0.138}/praisonaiagents/llm/llm.py

@@ -302,6 +302,42 @@ class LLM:
         
         return False
 
+    def _generate_ollama_tool_summary(self, tool_results: List[Any], response_text: str) -> Optional[str]:
+        """
+        Generate a summary from tool results for Ollama to prevent infinite loops.
+        
+        This prevents infinite loops where Ollama provides an empty response after a
+        tool call, expecting the user to prompt for a summary.
+
+        Args:
+            tool_results: The list of results from tool execution.
+            response_text: The text response from the LLM.
+
+        Returns:
+            A summary string if conditions are met, otherwise None.
+        """
+        # Constant for minimal response length check
+        OLLAMA_MIN_RESPONSE_LENGTH = 10
+        
+        # Only generate summary for Ollama with tool results
+        if not (self._is_ollama_provider() and tool_results):
+            return None
+
+        # If response is substantial, no summary needed
+        if response_text and len(response_text.strip()) > OLLAMA_MIN_RESPONSE_LENGTH:
+            return None
+            
+        # Build tool summary efficiently
+        summary_lines = ["Based on the tool execution results:"]
+        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}")
+        
+        return "\n".join(summary_lines)
+
     def _format_ollama_tool_result_message(self, function_name: str, tool_result: Any) -> Dict[str, str]:
         """
         Format tool result message for Ollama provider.
@@ -858,7 +894,13 @@ class LLM:
                                 message=original_prompt,
                                 response=response_content,
                                 markdown=markdown,
-                                generation_time=generation_time_val
+                                generation_time=generation_time_val,
+                                agent_name=agent_name,
+                                agent_role=agent_role,
+                                agent_tools=agent_tools,
+                                task_name=task_name,
+                                task_description=task_description,
+                                task_id=task_id
                             )
                             callback_executed = True
                     
@@ -927,7 +969,13 @@ class LLM:
                                     message=original_prompt,
                                     response=response_text,
                                     markdown=markdown,
-                                    generation_time=time.time() - current_time
+                                    generation_time=time.time() - current_time,
+                                    agent_name=agent_name,
+                                    agent_role=agent_role,
+                                    agent_tools=agent_tools,
+                                    task_name=task_name,
+                                    task_description=task_description,
+                                    task_id=task_id
                                 )
                                 callback_executed = True
 
@@ -981,7 +1029,13 @@ class LLM:
                                     message=original_prompt,
                                     response=response_text,
                                     markdown=markdown,
-                                    generation_time=time.time() - current_time
+                                    generation_time=time.time() - current_time,
+                                    agent_name=agent_name,
+                                    agent_role=agent_role,
+                                    agent_tools=agent_tools,
+                                    task_name=task_name,
+                                    task_description=task_description,
+                                    task_id=task_id
                                 )
                                 callback_executed = True
                     
@@ -1072,13 +1126,19 @@ class LLM:
                             final_response_text = response_text.strip()
                             break
                         
+                        # Special handling for Ollama to prevent infinite loops
+                        tool_summary = self._generate_ollama_tool_summary(tool_results, response_text)
+                        if tool_summary:
+                            final_response_text = tool_summary
+                            break
+                        
                         # Otherwise, continue the loop to check if more tools are needed
                         iteration_count += 1
                         continue
                     else:
                         # No tool calls, we're done with this iteration
                         # If we've executed tools in previous iterations, this response contains the final answer
-                        if iteration_count > 0:
+                        if iteration_count > 0 and not final_response_text:
                             final_response_text = response_text.strip() if response_text else ""
                         break
                         
@@ -1133,7 +1193,13 @@ class LLM:
                     message=original_prompt,
                     response=response_content,
                     markdown=markdown,
-                    generation_time=generation_time_val
+                    generation_time=generation_time_val,
+                    agent_name=agent_name,
+                    agent_role=agent_role,
+                    agent_tools=agent_tools,
+                    task_name=task_name,
+                    task_description=task_description,
+                    task_id=task_id
                 )
                 callback_executed = True
             
@@ -1162,7 +1228,13 @@ class LLM:
                         message=original_prompt,
                         response=response_text,
                         markdown=markdown,
-                        generation_time=time.time() - start_time
+                        generation_time=time.time() - start_time,
+                        agent_name=agent_name,
+                        agent_role=agent_role,
+                        agent_tools=agent_tools,
+                        task_name=task_name,
+                        task_description=task_description,
+                        task_id=task_id
                     )
                     callback_executed = True
                 return response_text
@@ -1182,7 +1254,13 @@ class LLM:
                         message=original_prompt,
                         response=response_text,
                         markdown=markdown,
-                        generation_time=time.time() - start_time
+                        generation_time=time.time() - start_time,
+                        agent_name=agent_name,
+                        agent_role=agent_role,
+                        agent_tools=agent_tools,
+                        task_name=task_name,
+                        task_description=task_description,
+                        task_id=task_id
                     )
                     callback_executed = True
                 
@@ -1815,13 +1893,19 @@ Output MUST be JSON with 'reflection' and 'satisfactory'.
                         final_response_text = response_text.strip()
                         break
                     
+                    # Special handling for Ollama to prevent infinite loops
+                    tool_summary = self._generate_ollama_tool_summary(tool_results, response_text)
+                    if tool_summary:
+                        final_response_text = tool_summary
+                        break
+                    
                     # Continue the loop to check if more tools are needed
                     iteration_count += 1
                     continue
                 else:
                     # No tool calls, we're done with this iteration
                     # If we've executed tools in previous iterations, this response contains the final answer
-                    if iteration_count > 0:
+                    if iteration_count > 0 and not final_response_text:
                         final_response_text = response_text.strip()
                     break
 

{praisonaiagents-0.0.136 → praisonaiagents-0.0.138}/praisonaiagents.egg-info/PKG-INFO

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

{praisonaiagents-0.0.136 → praisonaiagents-0.0.138}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "praisonaiagents"
-version = "0.0.136"
+version = "0.0.138"
 description = "Praison AI agents for completing complex tasks with Self Reflection Agents"
 requires-python = ">=3.10"
 authors = [

{praisonaiagents-0.0.136 → praisonaiagents-0.0.138}/tests/test_fix_comprehensive.py

@@ -57,7 +57,7 @@ print("=" * 60)
 
 agent_gpt4 = Agent(
     instructions="You are a helpful assistant. You can use the tools provided to you to help the user. When asked to multiply a stock price, first get the stock price, then multiply it.",
-    llm="gpt-4o",
+    llm="gpt-4o-mini",
     tools=[get_stock_price, multiply],
     verbose=True
 )