Skip to content

Commit 7a0bd44

Browse files
committed
fix deprecated utcnow usage
1 parent bc75b94 commit 7a0bd44

4 files changed

Lines changed: 10 additions & 10 deletions

File tree

agent/base_agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def __init__(self, agent_name: str, llm_client, team_mode: bool, turn_delay=0):
5757
self.max_consecutive_tools = 20
5858
self.group_chat_messages = []
5959
self.last_logged_index = 0 # Last index of the group chat messages that were logged
60-
self.last_log_time = datetime.datetime.utcnow().isoformat() # Last time a log was sent
60+
self.last_log_time = datetime.datetime.now(datetime.timezone.utc).isoformat() # Last time a log was sent
6161
self.turn_delay = turn_delay
6262

6363
self.name = agent_name
@@ -74,7 +74,7 @@ def check_and_send_work_log(self, conversation):
7474
# Check if there are new messages since the last log
7575
new_messages = conversation[self.last_logged_index:]
7676
first_timestamp = self.last_log_time
77-
last_timestamp = datetime.datetime.utcnow().isoformat()
77+
last_timestamp = datetime.datetime.now(datetime.timezone.utc).isoformat()
7878
success = send_work_log(self.name, new_messages, first_timestamp, last_timestamp)
7979
if success:
8080
self.steps_since_last_log = 0

group_chat/group_chat.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os
22
import threading
3-
from datetime import datetime
3+
from datetime import datetime, timezone
44
from typing import List
55

66
from fastapi import FastAPI
@@ -52,7 +52,7 @@ def load_messages():
5252
# Add an initial message from the "supervisor"
5353
initial_message = StoredMessage(
5454
username="supervisor",
55-
timestamp=datetime.utcnow().isoformat(),
55+
timestamp=datetime.now(timezone.utc).isoformat(),
5656
message=os.getenv("INITIAL_GROUP_CHAT_MESSAGE", INITIAL_MESSAGE),
5757
)
5858
messages.append(initial_message)
@@ -96,7 +96,7 @@ async def send_message(msg: Message):
9696
global message_count
9797
message_count += 1
9898

99-
now = datetime.utcnow().isoformat()
99+
now = datetime.now(timezone.utc).isoformat()
100100
stored = StoredMessage(username=msg.username, timestamp=now, message=msg.message)
101101
with lock:
102102
print(f"Received message by {msg.username}: {msg.message}")

group_work_log/group_work_log.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import threading
22
from contextlib import asynccontextmanager
3-
from datetime import datetime
3+
from datetime import datetime, timezone
44
import os
55
from typing import List, Optional, Dict, Any
66

@@ -80,7 +80,7 @@ def read_previous_summaries():
8080
if len(timespan_parts) >= 2:
8181
current_timestamp = timespan_parts[1].strip()
8282
else:
83-
current_timestamp = datetime.utcnow().isoformat()
83+
current_timestamp = datetime.now(timezone.utc).isoformat()
8484

8585
else:
8686
# Continue building the current summary
@@ -189,7 +189,7 @@ async def submit_worklog(request: WorklogRequest):
189189
if not messages:
190190
raise HTTPException(status_code=400, detail="Empty messages provided")
191191

192-
now = datetime.utcnow().isoformat()
192+
now = datetime.now(timezone.utc).isoformat()
193193
response = {"status": "ok"}
194194

195195
with lock:

tests/unit/group_work_log/test_group_work_log.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,10 +107,10 @@ def test_returns_empty_list_initially(self, client):
107107
def test_after_timestamp_filters_older_entries(self, client):
108108
client.post("/submit-worklog", json=_WORKLOG_PAYLOAD)
109109
import time
110-
from datetime import datetime
110+
from datetime import datetime, timezone
111111
time.sleep(0.01)
112112
# Always include microseconds to satisfy server-side strptime validation
113-
cutoff = datetime.utcnow().strftime("%Y-%m-%dT%H:%M:%S.%f")
113+
cutoff = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%S.%f")
114114
time.sleep(0.01)
115115
client.post("/submit-worklog", json={**_WORKLOG_PAYLOAD, "agent_name": "AgentB"})
116116
r = client.get(f"/summaries?after_timestamp={cutoff}")

0 commit comments

Comments
 (0)