Closed tomusher closed 4 weeks ago
In some cases, such as when passing a mock_response to completion, LiteLLM returns a generator rather than a CustomStreamWrapper.
mock_response
completion
CustomStreamWrapper
This doesn't seem to break anything except for tests, so this just adds a test and a fix for it - likely only useful for future tests.
Looks like this only works with LiteLLM 1.41.15, so tests will pass once #76 is merged
In some cases, such as when passing a
mock_response
tocompletion
, LiteLLM returns a generator rather than aCustomStreamWrapper
.This doesn't seem to break anything except for tests, so this just adds a test and a fix for it - likely only useful for future tests.