File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 5
5
from guidellm .backend .response import ResponseSummary
6
6
from guidellm .request .request import GenerationRequest
7
7
8
- __all__ = ["RequestSession" ]
8
+ __all__ = ["GenerativeRequestSession" , " RequestSession" ]
9
9
10
10
# TODO: Replace with specific types that implement needed features
11
11
RequestT = TypeVar ("RequestT" )
@@ -36,7 +36,7 @@ def __init__(self, prompts: list[GenerationRequest]) -> None:
36
36
self .prompts = prompts
37
37
self .responses : list [str ] = []
38
38
39
- def get_request (self ) -> GenerationRequest :
39
+ def get_next_request (self ) -> GenerationRequest :
40
40
completed_responses = len (self .responses )
41
41
base_request = self .prompts [completed_responses ].model_copy ()
42
42
base_request .content = "" .join (
@@ -53,6 +53,9 @@ def get_request(self) -> GenerationRequest:
53
53
54
54
return base_request
55
55
56
+ def get_next_delay (self ) -> float :
57
+ return 0.0
58
+
56
59
def push_response (self , response : ResponseSummary ) -> None :
57
60
if len (self .responses ) < len (self .prompts ):
58
61
if response .response_output_tokens is not None :
You can’t perform that action at this time.
0 commit comments