phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) linting errors

Browse source
This commit is contained in:
ishaan-jaff 2023-10-31 14:43:10 -07:00
parent b301e4ead7
commit 4875af17a1
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/main.py
View file

@ -525,7 +525,7 @@ def completion(
# https://platform.openai.com/docs/api-reference/completions/create
prompt = messages[0]["content"]
else:
prompt = " ".join([message["content"] for message in messages])
prompt = " ".join([message["content"] for message in messages]) # type: ignore
## LOGGING
logging.pre_call(
input=prompt,
@ -1188,7 +1188,7 @@ def completion(
model_response["choices"][0]["message"]["content"] = response_string
model_response["created"] = time.time()
model_response["model"] = "ollama/" + model
prompt_tokens = len(encoding.encode(prompt))
prompt_tokens = len(encoding.encode(prompt)) # type: ignore
completion_tokens = len(encoding.encode(response_string))
model_response["usage"] = {
"prompt_tokens": prompt_tokens,
@ -1284,7 +1284,7 @@ def completion(
)
"""
prompt = " ".join([message["content"] for message in messages])
prompt = " ".join([message["content"] for message in messages]) # type: ignore
resp = requests.post(url, json={
'model': model,
'params': {