Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
32 changes: 17 additions & 15 deletions veadk/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,14 +151,18 @@ async def wrapper(
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} {event_metadata}")
elif event.content is not None and event.content.parts:
for part in event.content.parts:
if len(part.text.strip()) > 0:
final_output = part.text
if part.thought:
logger.debug(
f"Thinking output: {final_output} {event_metadata}"
)
else:
logger.debug(
f"Event output: {final_output} {event_metadata}"
)

post_run_process(self)

Expand Down Expand Up @@ -502,13 +506,11 @@ async def event_generator():
yield event

async for event in event_generator():
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
if event.content is not None and event.content.parts:
for part in event.content.parts:
if not part.thought and len(part.text.strip()) > 0:
final_output = part.text
break
except LlmCallsLimitExceededError as e:
logger.warning(f"Max number of llm calls limit exceeded: {e}")
final_output = ""
Expand Down