feat: add kwargs support and StepWindowMemory

- Add kwargs field in model config for LiteLLM params (temperature, reasoning_effort, etc.)
- Add StepWindowMemory for step-based history trimming with image support
- Update config_example.yaml with detailed usage examples
- Bump version to 0.1.1

Author: SNHuan

easyagent/__init__.py

@@ -4,7 +4,7 @@
 from easyagent.memory import BaseMemory, SlidingWindowMemory, SummaryMemory
 from easyagent.tool import Tool, ToolManager, register_tool
 
-__version__ = "0.1.0"
+__version__ = "0.1.1"
 __all__ = [
     # Agent
     "ReactAgent",

easyagent/memory/__init__.py

@@ -1,6 +1,7 @@
 from easyagent.memory.base import BaseMemory
 from easyagent.memory.sliding_window import SlidingWindowMemory
+from easyagent.memory.step_window import StepWindowMemory
 from easyagent.memory.summary import SummaryMemory
 
-__all__ = ["BaseMemory", "SlidingWindowMemory", "SummaryMemory"]
+__all__ = ["BaseMemory", "SlidingWindowMemory", "StepWindowMemory", "SummaryMemory"]
 

easyagent/memory/step_window.py

@@ -0,0 +1,77 @@
+from __future__ import annotations
+
+from typing import List
+
+from easyagent.memory.base import BaseMemory
+from easyagent.model.schema import Message
+
+
+class StepWindowMemory(BaseMemory):
+    """Memory strategy that limits history by user message steps, with optional image trimming."""
+
+    def __init__(self, text_steps: int = 10, image_steps: int = 3) -> None:
+        self._text_steps = text_steps
+        self._image_steps = image_steps
+        self._messages: List[Message] = []
+
+    def add(self, message: Message) -> None:
+        self._messages.append(message)
+
+    def get_messages(self) -> list[Message]:
+        messages = self._trim_text(self._messages)
+        return self._trim_images(messages)
+
+    def clear(self) -> None:
+        self._messages.clear()
+
+    @property
+    def token_count(self) -> int:
+        return len(self._messages)
+
+    def _trim_text(self, messages: List[Message]) -> List[Message]:
+        if self._text_steps is None or self._text_steps <= 0:
+            return list(messages)
+        user_indices = [idx for idx, msg in enumerate(messages) if msg.role == "user"]
+        if len(user_indices) <= self._text_steps:
+            return list(messages)
+        start = user_indices[-self._text_steps]
+        return list(messages[start:])
+
+    def _trim_images(self, messages: List[Message]) -> List[Message]:
+        if self._image_steps is None:
+            return list(messages)
+
+        user_positions = [idx for idx, msg in enumerate(messages) if msg.role == "user"]
+        if self._image_steps <= 0:
+            cutoff = len(messages)
+        elif len(user_positions) > self._image_steps:
+            cutoff = user_positions[-self._image_steps]
+        else:
+            return list(messages)
+
+        trimmed: List[Message] = []
+        for idx, msg in enumerate(messages):
+            if msg.role == "user" and idx < cutoff:
+                trimmed.append(_strip_images(msg))
+            else:
+                trimmed.append(msg)
+        return trimmed
+
+
+def _strip_images(message: Message) -> Message:
+    content = message.content
+    if isinstance(content, list):
+        filtered = [
+            item
+            for item in content
+            if not (isinstance(item, dict) and item.get("type") == "image_url")
+        ]
+        return message.model_copy(update={"content": filtered})
+    if isinstance(content, dict):
+        cleaned = dict(content)
+        if "images" in cleaned:
+            cleaned["images"] = []
+        if "image_urls" in cleaned:
+            cleaned["image_urls"] = []
+        return message.model_copy(update={"content": cleaned})
+    return message

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "easy-agent-sdk"
-version = "0.1.0"
+version = "0.1.1"
 description = "A lightweight AI Agent framework built on LiteLLM with ReAct reasoning, tool calling, and smart memory."
 readme = "README.md"
 license = {text = "MIT"}