Flexibility to add already planned task (aiplanethub#54)

* flexibility to the add the already planned task. Limit max_iterations = 20

* avoid ddgs max_results to exceed 15

---------

Co-authored-by: lucifertrj <[email protected]>

Author: tarun-aiplanet

src/openagi/actions/tools/ddg_search.py

@@ -3,7 +3,7 @@
 from openagi.actions.base import BaseAction
 from pydantic import Field
 from duckduckgo_search import DDGS
-
+import logging
 
 class DuckDuckGoSearch(BaseAction):
     """Use this Action to search DuckDuckGo for a query."""
@@ -31,6 +31,10 @@ def _get_ddgs(self):
         return DDGS()
 
     def execute(self):
+        if self.max_results > 15:
+            logging.info("Over threshold value... Limiting the Max results to 15")
+            self.max_results = 15
+        
         result = self._get_ddgs().text(
             self.query,
             max_results=self.max_results,

src/openagi/agent.py

@@ -45,7 +45,7 @@ class Admin(BaseModel):
         description="Actions that the Agent supports", default_factory=list
     )
     max_iterations: int = Field(
-        default=20, description="Maximum number of steps to achieve the objective."
+        default=15, description="Maximum number of steps to achieve the objective."
     )
     output_format: OutputFormat = Field(
         default=OutputFormat.markdown,
@@ -420,11 +420,12 @@ def single_agent_execution(self, query: str, description: str, task_lists: TaskL
         logging.debug(f"Execution Completed for Session ID - {self.memory.session_id}")
         return output
 
-    def run(self, query: str, description: str):
+    def run(self, query: str, description: str,planned_tasks: Optional[List[Dict]] = None):
         logging.info("Running Admin Agent...")
         logging.info(f"SessionID - {self.memory.session_id}")
 
-        planned_tasks = self.run_planner(query=query, descripton=description)
+        if planned_tasks is None:
+            planned_tasks = self.run_planner(query=query, descripton=description)
 
         logging.info("Tasks Planned...")
         logging.debug(f"{planned_tasks=}")