PR fix

Author: VerioN1

decart/realtime/client.py

@@ -79,11 +79,9 @@ def _emit_error(self, error: DecartSDKError) -> None:
             except Exception as e:
                 logger.exception(f"Error in error callback: {e}")
 
-    async def set_prompt(self, prompt: str, enrich: bool = True, max_timeout: float = 15.0) -> None:
+    async def set_prompt(self, prompt: str, enrich: bool = True) -> None:
         if not prompt or not prompt.strip():
             raise InvalidInputError("Prompt cannot be empty")
-        if max_timeout <= 0 or max_timeout > 60:
-            raise InvalidInputError("max_timeout must be between 0 and 60 seconds")
 
         event, result = self._manager.register_prompt_wait(prompt)
 
@@ -93,7 +91,7 @@ async def set_prompt(self, prompt: str, enrich: bool = True, max_timeout: float
             )
 
             try:
-                await asyncio.wait_for(event.wait(), timeout=max_timeout)
+                await asyncio.wait_for(event.wait(), timeout=15.0)
             except asyncio.TimeoutError:
                 raise DecartSDKError("Prompt acknowledgment timed out")
 

tests/test_realtime_unit.py

@@ -220,8 +220,10 @@ def register_prompt_wait(prompt):
 
         from decart.errors import DecartSDKError
 
-        with pytest.raises(DecartSDKError) as exc_info:
-            await realtime_client.set_prompt("New prompt", max_timeout=0.01)
+        # Mock asyncio.wait_for to immediately raise TimeoutError
+        with patch("asyncio.wait_for", side_effect=asyncio.TimeoutError):
+            with pytest.raises(DecartSDKError) as exc_info:
+                await realtime_client.set_prompt("New prompt")
 
         assert "timed out" in str(exc_info.value)
         mock_manager.unregister_prompt_wait.assert_called_with("New prompt")