linting type fixes

This commit is contained in:
ishaan-jaff 2023-08-18 11:36:06 -07:00
parent fbcb6d1c20
commit 5e7d22512d
6 changed files with 13 additions and 12 deletions

View file

@ -265,8 +265,8 @@ def completion(
or api_key
or litellm.replicate_key
)
# set replicate kye
os.environ["REPLICATE_API_TOKEN"] = replicate_key
# set replicate key
os.environ["REPLICATE_API_TOKEN"]: str = replicate_key
prompt = " ".join([message["content"] for message in messages])
input = {"prompt": prompt}
if "max_tokens" in optional_params: