From 9d5518753a192dc0e5431f909a013b2bc656f3d0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniele=20Trifir=C3=B2?= <dtrifiro@redhat.com>
Date: Tue, 2 Jul 2024 15:03:28 +0200
Subject: [PATCH] fix _convert_output usage

related to #15 (973d9ef)
---
 src/vllm_tgis_adapter/grpc/grpc_server.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/vllm_tgis_adapter/grpc/grpc_server.py b/src/vllm_tgis_adapter/grpc/grpc_server.py
index efbb2136..14bc022e 100644
--- a/src/vllm_tgis_adapter/grpc/grpc_server.py
+++ b/src/vllm_tgis_adapter/grpc/grpc_server.py
@@ -366,10 +366,10 @@ async def GenerateStream(
             last_response = self._convert_output(
                 output,
                 resp_options,
-                max_is_tok_limit,
-                time_limit_reached,
-                last_output_length,
-                last_token_count,
+                max_is_token_limit=max_is_tok_limit,
+                time_limit_reached=time_limit_reached,
+                text_start_offset=last_output_length,
+                token_start_offset=last_token_count,
             )
             yield last_response