add log methods to run

- adding updated log methods to run
pull/615/head
Sambhav Dixit 2 months ago committed by GitHub
parent 38db41b08c
commit 0d2d7fd439
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -783,6 +783,8 @@ class Agent:
or loop_count < self.max_loops
):
loop_count += 1
# Log step start
current_step_id = f"step_{loop_count}_{uuid.uuid4().hex}"
self.loop_count_print(loop_count, self.max_loops)
print("\n")
@ -837,6 +839,9 @@ class Agent:
# Convert to a str if the response is not a str
response = self.llm_output_parser(response)
# Log step metadata
step_meta = self.log_step_metadata(loop_count, task_prompt, response)
# Print
if self.streaming_on is True:
self.stream_response(response)

Loading…
Cancel
Save