headers and custom gateway

This commit is contained in:
maamalama 2024-07-05 20:17:38 -07:00
parent c99a5a58ec
commit 2c9560239a
2 changed files with 50 additions and 2 deletions

View file

@ -807,6 +807,7 @@ class Logging:
print_verbose("reaches helicone for logging!")
model = self.model
messages = self.model_call_details["input"]
kwargs = self.model_call_details
heliconeLogger.log_success(
model=model,
messages=messages,
@ -814,6 +815,7 @@ class Logging:
start_time=start_time,
end_time=end_time,
print_verbose=print_verbose,
kwargs=kwargs,
)
if callback == "langfuse":
global langFuseLogger