Technologic101 commited on
Commit
7ec8434
·
1 Parent(s): 1c12a94

task: init RAG once

Browse files
Files changed (1) hide show
  1. src/tools/design_retriever.py +3 -2
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