Skip to content

Commit 8277ca5

Browse files
make gpt-5 work
1 parent 40b76c2 commit 8277ca5

1 file changed

Lines changed: 12 additions & 5 deletions

File tree

src/git_bob/_endpoints.py

Lines changed: 12 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -111,11 +111,18 @@ def prompt_openai(message: str, model="gpt-4o-2024-08-06", image=None, max_accum
111111
for _ in range(0, max_accumulated_responses):
112112

113113
# submit prompt
114-
response = client.chat.completions.create(
115-
model=model,
116-
messages=message,
117-
max_tokens=max_response_tokens,
118-
)
114+
if model.startswith("gpt-5"):
115+
response = client.chat.completions.create(
116+
model=model,
117+
messages=message,
118+
max_completion_tokens=max_response_tokens,
119+
)
120+
else:
121+
response = client.chat.completions.create(
122+
model=model,
123+
messages=message,
124+
max_tokens=max_response_tokens,
125+
)
119126

120127
result = append_result(result, response.choices[0].message.content)
121128
print("finish_reason", response.choices[0].finish_reason)

0 commit comments

Comments
 (0)