type declaration cleanup for agent unit tests

Author: sid-rl

tests/smoketests/sdk/test_agent.py

@@ -6,6 +6,7 @@
 
 from runloop_api_client.sdk import RunloopSDK
 from tests.smoketests.utils import unique_name
+from runloop_api_client.types.shared_params import AgentSource
 
 pytestmark = [pytest.mark.smoketest]
 
@@ -112,7 +113,7 @@ def test_get_agent_by_id(self, sdk_client: RunloopSDK) -> None:
     @pytest.mark.timeout(THIRTY_SECOND_TIMEOUT)
     def test_list_multiple_agents(self, sdk_client: RunloopSDK) -> None:
         """Test listing multiple agents after creation."""
-        source_config = {
+        source_config: AgentSource = {
             "type": "npm",
             "npm": {
                 "package_name": "@runloop/hello-world-agent",

tests/smoketests/sdk/test_async_agent.py

@@ -6,6 +6,7 @@
 
 from runloop_api_client.sdk import AsyncRunloopSDK
 from tests.smoketests.utils import unique_name
+from runloop_api_client.types.shared_params import AgentSource
 
 pytestmark = [pytest.mark.smoketest, pytest.mark.asyncio]
 
@@ -112,7 +113,7 @@ async def test_get_agent_by_id(self, async_sdk_client: AsyncRunloopSDK) -> None:
     @pytest.mark.timeout(THIRTY_SECOND_TIMEOUT)
     async def test_list_multiple_agents(self, async_sdk_client: AsyncRunloopSDK) -> None:
         """Test listing multiple agents after creation."""
-        source_config = {
+        source_config: AgentSource = {
             "type": "npm",
             "npm": {
                 "package_name": "@runloop/hello-world-agent",