chore: refine test cases (#621)

TEN-framework · Jan 26, 2025 · 358090e · 358090e
1 parent 6eafc31
commit 358090e
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 5 deletions.
diff --git a/..._set_prop_2_python_app/ten_packages/extension/default_async_extension_python/extension.py b/..._set_prop_2_python_app/ten_packages/extension/default_async_extension_python/extension.py
@@ -15,17 +15,19 @@
 class DefaultAsyncExtension(AsyncExtension):
 
     async def on_init(self, ten_env: AsyncTenEnv) -> None:
-        test_prop_json_str = await ten_env.get_property_to_json("test_prop")
+        test_prop_json_str = await ten_env.get_property_to_json(
+            "params.billing_type"
+        )
         test_prop_json = json.loads(test_prop_json_str)
-        assert test_prop_json == 32
+        assert test_prop_json == "agent_billing"
 
         test_prop_json_str = await ten_env.get_property_to_json()
         test_prop_json = json.loads(test_prop_json_str)
-        assert test_prop_json["test_prop"] == 32
+        assert test_prop_json["params"]["billing_type"] == "agent_billing"
 
         test_prop_json_str = await ten_env.get_property_to_json("")
         test_prop_json = json.loads(test_prop_json_str)
-        assert test_prop_json["test_prop"] == 32
+        assert test_prop_json["params"]["billing_type"] == "agent_billing"
 
     async def on_cmd(self, ten_env: AsyncTenEnv, cmd: Cmd) -> None:
         # Send a new command to other extensions and wait for the result. The

diff --git a/...set_prop_2_python_app/ten_packages/extension/default_async_extension_python/property.json b/...set_prop_2_python_app/ten_packages/extension/default_async_extension_python/property.json
@@ -1,3 +1,15 @@
 {
-  "test_prop": 32
+  "base_url": "xxx",
+  "api_key": "yyy",
+  "model": "qwen2.5-duplex-beta250106",
+  "temperature": 0.8,
+  "top_p": 0.8,
+  "prompt_version": "",
+  "language": "zh-CN",
+  "max_history": 50,
+  "force_threshold_ms": 3000,
+  "vendor": "openai",
+  "params": {
+    "billing_type": "agent_billing"
+  }
 }