mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-07 08:26:10 +00:00
docs(batching.md): add batch completion to docs
This commit is contained in:
parent
e3000504f9
commit
2ee599b848
1 changed files with 70 additions and 1 deletions
|
|
@ -1,3 +1,6 @@
|
|||
import Tabs from '@theme/Tabs';
|
||||
import TabItem from '@theme/TabItem';
|
||||
|
||||
# Batching Completion()
|
||||
LiteLLM allows you to:
|
||||
* Send many completion calls to 1 model
|
||||
|
|
@ -51,6 +54,9 @@ This makes parallel calls to the specified `models` and returns the first respon
|
|||
|
||||
Use this to reduce latency
|
||||
|
||||
<Tabs>
|
||||
<TabItem value="sdk" label="SDK">
|
||||
|
||||
### Example Code
|
||||
```python
|
||||
import litellm
|
||||
|
|
@ -68,8 +74,70 @@ response = batch_completion_models(
|
|||
print(result)
|
||||
```
|
||||
|
||||
|
||||
|
||||
</TabItem>
|
||||
<TabItem value="proxy" label="PROXY">
|
||||
|
||||
[how to setup proxy config](../proxy/configs.md)
|
||||
|
||||
Just pass a comma-separated string of model names and the flag `fastest_response=True`.
|
||||
|
||||
<Tabs>
|
||||
<TabItem value="curl" label="curl">
|
||||
|
||||
```bash
|
||||
|
||||
curl -X POST 'http://localhost:4000/chat/completions' \
|
||||
-H 'Content-Type: application/json' \
|
||||
-H 'Authorization: Bearer sk-1234' \
|
||||
-D '{
|
||||
"model": "gpt-3.5-turbo, bedrock-anthropic-claude-3", # 👈 Comma-separated models
|
||||
"messages": [
|
||||
{
|
||||
"role": "user",
|
||||
"content": "What'\''s the weather like in Boston today?"
|
||||
}
|
||||
],
|
||||
"stream": true,
|
||||
"fastest_response": true # 👈 FLAG
|
||||
}
|
||||
|
||||
'
|
||||
```
|
||||
|
||||
</TabItem>
|
||||
<TabItem value="openai" label="OpenAI SDK">
|
||||
|
||||
```python
|
||||
import openai
|
||||
client = openai.OpenAI(
|
||||
api_key="anything",
|
||||
base_url="http://0.0.0.0:4000"
|
||||
)
|
||||
|
||||
# request sent to model set on litellm proxy, `litellm --model`
|
||||
response = client.chat.completions.create(
|
||||
model="gpt-3.5-turbo, bedrock-anthropic-claude-3", # 👈 Comma-separated models
|
||||
messages = [
|
||||
{
|
||||
"role": "user",
|
||||
"content": "this is a test request, write a short poem"
|
||||
}
|
||||
],
|
||||
extra_body={"fastest_response": true} # 👈 FLAG
|
||||
)
|
||||
|
||||
print(response)
|
||||
```
|
||||
|
||||
</TabItem>
|
||||
</Tabs>
|
||||
</TabItem>
|
||||
</Tabs>
|
||||
|
||||
### Output
|
||||
Returns the first response
|
||||
Returns the first response in OpenAI format. Cancels other LLM API calls.
|
||||
```json
|
||||
{
|
||||
"object": "chat.completion",
|
||||
|
|
@ -95,6 +163,7 @@ Returns the first response
|
|||
}
|
||||
```
|
||||
|
||||
|
||||
## Send 1 completion call to many models: Return All Responses
|
||||
This makes parallel calls to the specified models and returns all responses
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue