case format

Author: root

fastdeploy/engine/common_engine.py

@@ -1726,11 +1726,11 @@ def _decode_token(self, token_ids, req_id, is_end):
                 token_ids = cum_tokens[prefix_offset:read_offset]
             else:
                 token_ids = []
-                
+
             if is_end and delta_text == "" and len(cum_tokens) > 0:
                 read_offset = self.data_processor.decode_status[req_id][1]
                 token_ids = cum_tokens[read_offset:]
-                
+
             if is_end:
                 del self.data_processor.decode_status[req_id]
         return delta_text, token_ids

tests/engine/test_decode_token.py

@@ -25,8 +25,7 @@
 from fastdeploy.input.text_processor import DataProcessor
 from fastdeploy.utils import envs
 
-
-MODEL_PATH = "/root/paddlejob/workspace/gzz/ERNIE-4.5-0.3B-Base-PT"
+MODEL_PATH = os.getenv("MODEL_PATH","") + "/ERNIE-4.5-0.3B-Paddle"
 
 
 class TestDecodeToken(unittest.TestCase):
@@ -74,18 +73,15 @@ def test_incremental_decoding_and_cleanup(self):
             delta_text, _ = self.engine._decode_token([], self.req_id, is_end=True)
             self._assert_cleaned_up()
 
-   
-
     def test_undecoded_tokens_on_end(self):
         """Test that tokens which produce no visible text during streaming
         are force-decoded when is_end=True"""
         with patch.object(envs, 'FD_ENABLE_RETURN_TEXT', True), \
              patch.dict(os.environ, {'DEBUG_DECODE': '1'}):
-            all_delta=""
-            
+            all_delta = ""
+
             delta_text, _ = self.engine._decode_token([109584], self.req_id, is_end=False)
             all_delta += delta_text
-                
 
             # Now end the stream - force decode should recover any remaining text
             delta_end, _ = self.engine._decode_token([109584], self.req_id, is_end=False)
@@ -99,8 +95,6 @@ def test_undecoded_tokens_on_end(self):
             self.assertEqual(token_ids, [109584, 109584, 109584])
             self._assert_cleaned_up()
 
-   
-
 
 if __name__ == "__main__":
     unittest.main()