oagi 0.4.3__py3-none-any.whl → 0.5.0__py3-none-any.whl

oagi/async_client.py

@@ -91,6 +91,8 @@ class AsyncClient:
         task_id: str | None = None,
         instruction: str | None = None,
         max_actions: int | None = 5,
+        last_task_id: str | None = None,
+        history_steps: int | None = None,
         api_version: str | None = None,
     ) -> LLMResponse:
         """
@@ -103,6 +105,8 @@ class AsyncClient:
             task_id: Task ID for continuing existing task
             instruction: Additional instruction when continuing a session (only works with task_id)
             max_actions: Maximum number of actions to return (1-20)
+            last_task_id: Previous task ID to retrieve history from (only works with task_id)
+            history_steps: Number of historical steps to include from last_task_id (default: 1, max: 10)
             api_version: API version header
 
         Returns:
@@ -127,6 +131,10 @@ class AsyncClient:
             payload["instruction"] = instruction
         if max_actions is not None:
             payload["max_actions"] = max_actions
+        if last_task_id is not None:
+            payload["last_task_id"] = last_task_id
+        if history_steps is not None:
+            payload["history_steps"] = history_steps
 
         logger.info(f"Making async API request to /v1/message with model: {model}")
         logger.debug(

oagi/async_short_task.py

@@ -30,12 +30,28 @@ class AsyncShortTask(AsyncTask):
         max_steps: int = 5,
         executor: AsyncActionHandler = None,
         image_provider: AsyncImageProvider = None,
+        last_task_id: str | None = None,
+        history_steps: int | None = None,
     ) -> bool:
-        """Run the task in automatic mode with the provided executor and image provider."""
+        """Run the task in automatic mode with the provided executor and image provider.
+
+        Args:
+            task_desc: Task description
+            max_steps: Maximum number of steps
+            executor: Async handler to execute actions
+            image_provider: Async provider for screenshots
+            last_task_id: Previous task ID to retrieve history from
+            history_steps: Number of historical steps to include
+        """
         logger.info(
             f"Starting async auto mode for task: '{task_desc}' (max_steps: {max_steps})"
         )
-        await self.init_task(task_desc, max_steps=max_steps)
+        await self.init_task(
+            task_desc,
+            max_steps=max_steps,
+            last_task_id=last_task_id,
+            history_steps=history_steps,
+        )
 
         for i in range(max_steps):
             logger.debug(f"Async auto mode step {i + 1}/{max_steps}")

oagi/async_task.py

@@ -29,10 +29,27 @@ class AsyncTask:
         self.task_id: str | None = None
         self.task_description: str | None = None
         self.model = model
+        self.last_task_id: str | None = None
+        self.history_steps: int | None = None
 
-    async def init_task(self, task_desc: str, max_steps: int = 5):
-        """Initialize a new task with the given description."""
+    async def init_task(
+        self,
+        task_desc: str,
+        max_steps: int = 5,
+        last_task_id: str | None = None,
+        history_steps: int | None = None,
+    ):
+        """Initialize a new task with the given description.
+
+        Args:
+            task_desc: Task description
+            max_steps: Maximum number of steps (for logging)
+            last_task_id: Previous task ID to retrieve history from
+            history_steps: Number of historical steps to include (default: 1)
+        """
         self.task_description = task_desc
+        self.last_task_id = last_task_id
+        self.history_steps = history_steps
         response = await self.client.create_message(
             model=self.model,
             screenshot="",
@@ -41,6 +58,10 @@ class AsyncTask:
         )
         self.task_id = response.task_id  # Reset task_id for new task
         logger.info(f"Async task initialized: '{task_desc}' (max_steps: {max_steps})")
+        if last_task_id:
+            logger.info(
+                f"Will include {history_steps or 1} steps from previous task: {last_task_id}"
+            )
 
     async def step(
         self, screenshot: Image | bytes, instruction: str | None = None
@@ -74,6 +95,8 @@ class AsyncTask:
                 task_description=self.task_description,
                 task_id=self.task_id,
                 instruction=instruction,
+                last_task_id=self.last_task_id if self.task_id else None,
+                history_steps=self.history_steps if self.task_id else None,
             )
 
             # Update task_id from response

oagi/short_task.py

@@ -30,12 +30,28 @@ class ShortTask(Task):
         max_steps: int = 5,
         executor: ActionHandler = None,
         image_provider: ImageProvider = None,
+        last_task_id: str | None = None,
+        history_steps: int | None = None,
     ) -> bool:
-        """Run the task in automatic mode with the provided executor and image provider."""
+        """Run the task in automatic mode with the provided executor and image provider.
+
+        Args:
+            task_desc: Task description
+            max_steps: Maximum number of steps
+            executor: Handler to execute actions
+            image_provider: Provider for screenshots
+            last_task_id: Previous task ID to retrieve history from
+            history_steps: Number of historical steps to include
+        """
         logger.info(
             f"Starting auto mode for task: '{task_desc}' (max_steps: {max_steps})"
         )
-        self.init_task(task_desc, max_steps=max_steps)
+        self.init_task(
+            task_desc,
+            max_steps=max_steps,
+            last_task_id=last_task_id,
+            history_steps=history_steps,
+        )
 
         for i in range(max_steps):
             logger.debug(f"Auto mode step {i + 1}/{max_steps}")

oagi/sync_client.py

@@ -130,6 +130,8 @@ class SyncClient:
         task_id: str | None = None,
         instruction: str | None = None,
         max_actions: int | None = 5,
+        last_task_id: str | None = None,
+        history_steps: int | None = None,
         api_version: str | None = None,
     ) -> LLMResponse:
         """
@@ -142,6 +144,8 @@ class SyncClient:
             task_id: Task ID for continuing existing task
             instruction: Additional instruction when continuing a session (only works with task_id)
             max_actions: Maximum number of actions to return (1-20)
+            last_task_id: Previous task ID to retrieve history from (only works with task_id)
+            history_steps: Number of historical steps to include from last_task_id (default: 1, max: 10)
             api_version: API version header
 
         Returns:
@@ -166,6 +170,10 @@ class SyncClient:
             payload["instruction"] = instruction
         if max_actions is not None:
             payload["max_actions"] = max_actions
+        if last_task_id is not None:
+            payload["last_task_id"] = last_task_id
+        if history_steps is not None:
+            payload["history_steps"] = history_steps
 
         logger.info(f"Making API request to /v1/message with model: {model}")
         logger.debug(

oagi/task.py

@@ -28,10 +28,27 @@ class Task:
         self.task_id: str | None = None
         self.task_description: str | None = None
         self.model = model
+        self.last_task_id: str | None = None
+        self.history_steps: int | None = None
 
-    def init_task(self, task_desc: str, max_steps: int = 5):
-        """Initialize a new task with the given description."""
+    def init_task(
+        self,
+        task_desc: str,
+        max_steps: int = 5,
+        last_task_id: str | None = None,
+        history_steps: int | None = None,
+    ):
+        """Initialize a new task with the given description.
+
+        Args:
+            task_desc: Task description
+            max_steps: Maximum number of steps (for logging)
+            last_task_id: Previous task ID to retrieve history from
+            history_steps: Number of historical steps to include (default: 1)
+        """
         self.task_description = task_desc
+        self.last_task_id = last_task_id
+        self.history_steps = history_steps
         response = self.client.create_message(
             model=self.model,
             screenshot="",
@@ -40,6 +57,10 @@ class Task:
         )
         self.task_id = response.task_id  # Reset task_id for new task
         logger.info(f"Task initialized: '{task_desc}' (max_steps: {max_steps})")
+        if last_task_id:
+            logger.info(
+                f"Will include {history_steps or 1} steps from previous task: {last_task_id}"
+            )
 
     def step(self, screenshot: Image | bytes, instruction: str | None = None) -> Step:
         """Send screenshot to the server and get the next actions.
@@ -71,6 +92,8 @@ class Task:
                 task_description=self.task_description,
                 task_id=self.task_id,
                 instruction=instruction,
+                last_task_id=self.last_task_id if self.task_id else None,
+                history_steps=self.history_steps if self.task_id else None,
             )
 
             # Update task_id from response

{oagi-0.4.3.dist-info → oagi-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: oagi
-Version: 0.4.3
+Version: 0.5.0
 Summary: Official API of OpenAGI Foundation
 Project-URL: Homepage, https://github.com/agiopen-org/oagi
 Author-email: OpenAGI Foundation <contact@agiopen.org>

{oagi-0.4.3.dist-info → oagi-0.5.0.dist-info}/RECORD

@@ -1,19 +1,19 @@
 oagi/__init__.py,sha256=m-Z121YCIwQOPXpTC8kd_UIJizcX8QuHyrSSguQ0KE0,2187
-oagi/async_client.py,sha256=oDj4kIdtaV37uopoAeClCFQTxrYRwHV2HwMAcMdVYwE,8455
+oagi/async_client.py,sha256=JN2PAhM5pRTzjKxScQm-Fb_mfsEUwyXgFDJ_5w5kQv8,8916
 oagi/async_pyautogui_action_handler.py,sha256=F-lKyePCONWI03WnSxpX_QwxONbvnfdQu51wTod6mdw,1614
 oagi/async_screenshot_maker.py,sha256=pI-dbLcYOzcO1ffgTmozAdbYJQNBPKA7hmqj1RxEmIY,1688
-oagi/async_short_task.py,sha256=jvFTbmXTxFlkpAwmWeZlxbTSv_RB7V561hxw6gUcigw,1961
+oagi/async_short_task.py,sha256=7WM89H4V7xaPjuQQlzjA5Q7JOoJLhZw-DotJPNm8uAA,2517
 oagi/async_single_step.py,sha256=QawXO4GyfMz6O9jV8QBC1vKxFuS9vjKQxxJ1nwgHBzI,2838
-oagi/async_task.py,sha256=bclqtgg7mI2WAp-62jOz044tVk4wruycpn9NYDncnA8,4145
+oagi/async_task.py,sha256=z6q1OUjvbnXB5mTimAkM5kTFmixJA6zmWLPujx8vMHc,5013
 oagi/exceptions.py,sha256=VMwVS8ouE9nHhBpN3AZMYt5_U2kGcihWaTnBhoQLquo,1662
 oagi/logging.py,sha256=CWe89mA5MKTipIvfrqSYkv2CAFNBSwHMDQMDkG_g64g,1350
 oagi/pil_image.py,sha256=Zp7YNwyE_AT25ZEFsWKbzMxbO8JOQsJ1Espph5ye8k8,3804
 oagi/pyautogui_action_handler.py,sha256=8IFbU4p907L4b3TV3Eeh0-c-pYL2lYw-_qf1r8TtPTw,9811
 oagi/screenshot_maker.py,sha256=sVuW7jn-K4FmLhmYI-akdNI-UVcTeBzh9P1_qJhoq1s,1282
-oagi/short_task.py,sha256=9l1PDX70vDUEX2CIJ66yaAtb96P3mK_m95JffspnYFI,1779
+oagi/short_task.py,sha256=prrtUExDbiwabFqm6lOvKfFhLjQwANyWe2z7-xT995w,2323
 oagi/single_step.py,sha256=djhGOHzA5Y3-9_ity9QiJr_ObZZ04blSmNZsLXXXfkg,2939
-oagi/sync_client.py,sha256=E6EgFIe-H91rdsPhF1puwrBTpOnKaL6JA1WHR4R-CLY,9395
-oagi/task.py,sha256=JfsugIhBrwDmi1xOEVQdqmXsGFK-H4p17-B4rM8kbWs,4001
+oagi/sync_client.py,sha256=rgLjsZ2TnFidq5UcZddN83WNQ0HpfBlVihzG1j3XY28,9856
+oagi/task.py,sha256=yHM8QpevP8RitnFXGgyJttGRXB8mtFPSDB33E0MSIrg,4869
 oagi/types/__init__.py,sha256=YXxL-30f92qAf9U6LZuVCtKFG-Pi3xahKedaNxyrxFE,766
 oagi/types/action_handler.py,sha256=NH8E-m5qpGqWcXzTSWfF7W0Xdp8SkzJsbhCmQ0B96cg,1075
 oagi/types/async_action_handler.py,sha256=k1AaqSkFcXlxwW8sn-w0WFHGsIqHFLbcOPrkknmSVug,1116
@@ -24,7 +24,7 @@ oagi/types/models/__init__.py,sha256=bVzzGxb6lVxAQyJpy0Z1QknSe-xC3g4OIDr7t-p_3Ys
 oagi/types/models/action.py,sha256=hh6mRRSSWgrW4jpZo71zGMCOcZpV5_COu4148uG6G48,967
 oagi/types/models/image_config.py,sha256=tl6abVg_-IAPLwpaWprgknXu7wRWriMg-AEVyUX73v0,1567
 oagi/types/models/step.py,sha256=RSI4H_2rrUBq_xyCoWKaq7JHdJWNobtQppaKC1l0aWU,471
-oagi-0.4.3.dist-info/METADATA,sha256=fdp0bxPUM_ajJpP1pGxC8OVQy24kgvCQTHckH85XGZI,4799
-oagi-0.4.3.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-oagi-0.4.3.dist-info/licenses/LICENSE,sha256=sy5DLA2M29jFT4UfWsuBF9BAr3FnRkYtnAu6oDZiIf8,1075
-oagi-0.4.3.dist-info/RECORD,,
+oagi-0.5.0.dist-info/METADATA,sha256=MQHpJB8UgoliDCNk55ce0klXvkQHVpiw9AmCylAY_jQ,4799
+oagi-0.5.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+oagi-0.5.0.dist-info/licenses/LICENSE,sha256=sy5DLA2M29jFT4UfWsuBF9BAr3FnRkYtnAu6oDZiIf8,1075
+oagi-0.5.0.dist-info/RECORD,,