Spaces:
Runtime error
Runtime error
Commit
·
8c57a85
1
Parent(s):
4698077
task: default to one example
Browse files
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 =
|
| 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
|