new(gpt.gblib): Wikipedia feature.
This commit is contained in:
parent
14b172187e
commit
ba0adc57af
1 changed files with 10 additions and 1 deletions
|
@ -50,6 +50,7 @@ import { Serialized } from "@langchain/core/load/serializable";
|
||||||
import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
|
import { BaseCallbackHandler } from "@langchain/core/callbacks/base";
|
||||||
import { pdfToPng, PngPageOutput } from 'pdf-to-png-converter';
|
import { pdfToPng, PngPageOutput } from 'pdf-to-png-converter';
|
||||||
import { DynamicStructuredTool } from "@langchain/core/tools";
|
import { DynamicStructuredTool } from "@langchain/core/tools";
|
||||||
|
import { WikipediaQueryRun } from "@langchain/community/tools/wikipedia_query_run";
|
||||||
import {
|
import {
|
||||||
BaseLLMOutputParser,
|
BaseLLMOutputParser,
|
||||||
OutputParserException,
|
OutputParserException,
|
||||||
|
@ -276,7 +277,9 @@ export class ChatServices {
|
||||||
AIMessagePromptTemplate.fromTemplate(
|
AIMessagePromptTemplate.fromTemplate(
|
||||||
`
|
`
|
||||||
Answer the question without calling any tool, but if there is a need to call:
|
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}
|
${toolsAsText}
|
||||||
|
|
||||||
Do not use any previous tools output in the chat_history.
|
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;
|
return functions;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue