bug fixes to logging

This commit is contained in:
Krrish Dholakia 2023-08-28 15:36:57 -07:00
parent f5257e6eff
commit 80fa61cc0e
3 changed files with 2 additions and 2 deletions

View file

@ -1627,7 +1627,7 @@ class CustomStreamWrapper:
chunk = next(self.completion_stream)
completion_obj['content'] = chunk['choices']['delta']
# LOGGING
self.logging_obj(completion_obj["content"])
self.logging_obj.post_call(completion_obj["content"])
# return this for all models
return {"choices": [{"delta": completion_obj}]}