Skip to content

Commit 1409245

Browse files
authored
Change default "role" from "system" to "user" (#53)
This is a PR for #52. I am not sure if there are other implications but with that change it works for Gemma 2 9b IT.
1 parent f794711 commit 1409245

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/guidellm/backend/openai.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ async def make_request(
106106
stream = await self._async_client.chat.completions.create(
107107
model=self.model,
108108
messages=[
109-
{"role": "system", "content": request.prompt},
109+
{"role": "user", "content": request.prompt},
110110
],
111111
stream=True,
112112
**request_args,

tests/unit/backend/test_openai_backend.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,7 +171,7 @@ async def test_openai_backend_make_request(req, request_args, mock_openai_client
171171
assert backend._async_client.create_args == () # type: ignore
172172
assert backend._async_client.create_kwargs["model"] == "mock-model" # type: ignore
173173
assert backend._async_client.create_kwargs["messages"] == [ # type: ignore
174-
{"role": "system", "content": req.prompt}
174+
{"role": "user", "content": req.prompt}
175175
]
176176
assert backend._async_client.create_kwargs["stream"] # type: ignore
177177
assert backend._async_client.create_kwargs["n"] == 1 # type: ignore

0 commit comments

Comments
 (0)