test: add more logging for failing test

This commit is contained in:
Krrish Dholakia 2024-03-12 11:15:14 -07:00
parent 84c7a5b693
commit d07c813ef9
2 changed files with 8 additions and 2 deletions

View file

@ -9373,7 +9373,9 @@ class CustomStreamWrapper:
else:
chunk = next(self.completion_stream)
if chunk is not None and chunk != b"":
print_verbose(f"PROCESSED CHUNK PRE CHUNK CREATOR: {chunk}")
print_verbose(
f"PROCESSED CHUNK PRE CHUNK CREATOR: {chunk}; custom_llm_provider: {self.custom_llm_provider}"
)
response: Optional[ModelResponse] = self.chunk_creator(chunk=chunk)
print_verbose(f"PROCESSED CHUNK POST CHUNK CREATOR: {response}")