RAG and Tool usage , at the same time about langflow HOT 2 OPEN

from langchain.tools.retriever import create_retriever_tool
from langflow import CustomComponent
from langflow.field_typing import BaseRetriever, Tool

class RetrieverToolComponent(CustomComponent):
    display_name = "RetrieverTool"
    description = "Tool for interacting with retriever"

    def build_config(self):
        return {
            "retriever": {
                "display_name": "Retriever",
                "info": "Retriever to interact with",
                "type": BaseRetriever,
            },
            "name": {"display_name": "Name", "info": "Name of the tool"},
            "description": {"display_name": "Description", "info": "Description of the tool"},
        }

    def build(
        self,
        retriever: BaseRetriever,
        name: str,
        description: str,
    ) -> Tool:
        return create_retriever_tool(
            retriever=retriever,
            name=name,
            description=description,
        )

def get_mobile_data_balance(mobile_number):
    # Logic to get mobile data balance
    balance = "23GB"  # Example balance
    return balance

def send_msg(mobile_number):
    # Logic to send OTP
    otp = "8822"  # Example OTP
    return otp

def otp_validation_and_return_data(otp):
    # Logic to validate OTP and return data balance
    if otp == "8822":
        return "Your data balance is 23GB"
    else:
        return "Invalid OTP"

from langflow.graph.state_manager import GraphStateManager

state_manager = GraphStateManager()

def conversation_flow(user_input, user_id):
    state = state_manager.get_state(user_id)
    
    if state == "awaiting_mobile_number":
        mobile_number = user_input
        send_msg(mobile_number)
        state_manager.update_state(user_id, "awaiting_otp")
        return "Please enter the OTP sent to you"
    
    elif state == "awaiting_otp":
        otp = user_input
        response = otp_validation_and_return_data(otp)
        state_manager.update_state(user_id, "default")
        return response
    
    elif "data balance" in user_input:
        state_manager.update_state(user_id, "awaiting_mobile_number")
        return "Please enter your mobile number"
    
    else:
        # Handle general broadband-related queries using RAG model
        response = retriever_tool.retrieve(user_input)
        return response

# Example integration
retriever_tool = RetrieverToolComponent().build(
    retriever=my_retriever_instance,
    name="MyRetrieverTool",
    description="A tool to retrieve information"
)

# Simulate a user input
user_id = "12345"
user_input = "Can you tell me my data balance?"
response = conversation_flow(user_input, user_id)
print(response)  # Output: Please enter your mobile number