Technologic101/imagineui

1from nodes.design_rag import DesignRAG2from langgraph.graph import MessagesState3from langchain_core.messages import SystemMessage4 5# this should be done at app level is used elsewhere6rag = DesignRAG()7 8async def design_retriever_tool(state: MessagesState, num_examples: int = 1):9    """10        Retrieves similar designs based on style requirements11        Name: query_similar_designs12    """13    14    result = await rag.query_similar_designs(state["messages"], num_examples)15 16    return SystemMessage(content=result)17 18