test: 修复路径导入错误，更新测试内容

Change-Id: I07823788f3f9c1ca4e431ff47412088d89f38ba0
Signed-off-by: OhYee <oyohyee@oyohyee.com>

Author: OhYee

tests/e2e/__test_model_async_template.py

@@ -18,7 +18,7 @@
 import os
 import time
 
-from pydantic_ai import ModelResponse
+from litellm.types.utils import ModelResponse
 import pydash
 import pytest
 

tests/e2e/__test_sandbox_template_async_template.py

@@ -13,6 +13,7 @@
 import datetime
 import time
 
+import pydash
 import pytest
 
 from agentrun.sandbox import Sandbox, Template
@@ -263,7 +264,10 @@ async def test_update_template_async(self, template_name: str):
             # assert template.description == "原始描述"
             assert template.cpu == 2.0
             assert template.memory == 2048
-            assert template.environment_variables == {"OLD_KEY": "old_value"}
+            assert (
+                pydash.get(template.environment_variables, "OLD_KEY")
+                == "old_value"
+            )
             assert template.created_at is not None
             created_at = datetime.datetime.strptime(
                 template.created_at, "%Y-%m-%dT%H:%M:%S.%f%z"

tests/e2e/test_model.py

@@ -28,7 +28,7 @@
 import os
 import time
 
-from pydantic_ai import ModelResponse
+from litellm.types.utils import ModelResponse
 import pydash
 import pytest
 

tests/e2e/test_sandbox_template.py

@@ -23,6 +23,7 @@
 import datetime
 import time
 
+import pydash
 import pytest
 
 from agentrun.sandbox import Sandbox, Template
@@ -472,7 +473,10 @@ async def test_update_template_async(self, template_name: str):
             # assert template.description == "原始描述"
             assert template.cpu == 2.0
             assert template.memory == 2048
-            assert template.environment_variables == {"OLD_KEY": "old_value"}
+            assert (
+                pydash.get(template.environment_variables, "OLD_KEY")
+                == "old_value"
+            )
             assert template.created_at is not None
             created_at = datetime.datetime.strptime(
                 template.created_at, "%Y-%m-%dT%H:%M:%S.%f%z"
@@ -561,7 +565,10 @@ def test_update_template(self, template_name: str):
             # assert template.description == "原始描述"
             assert template.cpu == 2.0
             assert template.memory == 2048
-            assert template.environment_variables == {"OLD_KEY": "old_value"}
+            assert (
+                pydash.get(template.environment_variables, "OLD_KEY")
+                == "old_value"
+            )
             assert template.created_at is not None
             created_at = datetime.datetime.strptime(
                 template.created_at, "%Y-%m-%dT%H:%M:%S.%f%z"