Merge pull request #4607 from maamalama/helicone-cohere

Helicone Headers & Cohere support
This commit is contained in:
Krish Dholakia 2024-07-11 22:01:44 -07:00 committed by GitHub
commit 5ad341d0ff
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 158 additions and 26 deletions

View file

@ -811,6 +811,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,
@ -818,6 +819,7 @@ class Logging:
start_time=start_time,
end_time=end_time,
print_verbose=print_verbose,
kwargs=kwargs,
)
if callback == "langfuse":
global langFuseLogger