Merge pull request #647 from nirga/traceloop-bug

fix: bugs in traceloop integration
2025-04-25 10:44:24 +00:00 · 2023-10-24 17:38:30 -07:00 · 2023-10-24 17:38:30 -07:00 · b86c679e71
commit b86c679e71
parent 54beb0f953 c21e86faec
4 changed files with 144 additions and 11 deletions
--- a/litellm/utils.py
+++ b/litellm/utils.py
@ -2411,6 +2411,15 @@ def handle_success(args, kwargs, result, start_time, end_time):
                        end_time=end_time,
                        print_verbose=print_verbose,
                    )
+                
+                elif callback == "traceloop":
+                    traceloopLogger.log_event(
+                        kwargs=kwargs,
+                        response_obj=result,
+                        start_time=start_time,
+                        end_time=end_time,
+                        print_verbose=print_verbose,
+                    )

                elif callback == "aispend":
                    print_verbose("reaches aispend for logging!")