Skip to content

GitLab

Commit d361404a authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

refac

parent 465c3a99
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
backend/apps/webui/routers/memories.py
View file @ d361404a
...@@ -101,6 +101,7 @@ async def update_memory_by_id( ...@@ -101,6 +101,7 @@ async def update_memory_by_id(
class QueryMemoryForm(BaseModel): class QueryMemoryForm(BaseModel):
content: str content: str
k: Optional[int] = 1
@router.post("/query") @router.post("/query")
...@@ -112,7 +113,7 @@ async def query_memory( ...@@ -112,7 +113,7 @@ async def query_memory(
results = collection.query( results = collection.query(
query_embeddings=[query_embedding], query_embeddings=[query_embedding],
n_results=1, # how many results to return n_results=form_data.k, # how many results to return
) )
return results return results
......
backend/data/config.json
View file @ d361404a
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
}, },
{ {
"title": ["Show me a code snippet", "of a website's sticky header"], "title": ["Show me a code snippet", "of a website's sticky header"],
],
"content": "Show me a code snippet of a website's sticky header in CSS and JavaScript." "content": "Show me a code snippet of a website's sticky header in CSS and JavaScript."
}, },
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment