fix: declare endpoint in initialization step

This commit is contained in:
Danny Avila 2024-09-01 09:29:14 -04:00
parent 60ee12d3e8
commit 6d30726113
No known key found for this signature in database
GPG key ID: 2DD9CC89B9B50364
3 changed files with 2 additions and 16 deletions

View file

@ -1,16 +0,0 @@
const { EModelEndpoint } = require('librechat-data-provider');
const AgentClient = require('~/server/controllers/agents/client');
const { logger } = require('~/config');
class BedrockClient extends AgentClient {
constructor(options = {}) {
super(options);
this.options.endpoint = EModelEndpoint.bedrock;
}
setOptions(options) {
logger.info('[api/server/controllers/bedrock/client.js] setOptions', options);
}
}
module.exports = BedrockClient;

View file

@ -108,6 +108,7 @@ const initializeClient = async ({ req, res, endpointOption }) => {
toolMap,
modelOptions,
eventHandlers,
endpoint: EModelEndpoint.agents,
configOptions: options.configOptions,
maxContextTokens:
agent.max_context_tokens ??

View file

@ -47,6 +47,7 @@ const initializeClient = async ({ req, res, endpointOption }) => {
modelOptions,
eventHandlers,
maxContextTokens,
endpoint: EModelEndpoint.bedrock,
configOptions: options.configOptions,
});
return { client };