feat: runner log change (#335)

* feat: runner log change

* feat: runner log change 2

Author: zakahan

veadk/runner.py

@@ -138,17 +138,23 @@ async def wrapper(
                 **kwargs,
             ):
                 yield event
+                event_metadata = f"| agent_name: {event.author} , user_id: {user_id} , session_id: {session_id} , invocation_id: {event.invocation_id}"
                 if event.get_function_calls():
                     for function_call in event.get_function_calls():
-                        logger.debug(f"Function call: {function_call}")
+                        logger.debug(f"Function call: {function_call} {event_metadata}")
+                elif event.get_function_responses():
+                    for function_response in event.get_function_responses():
+                        logger.debug(
+                            f"Function response: {function_response} {event_metadata}"
+                        )
                 elif (
                     event.content is not None
                     and event.content.parts
                     and event.content.parts[0].text is not None
                     and len(event.content.parts[0].text.strip()) > 0
                 ):
                     final_output = event.content.parts[0].text
-                    logger.debug(f"Event output: {final_output}")
+                    logger.debug(f"Event output: {final_output} {event_metadata}")
 
             post_run_process(self)
 
@@ -492,17 +498,13 @@ async def event_generator():
                         yield event
 
                 async for event in event_generator():
-                    if event.get_function_calls():
-                        for function_call in event.get_function_calls():
-                            logger.debug(f"Function call: {function_call}")
-                    elif (
+                    if (
                         event.content is not None
                         and event.content.parts
                         and event.content.parts[0].text is not None
                         and len(event.content.parts[0].text.strip()) > 0
                     ):
                         final_output = event.content.parts[0].text
-                        logger.debug(f"Event output: {final_output}")
             except LlmCallsLimitExceededError as e:
                 logger.warning(f"Max number of llm calls limit exceeded: {e}")
                 final_output = ""