diff --git a/docs/reference/vendors.md b/docs/reference/vendors.md
index 0ce84f3..073c0b6 100644
--- a/docs/reference/vendors.md
+++ b/docs/reference/vendors.md
@@ -554,7 +554,7 @@ For `nova-2` and `nova-3`, omit `api_key` to use Agora-managed credentials. For
 |---|---|---|---|---|
 | `api_key` | `str` | Yes | — | AssemblyAI API key |
 | `language` | `str` | Yes | — | Language code |
-| `uri` | `str` | No | `None` | AssemblyAI streaming WebSocket URL |
+| `ws_url` | `str` | No | `None` | AssemblyAI streaming WebSocket URL |
 | `additional_params` | `Dict[str, Any]` | No | `None` | Additional parameters |
 
 ### `AresSTT`
diff --git a/src/agora_agent/agentkit/vendors/stt.py b/src/agora_agent/agentkit/vendors/stt.py
index ab610a3..c276fca 100644
--- a/src/agora_agent/agentkit/vendors/stt.py
+++ b/src/agora_agent/agentkit/vendors/stt.py
@@ -195,7 +195,7 @@ class AssemblyAISTT(BaseSTT):
 
     api_key: str = Field(..., description="AssemblyAI API key")
     language: str = Field(..., description="Language code")
-    uri: Optional[str] = Field(default=None, description="AssemblyAI streaming WebSocket URL")
+    ws_url: Optional[str] = Field(default=None, description="AssemblyAI streaming WebSocket URL")
     additional_params: Optional[Dict[str, Any]] = Field(default=None)
 
     def to_config(self) -> Dict[str, Any]:
@@ -203,8 +203,8 @@ def to_config(self) -> Dict[str, Any]:
         params["api_key"] = self.api_key
         if self.language is not None:
             params["language"] = self.language
-        if self.uri is not None:
-            params["uri"] = self.uri
+        if self.ws_url is not None:
+            params["ws_url"] = self.ws_url
 
         config: Dict[str, Any] = {
             "vendor": "assemblyai",
diff --git a/tests/custom/test_stt_language.py b/tests/custom/test_stt_language.py
index 697d2ca..15dfd9d 100644
--- a/tests/custom/test_stt_language.py
+++ b/tests/custom/test_stt_language.py
@@ -163,12 +163,12 @@ def test_stt_vendor_params_match_documented_shapes() -> None:
         "language_code": "en-US",
     }
 
-    assemblyai_config = AssemblyAISTT(api_key="assembly-key", language="en-US", uri="wss://example.test/ws").to_config()
+    assemblyai_config = AssemblyAISTT(api_key="assembly-key", language="en-US", ws_url="wss://example.test/ws").to_config()
     assert "language" not in assemblyai_config
     assert assemblyai_config["params"] == {
         "api_key": "assembly-key",
         "language": "en-US",
-        "uri": "wss://example.test/ws",
+        "ws_url": "wss://example.test/ws",
     }
 
     assert MicrosoftSTT(key="ms-key", region="eastus", language="en-US").to_config()["params"] == {
@@ -193,7 +193,7 @@ def test_assemblyai_params_stay_nested_and_asr_language_comes_from_turn_detectio
         Agent(test_client(), turn_detection=TurnDetectionConfig(language="fr-FR"))
         .with_llm(OpenAI(api_key="llm-key", model="gpt-4o-mini", base_url="https://api.openai.com/v1/chat/completions"))
         .with_tts(ElevenLabsTTS(key="tts-key", voice_id="voice", model_id="eleven_flash_v2_5", base_url="wss://api.elevenlabs.io/v1"))
-        .with_stt(AssemblyAISTT(api_key="assembly-key", language="en-US", uri="wss://example.test/ws"))
+        .with_stt(AssemblyAISTT(api_key="assembly-key", language="en-US", ws_url="wss://example.test/ws"))
     )
 
     assert props["asr"] == {
@@ -202,7 +202,7 @@ def test_assemblyai_params_stay_nested_and_asr_language_comes_from_turn_detectio
         "params": {
             "api_key": "assembly-key",
             "language": "en-US",
-            "uri": "wss://example.test/ws",
+            "ws_url": "wss://example.test/ws",
         },
     }
     assert props["turn_detection"] == {"language": "fr-FR"}