Spaces:
Runtime error
Runtime error
Commit
·
7ec8434
1
Parent(s):
1c12a94
task: init RAG once
Browse files
src/tools/design_retriever.py
CHANGED
|
@@ -2,15 +2,16 @@ from nodes.design_rag import DesignRAG
|
|
| 2 |
from langgraph.graph import MessagesState
|
| 3 |
from langchain_core.messages import SystemMessage
|
| 4 |
|
|
|
|
|
|
|
|
|
|
| 5 |
async def design_retriever_tool(state: MessagesState, num_examples: int = 1):
|
| 6 |
"""
|
| 7 |
Retrieves similar designs based on style requirements
|
| 8 |
Name: query_similar_designs
|
| 9 |
"""
|
| 10 |
-
rag = DesignRAG() # Create instance
|
| 11 |
|
| 12 |
result = await rag.query_similar_designs(state["messages"], num_examples)
|
| 13 |
-
print("Here's the result: ", result)
|
| 14 |
|
| 15 |
return SystemMessage(content=result)
|
| 16 |
|
|
|
|
| 2 |
from langgraph.graph import MessagesState
|
| 3 |
from langchain_core.messages import SystemMessage
|
| 4 |
|
| 5 |
+
# this should be done at app level is used elsewhere
|
| 6 |
+
rag = DesignRAG()
|
| 7 |
+
|
| 8 |
async def design_retriever_tool(state: MessagesState, num_examples: int = 1):
|
| 9 |
"""
|
| 10 |
Retrieves similar designs based on style requirements
|
| 11 |
Name: query_similar_designs
|
| 12 |
"""
|
|
|
|
| 13 |
|
| 14 |
result = await rag.query_similar_designs(state["messages"], num_examples)
|
|
|
|
| 15 |
|
| 16 |
return SystemMessage(content=result)
|
| 17 |
|