⚡️ Speed up method AgentMemory.get_messages by 27%

[codeflash-ai]

📄 27% (0.27x) speedup for AgentMemory.get_messages in libs/agno/agno/memory/agent.py

⏱️ Runtime : 1.75 microseconds → 1.37 microsecond (best of 97 runs)

📝 Explanation and details

To optimize the get_messages method, we can precompute the dictionary representations of the messages if the messages are not expected to change frequently. This way, we avoid recomputation each time get_messages is called. Here’s a revised version of your code.

✅ Correctness verification report:

Test	Status
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	✅ 4 Passed
⏪ Replay Tests	🔘 None Found
🔎 Concolic Coverage Tests	✅ 2 Passed
📊 Tests Coverage	100.0%

🌀 Generated Regression Tests Details

from __future__ import annotations

from typing import Any, Dict, List, Optional

# imports
import pytest  # used for our unit tests
from agno.memory.agent import AgentMemory
from agno.memory.classifier import MemoryClassifier
from agno.memory.db import MemoryDb
from agno.memory.manager import MemoryManager
from agno.memory.memory import Memory, MemoryRetrieval
from agno.memory.summarizer import MemorySummarizer
from agno.memory.summary import SessionSummary
from agno.models.message import Message
from pydantic import BaseModel, ConfigDict

# unit tests

# Basic Functionality


def test_empty_messages_list():
    # Empty messages list
    agent_memory = AgentMemory(messages=[])
    codeflash_output = agent_memory.get_messages()












from __future__ import annotations

from typing import Any, Dict, List, Optional

# imports
import pytest  # used for our unit tests
from agno.memory.agent import AgentMemory
from agno.memory.classifier import MemoryClassifier
from agno.memory.db import MemoryDb
from agno.memory.manager import MemoryManager
from agno.memory.memory import Memory, MemoryRetrieval
from agno.memory.summarizer import MemorySummarizer
from agno.memory.summary import SessionSummary
from agno.models.message import Message
from pydantic import BaseModel, ConfigDict


# unit tests
def test_empty_messages():
    # Test with no messages
    agent_memory = AgentMemory(messages=[])
    codeflash_output = agent_memory.get_messages()















from agno.memory.agent import AgentMemory

def test_AgentMemory_get_messages():
    AgentMemory.get_messages(AgentMemory())

To edit these changes git checkout codeflash/optimize-AgentMemory.get_messages-m94ou5wr and push.