code: def inference(self, *args):
      prefill, decode_ids = args[0]
  
      # Prefill requests
      results = {}
      for req_id in prefill:
          results[req_id] = self._run_prefill(req_id)
  
      # Decode the rest
      decode_result = self._run_decode(decode_ids) if decode_ids else {}
      results.update(decode_result)
      return [results[i] for i in self.context.request_ids.values()]
  
file_path: serve/test/pytest/test_data/streaming/stream_handler.py
imports: import json
  import logging
  import types
  import torch
  from transformers import AutoModelForCausalLM, AutoTokenizer
  from ts.torch_handler.base_handler import BaseHandler
label_desc: Use image node_id 3 for calling the StreamingHandler obj's underlying member method code with example usage: obj.inference() and returns: unknown