Skip to content

Commit 2c8221c

Browse files
committed
chore: revert unneeded file change
1 parent 43bca12 commit 2c8221c

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

tests/models/test_litellm_extra_body.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,12 @@ async def fake_acompletion(model, messages=None, **kwargs):
2222
captured.update(kwargs)
2323
msg = Message(role="assistant", content="ok")
2424
choice = Choices(index=0, message=msg)
25-
return ModelResponse(
26-
choices=[choice],
27-
usage=Usage(0, 0, 0),
28-
)
25+
return ModelResponse(choices=[choice], usage=Usage(0, 0, 0))
2926

3027
monkeypatch.setattr(litellm, "acompletion", fake_acompletion)
3128
settings = ModelSettings(
32-
temperature=0.1, extra_body={"cached_content": "some_cache", "foo": 123}
29+
temperature=0.1,
30+
extra_body={"cached_content": "some_cache", "foo": 123}
3331
)
3432
model = LitellmModel(model="test-model")
3533

0 commit comments

Comments
 (0)