Skip to content

Commit a70a40a

Browse files
committed
contrib/strands: shorten strands import paths
1 parent a4636e4 commit a70a40a

3 files changed

Lines changed: 5 additions & 7 deletions

File tree

temporalio/contrib/strands/_plugin.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,8 @@
22
from contextlib import asynccontextmanager
33
from dataclasses import replace
44

5-
from strands.models import Model
6-
from strands.models.bedrock import BedrockModel
7-
from strands.tools.mcp.mcp_client import MCPClient
5+
from strands.models import BedrockModel, Model
6+
from strands.tools.mcp import MCPClient
87

98
from temporalio.contrib.pydantic import pydantic_data_converter
109
from temporalio.converter import DataConverter, DefaultPayloadConverter

temporalio/contrib/strands/_temporal_mcp_client.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,10 +5,9 @@
55

66
from mcp import ClientSession
77
from mcp.types import PaginatedRequestParams, Tool
8-
from strands.tools.mcp.mcp_agent_tool import MCPAgentTool
9-
from strands.tools.mcp.mcp_client import MCPClient
8+
from strands.tools import ToolProvider
9+
from strands.tools.mcp import MCPAgentTool, MCPClient
1010
from strands.tools.mcp.mcp_types import MCPToolResult
11-
from strands.tools.tool_provider import ToolProvider
1211
from strands.types.tools import AgentTool
1312

1413
from temporalio import activity

temporalio/contrib/strands/workflow.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from datetime import timedelta
1010
from typing import Any, TypeVar
1111

12-
from strands.hooks.registry import BaseHookEvent, HookCallback
12+
from strands.hooks import BaseHookEvent, HookCallback
1313
from strands.types.tools import AgentTool
1414

1515
from temporalio import workflow

0 commit comments

Comments
 (0)