Merge pull request #144 from yujonglee/fix-modelresponse-typing

Fix ModelResponse typing
This commit is contained in:
Krish Dholakia 2023-08-18 04:46:42 -07:00 committed by GitHub
commit 3d1934a829
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 11 deletions

View file

@ -40,7 +40,7 @@ def completion(
# model specific optional params
# used by text-bison only
top_k=40, request_timeout=0, # unused var for old version of OpenAI API
):
) -> ModelResponse:
try:
model_response = ModelResponse()
if azure: # this flag is deprecated, remove once notebooks are also updated.