diff --git a/docs/my-website/docs/completion/mock_requests.md b/docs/my-website/docs/completion/mock_requests.md index ac02bc7060..19dd72f115 100644 --- a/docs/my-website/docs/completion/mock_requests.md +++ b/docs/my-website/docs/completion/mock_requests.md @@ -20,12 +20,9 @@ mock_completion(model=model, messages=messages) model = "gpt-3.5-turbo" messages = [{"role": "user", "content": "Hey, I'm a mock request"}] response = litellm.mock_completion(model=model, messages=messages, stream=True) -complete_response = "" for chunk in response: print(chunk) # {'choices': [{'delta': {'role': 'assistant', 'content': 'Thi'}, 'finish_reason': None}]} complete_response += chunk["choices"][0]["delta"]["content"] -if complete_response == "": - raise Exception("Empty response received") ``` ## set mock response @@ -35,12 +32,9 @@ You can also customize the mock response text returned. By default it's set to - model = "gpt-3.5-turbo" messages = [{"role": "user", "content": "Hey, I'm a mock request"}] response = litellm.mock_completion(model=model, messages=messages, mock_response="My custom mock response", stream=True) -complete_response = "" for chunk in response: print(chunk) # {'choices': [{'delta': {'role': 'assistant', 'content': 'My '}, 'finish_reason': None}]} complete_response += chunk["choices"][0]["delta"]["content"] -if complete_response == "": - raise Exception("Empty response received") ``` ## (Non-streaming) Mock Response Object