fix(main.py): log hidden params for text completion calls

This commit is contained in:
Krrish Dholakia 2024-08-05 21:26:48 -07:00
parent 818a70838d
commit 7bf1b4d661
4 changed files with 23 additions and 24 deletions

View file

@ -371,6 +371,7 @@ class PredibaseChatCompletion(BaseLLM):
response_headers["llm_provider-{}".format(k)] = v
model_response._hidden_params["additional_headers"] = response_headers
return model_response
def completion(