fix(utils.py): support image gen logging to langfuse

This commit is contained in:
Krrish Dholakia 2024-02-16 16:12:52 -08:00
parent 5f9e141d1e
commit f57483ea70
5 changed files with 58 additions and 91 deletions

View file

@ -741,7 +741,7 @@ class AzureChatCompletion(BaseLLM):
response = azure_client.images.generate(**data, timeout=timeout) # type: ignore
## LOGGING
logging_obj.post_call(
input=input,
input=prompt,
api_key=api_key,
additional_args={"complete_input_dict": data},
original_response=response,