From fefdf04fb037444fde449c06babf6eba8486e2eb Mon Sep 17 00:00:00 2001 From: csgulati09 Date: Mon, 12 Aug 2024 13:53:28 +0530 Subject: [PATCH] fix: clean up --- portkey_ai/llms/langchain/portkey_langchain_callback.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/portkey_ai/llms/langchain/portkey_langchain_callback.py b/portkey_ai/llms/langchain/portkey_langchain_callback.py index 6b375cf..c45864f 100644 --- a/portkey_ai/llms/langchain/portkey_langchain_callback.py +++ b/portkey_ai/llms/langchain/portkey_langchain_callback.py @@ -253,7 +253,7 @@ def on_tool_end( self.event_array.append(self.event_map["tool_start_" + str(run_id)]) pass - def on_text( # Do we need to log this or not? This is just formatting of the text + def on_text( self, text: str, *, @@ -321,7 +321,6 @@ def on_agent_finish( end_time = int(datetime.now().timestamp()) total_time = (end_time - start_time) * 1000 - (self.main_span_id if parent_run_id is None else str(parent_run_id)) response_payload = self.on_agent_finish_transformer(finish) self.event_map["agent_action_" + str(run_id)]["response"] = response_payload self.event_map["agent_action_" + str(run_id)]["response"][