fix: fixing mypy linting errors and being backwards compatible for azure=true flag

This commit is contained in:
Krrish Dholakia 2023-10-05 22:36:32 -07:00
parent 25d8f45817
commit 060a2e40b2
7 changed files with 17 additions and 7 deletions

View file

@ -26,7 +26,7 @@ class AnthropicConfig():
to pass metadata to anthropic, it's {"user_id": "any-relevant-information"}
"""
max_tokens_to_sample: Optional[int]=256 # anthropic requires a default
stop_sequences: Optional[list[str]]=None
stop_sequences: Optional[list]=None
temperature: Optional[int]=None
top_p: Optional[int]=None
top_k: Optional[int]=None
@ -34,7 +34,7 @@ class AnthropicConfig():
def __init__(self,
max_tokens_to_sample: Optional[int]=256, # anthropic requires a default
stop_sequences: Optional[list[str]]=None,
stop_sequences: Optional[list]=None,
temperature: Optional[int]=None,
top_p: Optional[int]=None,
top_k: Optional[int]=None,