Skip to content

Commit 4a48e49

Browse files
committed
Allow cost estimation on gpt-4-0613 (same as gpt-4-0314).
1 parent 9fa2a63 commit 4a48e49

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/gptcmd/message.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -339,7 +339,10 @@ async def asend(self) -> Union[Message, AioMessageStream]:
339339
def _post_send(self, resp, stream_cls: Type[S]) -> Union[Message, S]:
340340
"Handles API response objects generated by openai."
341341
if not self.stream:
342-
if resp.model != "gpt-4-0314" and resp.model.startswith("gpt-4"):
342+
if resp.model not in (
343+
"gpt-4-0314",
344+
"gpt-4-0613",
345+
) and resp.model.startswith("gpt-4"):
343346
self.prompt_tokens = None
344347
self.sampled_tokens = None
345348
if self.prompt_tokens is not None:

0 commit comments

Comments
 (0)