Skip to content

Commit 6a29f8e

Browse files
authored
cleanup: add type declaration for source in agent unit test, upgrade uv lock (#709)
* uv lock upgrade * type declaration cleanup for agent unit tests
1 parent 80e662a commit 6a29f8e

3 files changed

Lines changed: 38 additions & 36 deletions

File tree

tests/smoketests/sdk/test_agent.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
from runloop_api_client.sdk import RunloopSDK
88
from tests.smoketests.utils import unique_name
9+
from runloop_api_client.types.shared_params import AgentSource
910

1011
pytestmark = [pytest.mark.smoketest]
1112

@@ -112,7 +113,7 @@ def test_get_agent_by_id(self, sdk_client: RunloopSDK) -> None:
112113
@pytest.mark.timeout(THIRTY_SECOND_TIMEOUT)
113114
def test_list_multiple_agents(self, sdk_client: RunloopSDK) -> None:
114115
"""Test listing multiple agents after creation."""
115-
source_config = {
116+
source_config: AgentSource = {
116117
"type": "npm",
117118
"npm": {
118119
"package_name": "@runloop/hello-world-agent",

tests/smoketests/sdk/test_async_agent.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
from runloop_api_client.sdk import AsyncRunloopSDK
88
from tests.smoketests.utils import unique_name
9+
from runloop_api_client.types.shared_params import AgentSource
910

1011
pytestmark = [pytest.mark.smoketest, pytest.mark.asyncio]
1112

@@ -112,7 +113,7 @@ async def test_get_agent_by_id(self, async_sdk_client: AsyncRunloopSDK) -> None:
112113
@pytest.mark.timeout(THIRTY_SECOND_TIMEOUT)
113114
async def test_list_multiple_agents(self, async_sdk_client: AsyncRunloopSDK) -> None:
114115
"""Test listing multiple agents after creation."""
115-
source_config = {
116+
source_config: AgentSource = {
116117
"type": "npm",
117118
"npm": {
118119
"package_name": "@runloop/hello-world-agent",

0 commit comments

Comments
 (0)