Skip to content

Commit da00b95

Browse files
committed
style(adk): fix ruff format on Bedrock embedding tests
Wrap long mock_client.invoke_model lines to comply with ruff line-length formatting rules. Signed-off-by: Gonçalo Faustino <goncalo.santos@wellhub.com> Made-with: Cursor
1 parent 0e64cfd commit da00b95

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

python/packages/kagent-adk/tests/unittests/test_embedding.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,9 @@ async def test_bedrock_embed(self):
156156
svc = make_service(provider="bedrock", model="amazon.titan-embed-text-v1")
157157
vec = [0.1] * 1536
158158
mock_client = mock.MagicMock()
159-
mock_client.invoke_model = mock.MagicMock(return_value={"body": mock.MagicMock(read=lambda: b'{"embedding": ' + str(vec).encode() + b"}")})
159+
mock_client.invoke_model = mock.MagicMock(
160+
return_value={"body": mock.MagicMock(read=lambda: b'{"embedding": ' + str(vec).encode() + b"}")}
161+
)
160162

161163
with mock.patch("boto3.client", return_value=mock_client):
162164
result = await svc._generate_embedding_async("hello world")
@@ -168,7 +170,9 @@ async def test_bedrock_embed_uses_region_from_env(self):
168170
svc = make_service(provider="bedrock", model="amazon.titan-embed-text-v1")
169171
vec = [0.5] * 1536
170172
mock_client = mock.MagicMock()
171-
mock_client.invoke_model = mock.MagicMock(return_value={"body": mock.MagicMock(read=lambda: b'{"embedding": ' + str(vec).encode() + b"}")})
173+
mock_client.invoke_model = mock.MagicMock(
174+
return_value={"body": mock.MagicMock(read=lambda: b'{"embedding": ' + str(vec).encode() + b"}")}
175+
)
172176

173177
with (
174178
mock.patch.dict("os.environ", {"AWS_REGION": "eu-west-1", "AWS_DEFAULT_REGION": ""}),

0 commit comments

Comments
 (0)