Fimeg/letta-server
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: patch .utcnow warning (#1702)

Browse Source
This commit is contained in:
Charles Packer
2025-04-14 12:55:34 -07:00
committed by GitHub
parent 90f138c829
commit 2ab267aa84
11 changed files with 26 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/personal_assistant_demo/google_calendar_test_setup.py
View File

@@ -40,7 +40,7 @@ def main():
service = build("calendar", "v3", credentials=creds)
# Call the Calendar API
now = datetime.datetime.utcnow().isoformat() + "Z" # 'Z' indicates UTC time
now = datetime.datetime.now(datetime.timezone.utc).isoformat() + "Z" # 'Z' indicates UTC time
print("Getting the upcoming 10 events")
events_result = (
service.events()

6
letta/groups/sleeptime_multi_agent.py
View File

@@ -1,6 +1,6 @@
import asyncio
import threading
from datetime import datetime
from datetime import datetime, timezone
from typing import List, Optional
from letta.agent import Agent, AgentState
@@ -154,7 +154,7 @@ class SleeptimeMultiAgent(Agent):
)
job_update = JobUpdate(
status=JobStatus.completed,
completed_at=datetime.utcnow(),
completed_at=datetime.now(timezone.utc),
metadata={
"result": result.model_dump(mode="json"),
"agent_id": participant_agent.agent_state.id,
@@ -165,7 +165,7 @@ class SleeptimeMultiAgent(Agent):
except Exception as e:
job_update = JobUpdate(
status=JobStatus.failed,
completed_at=datetime.utcnow(),
completed_at=datetime.now(timezone.utc),
metadata={"error": str(e)},
)
self.job_manager.update_job_by_id(job_id=run_id, job_update=job_update, actor=self.user)

4
letta/orm/base.py
View File

@@ -1,4 +1,4 @@
from datetime import datetime
from datetime import datetime, timezone
from typing import Optional
from sqlalchemy import Boolean, DateTime, String, func, text
@@ -25,7 +25,7 @@ class CommonSqlalchemyMetaMixins(Base):
timestamp (Optional[datetime]): The timestamp to set.
If None, uses the current UTC time.
"""
self.updated_at = timestamp or datetime.utcnow()
self.updated_at = timestamp or datetime.now(timezone.utc)
def _set_created_and_updated_by_fields(self, actor_id: str) -> None:
"""Populate created_by_id and last_updated_by_id based on actor."""

6
letta/server/rest_api/routers/v1/agents.py
View File

@@ -1,6 +1,6 @@
import json
import traceback
from datetime import datetime
from datetime import datetime, timezone
from typing import Annotated, Any, List, Optional
from fastapi import APIRouter, BackgroundTasks, Body, Depends, File, Header, HTTPException, Query, UploadFile, status
@@ -729,7 +729,7 @@ async def process_message_background(
# Update job status to completed
job_update = JobUpdate(
status=JobStatus.completed,
completed_at=datetime.utcnow(),
completed_at=datetime.now(timezone.utc),
metadata={"result": result.model_dump(mode="json")}, # Store the result in metadata
)
server.job_manager.update_job_by_id(job_id=job_id, job_update=job_update, actor=actor)
@@ -738,7 +738,7 @@ async def process_message_background(
# Update job status to failed
job_update = JobUpdate(
status=JobStatus.failed,
completed_at=datetime.utcnow(),
completed_at=datetime.now(timezone.utc),
metadata={"error": str(e)},
)
server.job_manager.update_job_by_id(job_id=job_id, job_update=job_update, actor=actor)

4
letta/services/passage_manager.py
View File

@@ -1,4 +1,4 @@
from datetime import datetime
from datetime import datetime, timezone
from typing import List, Optional
from openai import OpenAI
@@ -49,7 +49,7 @@ class PassageManager:
"organization_id": data["organization_id"],
"metadata_": data.get("metadata", {}),
"is_deleted": data.get("is_deleted", False),
"created_at": data.get("created_at", datetime.utcnow()),
"created_at": data.get("created_at", datetime.now(timezone.utc)),
}
if "agent_id" in data and data["agent_id"]:

4
tests/integration_test_summarizer.py
View File

@@ -1,7 +1,7 @@
import json
import os
import uuid
from datetime import datetime
from datetime import datetime, timezone
from typing import List
import pytest
@@ -55,7 +55,7 @@ def generate_message(role: str, text: str = None, tool_calls: List = None) -> Me
id="message-" + str(uuid.uuid4()),
role=MessageRole(role),
content=[TextContent(text=text or f"{role} message text")],
created_at=datetime.utcnow(),
created_at=datetime.now(timezone.utc),
tool_calls=tool_calls or [],
)

8
tests/test_llm_clients.py
View File

@@ -1,4 +1,4 @@
from datetime import datetime
from datetime import datetime, timezone
from unittest.mock import AsyncMock, patch
import pytest
@@ -35,10 +35,12 @@ def mock_agent_messages():
return {
"agent-1": [
PydanticMessage(
role=MessageRole.system, content=[{"type": "text", "text": "You are a helpful assistant."}], created_at=datetime.utcnow()
role=MessageRole.system,
content=[{"type": "text", "text": "You are a helpful assistant."}],
created_at=datetime.now(timezone.utc),
),
PydanticMessage(
role=MessageRole.user, content=[{"type": "text", "text": "What's the weather like?"}], created_at=datetime.utcnow()
role=MessageRole.user, content=[{"type": "text", "text": "What's the weather like?"}], created_at=datetime.now(timezone.utc)
),
]
}

4
tests/test_managers.py
View File

@@ -1768,7 +1768,7 @@ def test_agent_list_passages_filtering(server, default_user, sarah_agent, defaul
assert len(source_filtered) == 3
# Test date filtering
now = datetime.utcnow()
now = datetime.now(timezone.utc)
future_date = now + timedelta(days=1)
past_date = now - timedelta(days=1)
@@ -4302,7 +4302,7 @@ def test_job_messages_pagination(server: SyncServer, default_run, default_user,
def test_job_messages_ordering(server: SyncServer, default_run, default_user, sarah_agent):
"""Test that messages are ordered by created_at."""
# Create messages with different timestamps
base_time = datetime.utcnow()
base_time = datetime.now(timezone.utc)
message_times = [
base_time - timedelta(minutes=2),
base_time - timedelta(minutes=1),

4
tests/test_static_buffer_summarize.py
View File

@@ -1,5 +1,5 @@
import json
from datetime import datetime
from datetime import datetime, timezone
from unittest.mock import AsyncMock
import pytest
@@ -31,7 +31,7 @@ def messages():
Message(
role=MessageRole.user if i % 2 == 0 else MessageRole.assistant,
content=[TextContent(type="text", text=json.dumps({"message": f"Test message {i}"}))],
created_at=datetime.utcnow(),
created_at=datetime.now(timezone.utc),
)
for i in range(15)
]

4
tests/test_v1_routes.py
View File

@@ -1,4 +1,4 @@
from datetime import datetime
from datetime import datetime, timezone
from unittest.mock import MagicMock, Mock
import pytest
@@ -209,7 +209,7 @@ def test_upsert_base_tools(client, mock_sync_server, add_integers_tool):
def test_get_run_messages(client, mock_sync_server):
"""Test getting messages for a run."""
# Create properly formatted mock messages
current_time = datetime.utcnow()
current_time = datetime.now(timezone.utc)
mock_messages = [
UserMessage(
id=f"message-{i:08x}",

5
tests/utils.py
View File

@@ -1,7 +1,6 @@
import datetime
import os
import time
from datetime import datetime
from datetime import datetime, timezone
from importlib import util
from typing import Dict, Iterator, List, Tuple
@@ -34,7 +33,7 @@ class DummyDataConnector(DataConnector):
file_size=0, # Set to 0 as a placeholder
file_creation_date="1970-01-01", # Placeholder date
file_last_modified_date="1970-01-01", # Placeholder date
created_at=datetime.utcnow(),
created_at=datetime.now(timezone.utc),
)
self.file_to_text[file_metadata.id] = text