File tree 1 file changed +4
-4
lines changed
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -15,25 +15,25 @@ const hfDocs: BackendTool = {
15
15
"A search query which will be used to fetch the most relevant docs snippets regarding the user's query" ,
16
16
} ,
17
17
} ,
18
- async * call ( { prompt } , { messages, ip, username } ) {
18
+ async * call ( { query } , { messages, ip, username } ) {
19
19
const ipToken = await getIpToken ( ip , username ) ;
20
20
21
21
const userMessages = messages . filter ( ( { from } ) => from === "user" ) ;
22
22
const previousUserMessages = userMessages . slice ( 0 , - 1 ) ;
23
23
24
- const query =
24
+ const queryWithPreviousMsgs =
25
25
( previousUserMessages . length
26
26
? `Previous questions: \n${ previousUserMessages
27
27
. map ( ( { content } ) => `- ${ content } ` )
28
28
. join ( "\n" ) } `
29
29
: "" ) +
30
30
"\n\nCurrent Question: " +
31
- String ( prompt ) ;
31
+ String ( query ) ;
32
32
33
33
const outputs = await callSpace < string [ ] , string [ ] > (
34
34
"huggingchat/hf-docs" ,
35
35
"/predict" ,
36
- [ query , "RAG-friendly" ] ,
36
+ [ queryWithPreviousMsgs , "RAG-friendly" ] ,
37
37
ipToken
38
38
) ;
39
39
You can’t perform that action at this time.
0 commit comments