fixing test

Author: brnaba-aws

python/src/agent_squad/agents/strands_agent.py

@@ -15,7 +15,7 @@
 from strands.types.content import Messages
 from strands.agent.conversation_manager import ConversationManager
 from strands.types.traces import AttributeValue
-from strands.types.models import Model
+from strands.models.model import Model
 
 class StrandsAgent(Agent):
     """

python/src/tests/agents/test_strands_agent.py

@@ -13,7 +13,7 @@
 from agent_squad.agents.strands_agent import StrandsAgent
 from agent_squad.types import ConversationMessage, ParticipantRole
 from strands.agent.agent_result import AgentResult
-from strands.types.models import Model
+from strands.models.model import Model
 
 
 @pytest.fixture

python/test_requirements.txt

@@ -8,4 +8,4 @@ pytest-asyncio
 openai
 libsql-client
 ruff>=0.11.4
-strands-agents
+strands-agents==1.5.0