Technologic101 commited on
Commit
8c57a85
·
1 Parent(s): 4698077

task: default to one example

Browse files
Files changed (1) hide show
  1. src/tools/design_retriever.py +1 -1
src/tools/design_retriever.py CHANGED
@@ -2,7 +2,7 @@ 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 = 2):
6
  """
7
  Retrieves similar designs based on style requirements
8
  Name: query_similar_designs
 
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