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

Merge pull request #1112 from Undertone0809/add-cs-return-type-for-completion

Browse source 
refactor: add CustomStreamWrapper return type for completion
This commit is contained in:
Ishaan Jaff 2023-12-14 16:52:11 +05:30 • committed by GitHub
commit cbc8cb08c8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -285,7 +285,7 @@ def completion(
# Optional liteLLM function params
**kwargs,
) -> ModelResponse:
) -> Union[ModelResponse, CustomStreamWrapper]:
"""
Perform a completion() using any of litellm supported llms (example gpt-4, gpt-3.5-turbo, claude-2, command-nightly)
Parameters: