Skip to content

Conversation

anubhav756
Copy link
Contributor

@anubhav756 anubhav756 commented Jun 20, 2025

  • Remove client session management from the orchestration class

    • This is now managed by Toolbox SDK internally
  • Simplify prompt creation

    • This is handled by the respective tools
    • Tools descriptions, params, annotations, etc. are loaded from Toolbox
    • These are added through bind_tools from LangGraph
      • This enables removal of the custom response message creation (which was added as a TODO)
  • Add logged in user's token to the RunnableConfig

    • This is necessary so that the tools that require authentication can read the user's token if available
  • Simplify tools helper file by removing tools and helpers since those are now handled by Toolbox SDK internally

  • Add a Toolbox URL to connect to through integration tests

  • Removes ToolMessage while inserting ticket.

    • This was causing an issue with langchain-google-vertexai
    google.api_core.exceptions.InvalidArgument: 400 Please ensure that the number of function response parts should be equal to number of function call parts of the function call turn.
    
  • Remove unused human and AI messages post book ticket flow.

Diagram

image

Important

This PR depends on a couple of features from Toolbox SDK:

  • Support for optional parameters (#290)
  • Self-authenticated tools via RunnableConfig (#291)

Note

The failure in the integration test is expected. This PR is part of a series of changes, and the corresponding fix for this test is in a subsequent PR.

Reasoning

We've intentionally split the work into smaller, focused PRs to make the review process more manageable and efficient.

Merge Plan

All related PRs will be merged into the toolbox-main branch first. We will ensure all tests are passing on toolbox-main before merging the entire feature set into main.

@anubhav756 anubhav756 requested review from kurtisvg and Yuan325 June 20, 2025 14:00
@anubhav756 anubhav756 self-assigned this Jun 20, 2025
@anubhav756 anubhav756 requested a review from a team as a code owner June 20, 2025 14:00
@anubhav756 anubhav756 changed the title feat: Replace retrieval service with Toolbox feat: Integrate Toolbox SDK to replace Retrieval Service Jun 20, 2025
@anubhav756
Copy link
Contributor Author

/gcbrun

1 similar comment
@anubhav756
Copy link
Contributor Author

/gcbrun

This PR removes LangChain tools and Vertex AI Function Calling orchestration and consolidates on LangGraph.

It also flattens the directory structure and refactors to simplify some parts of code for easier understanding.
@anubhav756 anubhav756 force-pushed the anubhav-remove-orchs branch from 6763ed1 to 13051d3 Compare June 26, 2025 06:52
@anubhav756 anubhav756 force-pushed the anubhav-toolbox branch 2 times, most recently from c3b53e0 to 2cecab8 Compare June 26, 2025 11:01
@anubhav756 anubhav756 requested a review from Yuan325 June 29, 2025 04:19
Base automatically changed from anubhav-remove-orchs to toolbox-main July 3, 2025 06:27
@anubhav756 anubhav756 merged commit ac8669c into toolbox-main Jul 3, 2025
4 of 5 checks passed
@anubhav756 anubhav756 deleted the anubhav-toolbox branch July 3, 2025 08:45
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants