Skip to content

Commit d53cfe2

Browse files
Update graphgen/models/llm/openai_client.py
Co-authored-by: Copilot <[email protected]>
1 parent 55692c5 commit d53cfe2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

graphgen/models/llm/openai_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,8 +39,8 @@ def __init__(
3939
seed: Optional[int] = None,
4040
topk_per_token: int = 5, # number of topk tokens to generate for each token
4141
request_limit: bool = False,
42-
rpm: RPM = None,
43-
tpm: TPM = None,
42+
rpm: Optional[RPM] = None,
43+
tpm: Optional[TPM] = None,
4444
**kwargs: Any,
4545
):
4646
super().__init__(**kwargs)

0 commit comments

Comments
 (0)