refactor: update inputs in ToolCallingAgentComponent and add astream_events setup
#4240
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
We need to wait for the merging of pull request PR #4166, as the tool calling agent component currently produces an error until that is completed. Once PR #4166 is merged, it will resolve the issue, allowing us to proceed with merging this pull request and implementing further improvements to the tool-calling agents. Plan of Action: |
…date descriptions
- Introduced `process_agent_events` method to handle asynchronous event processing for agent execution. - Added individual async handlers for different event types: `handle_chain_start`, `handle_chain_end`, `handle_chat_model_stream`, `handle_tool_start`, and `handle_tool_end`. - Updated agent execution to use `astream_events` for improved event handling. - Enhanced logging for better traceability of agent and tool execution events.
…ut naming conventions
ogabrielluiz
force-pushed
the
improved-tca
branch
from
f7f940a
to
984257b
Compare
ogabrielluiz
changed the title
ToolCallingAgentComponent to enhance input handlingToolCallingAgentComponent and add astream_events setup
dosubot
bot
added
size:M
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This pull request introduces several enhancements and refactorings to improve agent processing and streamline the codebase. The most significant changes include adding a new method for processing agent events, updating type imports, and modifying input handling in the
ToolCallingAgentComponent.Enhancements to agent processing:
src/backend/base/langflow/base/agents/agent.py: Addedprocess_agent_eventsmethod to handle different agent events asynchronously, improving the logging and processing of agent events.src/backend/base/langflow/base/agents/agent.py: Updatedrun_agentmethod to useprocess_agent_eventsfor better event handling and logging.Codebase simplification:
src/backend/base/langflow/base/agents/agent.py: Replacedtyping.castwithcollections.abc.AsyncIteratorfor more accurate type hinting.Input handling improvements:
src/backend/base/langflow/components/agents/tool_calling.py: ReplacedMultilineInputwithMessageTextInputforsystem_promptandinput_valueto better represent the input data.src/backend/base/langflow/components/agents/tool_calling.py: Updatedcreate_agent_runnableto useinput_valuedirectly and ensure tools are provided as an empty list if not specified.