fix context bug

Author: aniketmaurya

src/litserve/loops/simple_loops.py

@@ -169,7 +169,7 @@ def run_batched_loop(
             response_queue_ids, uids, inputs = zip(*batches)
             num_inputs = len(inputs)
             try:
-                contexts = [{}] * num_inputs
+                contexts = [{} for _ in range(num_inputs)]
                 if hasattr(lit_spec, "populate_context"):
                     for input, context in zip(inputs, contexts):
                         lit_spec.populate_context(context, input)

src/litserve/loops/streaming_loops.py

@@ -155,7 +155,7 @@ def run_batched_streaming_loop(
             response_queue_ids, uids, inputs = zip(*batches)
             num_inputs = len(inputs)
             try:
-                contexts = [{}] * num_inputs
+                contexts = [{} for _ in range(num_inputs)]
                 if hasattr(lit_spec, "populate_context"):
                     for input, context in zip(inputs, contexts):
                         lit_spec.populate_context(context, input)