doc cleanup

This commit is contained in:
Krrish Dholakia
2023-09-11 16:41:49 -07:00
parent 54c9fb7f9e
commit f23b3d6ed1
@@ -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