Spaces:
Runtime error
Runtime error
File size: 522 Bytes
125ffe9 4698077 125ffe9 7ec8434 8c57a85 125ffe9 4698077 e9ce170 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from nodes.design_rag import DesignRAG
from langgraph.graph import MessagesState
from langchain_core.messages import SystemMessage
# this should be done at app level is used elsewhere
rag = DesignRAG()
async def design_retriever_tool(state: MessagesState, num_examples: int = 1):
"""
Retrieves similar designs based on style requirements
Name: query_similar_designs
"""
result = await rag.query_similar_designs(state["messages"], num_examples)
return SystemMessage(content=result)
|