phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

fix setting mock response

Browse source
This commit is contained in:
Toni Engelhardt 2023-09-16 17:03:32 +01:00 • committed by GitHub
parent 9cf32ff670
commit 73bea9345f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -82,8 +82,7 @@ def mock_completion(model: str, messages: List, stream: bool = False, mock_respo
response = mock_completion_streaming_obj(model_response, mock_response=mock_response, model=model) response = mock_completion_streaming_obj(model_response, mock_response=mock_response, model=model)
return response return response
completion_response = "This is a mock request" model_response["choices"][0]["message"]["content"] = mock_response
model_response["choices"][0]["message"]["content"] = completion_response
model_response["created"] = time.time() model_response["created"] = time.time()
model_response["model"] = model model_response["model"] = model
return model_response return model_response