Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: assistants agent improvements #5581

Open
wants to merge 9 commits into
base: main
Choose a base branch
from

Conversation

phact
Copy link
Collaborator

@phact phact commented Jan 8, 2025

This PR homogenizes the assistants manager component with the new Agent component and makes it compatible with existing Langflow tools. Code changes include the addition of a new utility function for wrapping tools, updates to the AstraAssistantManager class to support new input types and outputs, and the introduction of file handling and vector store integration.

Utility Function Enhancements:

  • Added wrap_base_tool_as_tool_interface function to dynamically wrap BaseTool instances with ToolInterface, including schema handling and Pydantic model creation.

Astra Assistant Manager Updates:

  • Introduced new input types (HandleInput, FileInput) and updated existing ones in AstraAssistantManager to support more flexible tool and file handling. [1] [2]
  • Added new output Vector Store Id and corresponding methods to manage vector store integration. [1] [2]
  • Enhanced process_inputs method to handle file uploads, create vector stores, and manage tool usage events. [1] [2]

@dosubot dosubot bot added the size:L This PR changes 100-499 lines, ignoring generated files. label Jan 8, 2025
@phact phact requested a review from ogabrielluiz January 8, 2025 06:57
@dosubot dosubot bot added the enhancement New feature or request label Jan 8, 2025
@phact phact changed the title Assistants agent improvements feat: assistants agent improvements Jan 8, 2025
@github-actions github-actions bot added enhancement New feature or request and removed enhancement New feature or request labels Jan 8, 2025
@phact phact enabled auto-merge January 8, 2025 07:09
Copy link

codspeed-hq bot commented Jan 8, 2025

CodSpeed Performance Report

Merging #5581 will not alter performance

Comparing assistants_agent_improvements (6611609) with main (e2ff7b3)

Summary

✅ 15 untouched benchmarks

@github-actions github-actions bot added enhancement New feature or request and removed enhancement New feature or request labels Jan 8, 2025
@phact phact self-assigned this Jan 8, 2025
@github-actions github-actions bot added enhancement New feature or request and removed enhancement New feature or request labels Jan 8, 2025
@github-actions github-actions bot added enhancement New feature or request and removed enhancement New feature or request labels Jan 10, 2025
@phact
Copy link
Collaborator Author

phact commented Jan 10, 2025

Flow
image

Tool call
image

File Search
image

@github-actions github-actions bot added enhancement New feature or request and removed enhancement New feature or request labels Jan 10, 2025
@github-actions github-actions bot added enhancement New feature or request and removed enhancement New feature or request labels Jan 11, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request size:L This PR changes 100-499 lines, ignoring generated files.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant