zrb 1.8.13__tar.gz → 1.8.15__tar.gz

{zrb-1.8.13 → zrb-1.8.15}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: zrb
-Version: 1.8.13
+Version: 1.8.15
 Summary: Your Automation Powerhouse
 Home-page: https://github.com/state-alchemists/zrb
 License: AGPL-3.0-or-later
@@ -25,6 +25,7 @@ Requires-Dist: libcst (>=1.7.0,<2.0.0)
 Requires-Dist: openai (>=1.86.0,<2.0.0) ; extra == "rag" or extra == "all"
 Requires-Dist: pdfplumber (>=0.11.6,<0.12.0) ; extra == "rag" or extra == "all"
 Requires-Dist: playwright (>=1.53.0,<2.0.0) ; extra == "playwright" or extra == "all"
+Requires-Dist: prompt-toolkit (>=3.0.51,<4.0.0)
 Requires-Dist: psutil (>=7.0.0,<8.0.0)
 Requires-Dist: pydantic-ai (>=0.3.4,<0.4.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)

{zrb-1.8.13 → zrb-1.8.15}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zrb"
-version = "1.8.13"
+version = "1.8.15"
 description = "Your Automation Powerhouse"
 authors = ["Go Frendi Gunawan <gofrendiasgard@gmail.com>"]
 license = "AGPL-3.0-or-later"
@@ -55,6 +55,7 @@ openai = {version = "^1.86.0", optional = true}
 playwright = {version = "^1.53.0", optional = true}
 psutil = "^7.0.0"
 pyjwt = "^2.10.1"
+prompt-toolkit = "^3.0.51"
 
 [tool.poetry.extras]
 # poetry install -E rag

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/builtin/llm/chat_session.py

@@ -25,6 +25,8 @@ async def read_user_prompt(ctx: AnyContext) -> str:
     Returns:
         The final result from the LLM session.
     """
+    from prompt_toolkit import PromptSession
+
     _show_info(ctx)
     final_result = ""
     ctx.print(stylize_faint("🧑 >> ") + f"{ctx.input.message}", plain=True)
@@ -41,10 +43,12 @@ async def read_user_prompt(ctx: AnyContext) -> str:
         return final_result
     multiline_mode = False
     user_inputs = []
+    user_input_session = PromptSession()
     while True:
         await asyncio.sleep(0.01)
         ctx.print(stylize_faint("🧑 >> "), end="", plain=True)
-        user_input = input()
+        user_input = await user_input_session.prompt_async()
+        # user_input = input()
         # Handle special input
         if user_input.strip().lower() in ("/bye", "/quit"):
             user_prompt = "\n".join(user_inputs)
@@ -53,9 +57,9 @@ async def read_user_prompt(ctx: AnyContext) -> str:
             if result is not None:
                 final_result = result
             break
-        elif user_input.strip().lower() in ("/multi"):
+        elif user_input.strip().lower() in ("/multi",):
             multiline_mode = True
-        elif user_input.strip().lower() in ("/end"):
+        elif user_input.strip().lower() in ("/end",):
             multiline_mode = False
             user_prompt = "\n".join(user_inputs)
             user_inputs = []

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/builtin/llm/tool/sub_agent.py

@@ -3,23 +3,17 @@ from collections.abc import Callable
 from textwrap import dedent
 from typing import TYPE_CHECKING, Any, Coroutine
 
-if TYPE_CHECKING:
-    from pydantic_ai import Tool
-    from pydantic_ai.mcp import MCPServer
-    from pydantic_ai.models import Model
-    from pydantic_ai.settings import ModelSettings
-else:
-    Tool = Any
-    MCPServer = Any
-    Model = Any
-    ModelSettings = Any
-
 from zrb.context.any_context import AnyContext
 from zrb.task.llm.agent import create_agent_instance, run_agent_iteration
 from zrb.task.llm.config import get_model, get_model_settings
 from zrb.task.llm.prompt import get_combined_system_prompt
 
 if TYPE_CHECKING:
+    from pydantic_ai import Tool
+    from pydantic_ai.mcp import MCPServer
+    from pydantic_ai.models import Model
+    from pydantic_ai.settings import ModelSettings
+
     ToolOrCallable = Tool | Callable
 else:
     ToolOrCallable = Any
@@ -29,10 +23,10 @@ def create_sub_agent_tool(
     tool_name: str,
     tool_description: str,
     system_prompt: str | None = None,
-    model: str | Model | None = None,
-    model_settings: ModelSettings | None = None,
+    model: "str | Model | None" = None,
+    model_settings: "ModelSettings | None" = None,
     tools: list[ToolOrCallable] = [],
-    mcp_servers: list[MCPServer] = [],
+    mcp_servers: list["MCPServer"] = [],
 ) -> Callable[[AnyContext, str], Coroutine[Any, Any, str]]:
     """
     Create an LLM "sub-agent" tool function for use by a main LLM agent.

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/builtin/todo.py

@@ -11,7 +11,6 @@ from zrb.input.text_input import TextInput
 from zrb.task.make_task import make_task
 from zrb.util.file import read_file, write_file
 from zrb.util.todo import (
-    TodoTaskModel,
     add_duration,
     cascade_todo_task,
     get_visual_todo_card,
@@ -72,6 +71,8 @@ def _get_filter_input(
     alias="add",
 )
 def add_todo(ctx: AnyContext):
+    from zrb.util.todo_model import TodoTaskModel
+
     todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
@@ -108,6 +109,8 @@ def add_todo(ctx: AnyContext):
     alias="list",
 )
 def list_todo(ctx: AnyContext):
+    from zrb.util.todo_model import TodoTaskModel
+
     todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
@@ -123,6 +126,8 @@ def list_todo(ctx: AnyContext):
     alias="show",
 )
 def show_todo(ctx: AnyContext):
+    from zrb.util.todo_model import TodoTaskModel
+
     todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     todo_list: list[TodoTaskModel] = []
@@ -157,6 +162,8 @@ def show_todo(ctx: AnyContext):
     alias="complete",
 )
 def complete_todo(ctx: AnyContext):
+    from zrb.util.todo_model import TodoTaskModel
+
     todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
@@ -187,6 +194,8 @@ def complete_todo(ctx: AnyContext):
     alias="archive",
 )
 def archive_todo(ctx: AnyContext):
+    from zrb.util.todo_model import TodoTaskModel
+
     todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):
@@ -248,6 +257,8 @@ def archive_todo(ctx: AnyContext):
     alias="log",
 )
 def log_todo(ctx: AnyContext):
+    from zrb.util.todo_model import TodoTaskModel
+
     todo_file_path = os.path.join(CFG.TODO_DIR, "todo.txt")
     todo_list: list[TodoTaskModel] = []
     if os.path.isfile(todo_file_path):

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/llm_config.py

@@ -1,15 +1,12 @@
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
+
+from zrb.config import CFG
 
 if TYPE_CHECKING:
     from pydantic_ai.models import Model
     from pydantic_ai.providers import Provider
     from pydantic_ai.settings import ModelSettings
-else:
-    Model = Any
-    ModelSettings = Any
-    Provider = Any
 
-from zrb.config import CFG
 
 DEFAULT_PERSONA = (
     "You are a helpful and precise expert assistant. Your goal is to follow "
@@ -217,9 +214,9 @@ class LLMConfig:
         default_history_summarization_threshold: int | None = None,
         default_enrich_context: bool | None = None,
         default_context_enrichment_threshold: int | None = None,
-        default_model: Model | None = None,
-        default_model_settings: ModelSettings | None = None,
-        default_model_provider: Provider | None = None,
+        default_model: "Model | None" = None,
+        default_model_settings: "ModelSettings | None" = None,
+        default_model_provider: "Provider | None" = None,
     ):
         self._default_model_name = default_model_name
         self._default_model_base_url = default_base_url
@@ -266,13 +263,13 @@ class LLMConfig:
             return CFG.LLM_API_KEY
 
     @property
-    def default_model_settings(self) -> ModelSettings | None:
+    def default_model_settings(self) -> "ModelSettings | None":
         if self._default_model_settings is not None:
             return self._default_model_settings
         return None
 
     @property
-    def default_model_provider(self) -> Provider | str:
+    def default_model_provider(self) -> "Provider | str":
         if self._default_model_provider is not None:
             return self._default_model_provider
         if self.default_model_base_url is None and self.default_model_api_key is None:
@@ -332,7 +329,7 @@ class LLMConfig:
         return DEFAULT_CONTEXT_ENRICHMENT_PROMPT
 
     @property
-    def default_model(self) -> Model | str | None:
+    def default_model(self) -> "Model | str | None":
         if self._default_model is not None:
             return self._default_model
         model_name = self.default_model_name
@@ -396,10 +393,10 @@ class LLMConfig:
     def set_default_model_base_url(self, model_base_url: str):
         self._default_model_base_url = model_base_url
 
-    def set_default_model_provider(self, provider: Provider | str):
+    def set_default_model_provider(self, provider: "Provider | str"):
         self._default_model_provider = provider
 
-    def set_default_model(self, model: Model | str):
+    def set_default_model(self, model: "Model | str"):
         self._default_model = model
 
     def set_default_summarize_history(self, summarize_history: bool):
@@ -418,7 +415,7 @@ class LLMConfig:
     ):
         self._default_context_enrichment_threshold = context_enrichment_threshold
 
-    def set_default_model_settings(self, model_settings: ModelSettings):
+    def set_default_model_settings(self, model_settings: "ModelSettings"):
         self._default_model_settings = model_settings
 
 

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/runner/cli.py

@@ -7,7 +7,6 @@ from zrb.context.shared_context import SharedContext
 from zrb.group.any_group import AnyGroup
 from zrb.group.group import Group
 from zrb.runner.common_util import get_run_kwargs
-from zrb.runner.web_app import create_web_app
 from zrb.runner.web_auth_config import web_auth_config
 from zrb.session.session import Session
 from zrb.session_state_logger.session_state_logger_factory import session_state_logger
@@ -187,6 +186,8 @@ server_group = cli.add_group(
 async def start_server(_: AnyContext):
     from uvicorn import Config, Server
 
+    from zrb.runner.web_app import create_web_app
+
     app = create_web_app(cli, web_auth_config, session_state_logger)
     server = Server(
         Config(

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/runner/web_auth_config.py

@@ -1,9 +1,11 @@
-from typing import Callable
+from typing import TYPE_CHECKING, Callable
 
 from zrb.config import CFG
-from zrb.runner.web_schema.user import User
 from zrb.task.any_task import AnyTask
 
+if TYPE_CHECKING:
+    from zrb.runner.web_schema.user import User
+
 
 class WebAuthConfig:
     def __init__(
@@ -18,7 +20,7 @@ class WebAuthConfig:
         super_admin_password: str | None = None,
         guest_username: str | None = None,
         guest_accessible_tasks: list[AnyTask | str] = [],
-        find_user_by_username: Callable[[str], User | None] | None = None,
+        find_user_by_username: Callable[[str], "User | None"] | None = None,
     ):
         self._secret_key = secret_key
         self._access_token_expire_minutes = access_token_expire_minutes
@@ -29,7 +31,7 @@ class WebAuthConfig:
         self._super_admin_username = super_admin_username
         self._super_admin_password = super_admin_password
         self._guest_username = guest_username
-        self._user_list = []
+        self._user_list: list["User"] = []
         self._guest_accessible_tasks = guest_accessible_tasks
         self._find_user_by_username = find_user_by_username
 
@@ -92,7 +94,9 @@ class WebAuthConfig:
         return self._guest_accessible_tasks
 
     @property
-    def default_user(self) -> User:
+    def default_user(self) -> "User":
+        from zrb.runner.web_schema.user import User
+
         if self.enable_auth:
             return User(
                 username=self.guest_username,
@@ -108,7 +112,9 @@ class WebAuthConfig:
         )
 
     @property
-    def super_admin(self) -> User:
+    def super_admin(self) -> "User":
+        from zrb.runner.web_schema.user import User
+
         return User(
             username=self.super_admin_username,
             password=self.super_admin_password,
@@ -116,7 +122,7 @@ class WebAuthConfig:
         )
 
     @property
-    def user_list(self) -> list[User]:
+    def user_list(self) -> list["User"]:
         if not self.enable_auth:
             return [self.default_user]
         return self._user_list + [self.super_admin, self.default_user]
@@ -152,11 +158,11 @@ class WebAuthConfig:
         self._guest_accessible_tasks = tasks
 
     def set_find_user_by_username(
-        self, find_user_by_username: Callable[[str], User | None]
+        self, find_user_by_username: Callable[[str], "User | None"]
     ):
         self._find_user_by_username = find_user_by_username
 
-    def append_user(self, user: User):
+    def append_user(self, user: "User"):
         duplicates = [
             existing_user
             for existing_user in self.user_list
@@ -166,7 +172,7 @@ class WebAuthConfig:
             raise ValueError(f"User already exists {user.username}")
         self._user_list.append(user)
 
-    def find_user_by_username(self, username: str) -> User | None:
+    def find_user_by_username(self, username: str) -> "User | None":
         user = None
         if self._find_user_by_username is not None:
             user = self._find_user_by_username(username)

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/runner/web_route/task_session_api_route.py

@@ -9,16 +9,18 @@ from zrb.runner.web_auth_config import WebAuthConfig
 from zrb.runner.web_schema.session import NewSessionResponse
 from zrb.runner.web_util.user import get_user_from_request
 from zrb.session.session import Session
-from zrb.session_state_log.session_state_log import SessionStateLog, SessionStateLogList
 from zrb.session_state_logger.any_session_state_logger import AnySessionStateLogger
 from zrb.task.any_task import AnyTask
 from zrb.util.group import NodeNotFoundError, extract_node_from_args, get_node_path
 
 if TYPE_CHECKING:
-    # We want fastapi to only be loaded when necessary to decrease footprint
-
     from fastapi import FastAPI
 
+    from zrb.session_state_log.session_state_log import (
+        SessionStateLog,
+        SessionStateLogList,
+    )
+
 
 def serve_task_session_api(
     app: "FastAPI",
@@ -30,6 +32,11 @@ def serve_task_session_api(
     from fastapi import Query, Request
     from fastapi.responses import JSONResponse
 
+    from zrb.session_state_log.session_state_log import (
+        SessionStateLog,
+        SessionStateLogList,
+    )
+
     @app.post("/api/v1/task-sessions/{path:path}")
     async def create_new_task_session_api(
         path: str,
@@ -110,8 +117,10 @@ def serve_task_session_api(
 
 
 def sanitize_session_state_log_list(
-    task: AnyTask, session_state_log_list: SessionStateLogList
-) -> SessionStateLogList:
+    task: AnyTask, session_state_log_list: "SessionStateLogList"
+) -> "SessionStateLogList":
+    from zrb.session_state_log.session_state_log import SessionStateLogList
+
     return SessionStateLogList(
         total=session_state_log_list.total,
         data=[
@@ -122,8 +131,8 @@ def sanitize_session_state_log_list(
 
 
 def sanitize_session_state_log(
-    task: AnyTask, session_state_log: SessionStateLog
-) -> SessionStateLog:
+    task: AnyTask, session_state_log: "SessionStateLog"
+) -> "SessionStateLog":
     """
     In session, we create snake_case aliases of inputs.
     The purpose was to increase ergonomics, so that user can use `input.system_prompt`
@@ -131,6 +140,8 @@ def sanitize_session_state_log(
     However, when we serve the session through HTTP API,
     we only want to show the original input names.
     """
+    from zrb.session_state_log.session_state_log import SessionStateLog
+
     enhanced_inputs = session_state_log.input
     real_inputs = {}
     for real_input in task.inputs:

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/session/any_session.py

@@ -5,13 +5,14 @@ from typing import TYPE_CHECKING, Any, Coroutine, TypeVar
 
 from zrb.context.any_context import AnyContext
 from zrb.group.any_group import AnyGroup
-from zrb.session_state_log.session_state_log import SessionStateLog
 from zrb.session_state_logger.any_session_state_logger import AnySessionStateLogger
 from zrb.task_status.task_status import TaskStatus
 
 if TYPE_CHECKING:
-    from zrb.context import any_shared_context
-    from zrb.task import any_task
+    from zrb.context.any_shared_context import AnySharedContext
+    from zrb.session_state_log.session_state_log import SessionStateLog
+    from zrb.task.any_task import AnyTask
+
 
 TAnySession = TypeVar("TAnySession", bound="AnySession")
 
@@ -44,7 +45,7 @@ class AnySession(ABC):
 
     @property
     @abstractmethod
-    def shared_ctx(self) -> any_shared_context.AnySharedContext:
+    def shared_ctx(self) -> "AnySharedContext":
         """Shared context for this session"""
         pass
 
@@ -83,16 +84,16 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def set_main_task(self, main_task: any_task.AnyTask):
+    def set_main_task(self, main_task: "AnyTask"):
         """Set main task"""
         pass
 
     @abstractmethod
-    def as_state_log(self) -> SessionStateLog:
+    def as_state_log(self) -> "SessionStateLog":
         pass
 
     @abstractmethod
-    def get_ctx(self, task: any_task.AnyTask) -> AnyContext:
+    def get_ctx(self, task: "AnyTask") -> AnyContext:
         """Retrieves the context for a specific task.
 
         Args:
@@ -104,7 +105,7 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def defer_monitoring(self, task: any_task.AnyTask, coro: Coroutine):
+    def defer_monitoring(self, task: "AnyTask", coro: Coroutine):
         """Defers the execution of a task's monitoring coroutine for later processing.
 
         Args:
@@ -114,7 +115,7 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def defer_action(self, task: any_task.AnyTask, coro: Coroutine):
+    def defer_action(self, task: "AnyTask", coro: Coroutine):
         """Defers the execution of a task's coroutine for later processing.
 
         Args:
@@ -138,7 +139,7 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def register_task(self, task: any_task.AnyTask):
+    def register_task(self, task: "AnyTask"):
         """Registers a new task in the session.
 
         Args:
@@ -147,7 +148,7 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def get_root_tasks(self, task: any_task.AnyTask) -> list[any_task.AnyTask]:
+    def get_root_tasks(self, task: "AnyTask") -> list["AnyTask"]:
         """Retrieves the list of root tasks that should be executed first
         to run the given task.
 
@@ -160,7 +161,7 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def get_next_tasks(self, task: any_task.AnyTask) -> list[any_task.AnyTask]:
+    def get_next_tasks(self, task: "AnyTask") -> list["AnyTask"]:
         """Retrieves the list of tasks that should be executed after the given task.
 
         Args:
@@ -172,7 +173,7 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def get_task_status(self, task: any_task.AnyTask) -> TaskStatus:
+    def get_task_status(self, task: "AnyTask") -> TaskStatus:
         """Get tasks' status.
 
         Args:
@@ -184,7 +185,7 @@ class AnySession(ABC):
         pass
 
     @abstractmethod
-    def is_allowed_to_run(self, task: any_task.AnyTask):
+    def is_allowed_to_run(self, task: "AnyTask"):
         """Determines if the specified task is allowed to run based on its current state.
 
         Args:

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/session/session.py

@@ -1,15 +1,10 @@
 import asyncio
-from typing import Any, Coroutine
+from typing import TYPE_CHECKING, Any, Coroutine
 
 from zrb.context.any_shared_context import AnySharedContext
 from zrb.context.context import AnyContext, Context
 from zrb.group.any_group import AnyGroup
 from zrb.session.any_session import AnySession
-from zrb.session_state_log.session_state_log import (
-    SessionStateLog,
-    TaskStatusHistoryStateLog,
-    TaskStatusStateLog,
-)
 from zrb.session_state_logger.any_session_state_logger import AnySessionStateLogger
 from zrb.session_state_logger.session_state_logger_factory import session_state_logger
 from zrb.task.any_task import AnyTask
@@ -32,6 +27,9 @@ from zrb.util.group import get_node_path
 from zrb.util.string.name import get_random_name
 from zrb.xcom.xcom import Xcom
 
+if TYPE_CHECKING:
+    from zrb.session_state_log.session_state_log import SessionStateLog
+
 
 class Session(AnySession):
     def __init__(
@@ -139,7 +137,13 @@ class Session(AnySession):
         main_task_path = get_node_path(self._root_group, main_task)
         self._main_task_path = [] if main_task_path is None else main_task_path
 
-    def as_state_log(self) -> SessionStateLog:
+    def as_state_log(self) -> "SessionStateLog":
+        from zrb.session_state_log.session_state_log import (
+            SessionStateLog,
+            TaskStatusHistoryStateLog,
+            TaskStatusStateLog,
+        )
+
         task_status_log: dict[str, TaskStatusStateLog] = {}
         log_start_time = ""
         for task, task_status in self._task_status.items():

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/session_state_logger/any_session_state_logger.py

@@ -1,16 +1,21 @@
 import datetime
 from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
 
-from zrb.session_state_log.session_state_log import SessionStateLog, SessionStateLogList
+if TYPE_CHECKING:
+    from zrb.session_state_log.session_state_log import (
+        SessionStateLog,
+        SessionStateLogList,
+    )
 
 
 class AnySessionStateLogger(ABC):
     @abstractmethod
-    def write(self, session_log: SessionStateLog):
+    def write(self, session_log: "SessionStateLog"):
         pass
 
     @abstractmethod
-    def read(self, session_name: str) -> SessionStateLog:
+    def read(self, session_name: str) -> "SessionStateLog":
         pass
 
     @abstractmethod
@@ -21,5 +26,5 @@ class AnySessionStateLogger(ABC):
         max_start_time: datetime.datetime,
         page: int = 0,
         limit: int = 10,
-    ) -> SessionStateLogList:
+    ) -> "SessionStateLogList":
         pass

{zrb-1.8.13 → zrb-1.8.15}/src/zrb/session_state_logger/file_session_state_logger.py

@@ -1,16 +1,22 @@
 import datetime
 import os
+from typing import TYPE_CHECKING
 
-from zrb.session_state_log.session_state_log import SessionStateLog, SessionStateLogList
 from zrb.session_state_logger.any_session_state_logger import AnySessionStateLogger
 from zrb.util.file import read_file, write_file
 
+if TYPE_CHECKING:
+    from zrb.session_state_log.session_state_log import (
+        SessionStateLog,
+        SessionStateLogList,
+    )
+
 
 class FileSessionStateLogger(AnySessionStateLogger):
     def __init__(self, session_log_dir: str):
         self._session_log_dir = session_log_dir
 
-    def write(self, session_log: SessionStateLog):
+    def write(self, session_log: "SessionStateLog"):
         session_file_path = self._get_session_file_path(session_log.name)
         session_dir_path = os.path.dirname(session_file_path)
         if not os.path.isdir(session_dir_path):
@@ -22,7 +28,9 @@ class FileSessionStateLogger(AnySessionStateLogger):
         timeline_dir_path = self._get_timeline_dir_path(session_log)
         write_file(os.path.join(timeline_dir_path, session_log.name), "")
 
-    def read(self, session_name: str) -> SessionStateLog:
+    def read(self, session_name: str) -> "SessionStateLog":
+        from zrb.session_state_log.session_state_log import SessionStateLog
+
         session_file_path = self._get_session_file_path(session_name)
         return SessionStateLog.model_validate_json(read_file(session_file_path))
 
@@ -33,7 +41,9 @@ class FileSessionStateLogger(AnySessionStateLogger):
         max_start_time: datetime.datetime,
         page: int = 0,
         limit: int = 10,
-    ) -> SessionStateLogList:
+    ) -> "SessionStateLogList":
+        from zrb.session_state_log.session_state_log import SessionStateLogList
+
         matching_sessions = []
         # Traverse the timeline directory and filter sessions
         timeline_dir = os.path.join(self._session_log_dir, "_timeline", *task_path)
@@ -62,7 +72,7 @@ class FileSessionStateLogger(AnySessionStateLogger):
     def _get_session_file_path(self, session_name: str) -> str:
         return os.path.join(self._session_log_dir, f"{session_name}.json")
 
-    def _get_timeline_dir_path(self, session_log: SessionStateLog) -> str:
+    def _get_timeline_dir_path(self, session_log: "SessionStateLog") -> str:
         start_time = self._get_start_time(session_log)
         year = start_time.year
         month = start_time.month
@@ -80,7 +90,7 @@ class FileSessionStateLogger(AnySessionStateLogger):
         ]
         return os.path.join(self._session_log_dir, "_timeline", *paths)
 
-    def _get_start_time(self, session_log: SessionStateLog) -> datetime.datetime:
+    def _get_start_time(self, session_log: "SessionStateLog") -> datetime.datetime:
         return datetime.datetime.strptime(
             session_log.start_time, "%Y-%m-%d %H:%M:%S.%f"
         )