Skip to content

Commit

Permalink
fix: clean up
Browse files Browse the repository at this point in the history
  • Loading branch information
csgulati09 committed Aug 12, 2024
1 parent 6693afe commit fefdf04
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions portkey_ai/llms/langchain/portkey_langchain_callback.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
*,
Expand Down Expand Up @@ -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"][
Expand Down

0 comments on commit fefdf04

Please sign in to comment.