Skip to content

Commit

Permalink
Python 3.9 friendly UTC timezone.
Browse files Browse the repository at this point in the history
  • Loading branch information
gperetin committed Jun 13, 2024
1 parent 78215ac commit a6f54c8
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
6 changes: 3 additions & 3 deletions core/agents/executor.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from datetime import UTC, datetime
from datetime import datetime, timezone
from typing import Optional

from pydantic import BaseModel, Field
Expand Down Expand Up @@ -96,13 +96,13 @@ async def run(self) -> AgentResponse:
self.next_state.action = f'Skip "{cmd_name}"'
return AgentResponse.done(self)

started_at = datetime.now(UTC)
started_at = datetime.now(timezone.utc)

log.info(f"Running command `{cmd}` with timeout {timeout}s")
status_code, stdout, stderr = await self.process_manager.run_command(cmd, timeout=timeout)
llm_response = await self.check_command_output(cmd, timeout, stdout, stderr, status_code)

duration = (datetime.now(UTC) - started_at).total_seconds()
duration = (datetime.now(timezone.utc) - started_at).total_seconds()

self.complete()
self.next_state.action = f'Run "{cmd_name}"'
Expand Down
4 changes: 2 additions & 2 deletions core/proc/exec_log.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
from datetime import UTC, datetime
from datetime import datetime, timezone
from typing import Optional

from pydantic import BaseModel, Field


class ExecLog(BaseModel):
started_at: datetime = Field(default_factory=lambda: datetime.now(UTC))
started_at: datetime = Field(default_factory=lambda: datetime.now(timezone.utc))
duration: float = Field(description="The duration of the command/process run in seconds")
cmd: str = Field(description="The full command (as executed in the shell)")
cwd: str = Field(description="The working directory for the command (relative to project root)")
Expand Down

0 comments on commit a6f54c8

Please sign in to comment.