phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

bug fixes to logging

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

BIN
litellm/__pycache__/utils.cpython-311.pyc
View file

Binary file not shown.

2
litellm/tests/test_litedebugger_integration.py
View file

@ -22,7 +22,7 @@ response = completion(model="claude-instant-1", messages=[{"role": "user", "cont
# print(f"response: {response}") # print(f"response: {response}")
# # Test 2: On embedding call # # Test 2: On embedding call
# response = embedding(model="text-embedding-ada-002", input=["sample text"]) response = embedding(model="text-embedding-ada-002", input=["sample text"])
# print(f"response: {response}") # print(f"response: {response}")
# # Test 3: On streaming completion call # # Test 3: On streaming completion call

2
litellm/utils.py
View file

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