Skip to content

Commit ca2d9cb

Browse files
committed
fix
1 parent 892baa9 commit ca2d9cb

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/lib/server/tools/hfDocs.ts

+4-4
Original file line numberDiff line numberDiff line change
@@ -15,25 +15,25 @@ const hfDocs: BackendTool = {
1515
"A search query which will be used to fetch the most relevant docs snippets regarding the user's query",
1616
},
1717
},
18-
async *call({ prompt }, { messages, ip, username }) {
18+
async *call({ query }, { messages, ip, username }) {
1919
const ipToken = await getIpToken(ip, username);
2020

2121
const userMessages = messages.filter(({ from }) => from === "user");
2222
const previousUserMessages = userMessages.slice(0, -1);
2323

24-
const query =
24+
const queryWithPreviousMsgs =
2525
(previousUserMessages.length
2626
? `Previous questions: \n${previousUserMessages
2727
.map(({ content }) => `- ${content}`)
2828
.join("\n")}`
2929
: "") +
3030
"\n\nCurrent Question: " +
31-
String(prompt);
31+
String(query);
3232

3333
const outputs = await callSpace<string[], string[]>(
3434
"huggingchat/hf-docs",
3535
"/predict",
36-
[query, "RAG-friendly"],
36+
[queryWithPreviousMsgs, "RAG-friendly"],
3737
ipToken
3838
);
3939

0 commit comments

Comments
 (0)