GeneralBots/botserver
2
1
Fork 0
Code Issues Pull requests 1 Projects Releases 323 Packages Wiki Activity Actions

new(gpt.gblib): Wikipedia feature.

Browse source
This commit is contained in:
Rodrigo Rodriguez 2024-03-21 17:35:09 -03:00
parent 14b172187e
commit ba0adc57af
1 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
packages/gpt.gblib/services/ChatServices.ts
View file

@ -50,6 +50,7 @@ import { Serialized } from "@langchain/core/load/serializable";
import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
import { pdfToPng, PngPageOutput } from 'pdf-to-png-converter';
import { DynamicStructuredTool } from "@langchain/core/tools";
import { WikipediaQueryRun } from "@langchain/community/tools/wikipedia_query_run";
import {
BaseLLMOutputParser,
OutputParserException,
@ -276,7 +277,9 @@ export class ChatServices {
AIMessagePromptTemplate.fromTemplate(
`
Answer the question without calling any tool, but if there is a need to call:
You have access to the following set of tools. Here are the names and descriptions for each tool:
You have access to the following set of tools.
Here are the names and descriptions for each tool:
${toolsAsText}
Do not use any previous tools output in the chat_history.
@ -457,6 +460,12 @@ export class ChatServices {
});
const tool = new WikipediaQueryRun({
topKResults: 3,
maxDocContentLength: 4000,
});
functions.push(tool);
return functions;
}
}