Skip to content

Commit 720037d

Browse files
committed
Fixup GenerativeRequestSession
1 parent c5e9bd1 commit 720037d

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

src/guidellm/request/session.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from guidellm.backend.response import ResponseSummary
66
from guidellm.request.request import GenerationRequest
77

8-
__all__ = ["RequestSession"]
8+
__all__ = ["GenerativeRequestSession", "RequestSession"]
99

1010
# TODO: Replace with specific types that implement needed features
1111
RequestT = TypeVar("RequestT")
@@ -36,7 +36,7 @@ def __init__(self, prompts: list[GenerationRequest]) -> None:
3636
self.prompts = prompts
3737
self.responses: list[str] = []
3838

39-
def get_request(self) -> GenerationRequest:
39+
def get_next_request(self) -> GenerationRequest:
4040
completed_responses = len(self.responses)
4141
base_request = self.prompts[completed_responses].model_copy()
4242
base_request.content = "".join(
@@ -53,6 +53,9 @@ def get_request(self) -> GenerationRequest:
5353

5454
return base_request
5555

56+
def get_next_delay(self) -> float:
57+
return 0.0
58+
5659
def push_response(self, response: ResponseSummary) -> None:
5760
if len(self.responses) < len(self.prompts):
5861
if response.response_output_tokens is not None:

0 commit comments

Comments
 (0)