fix: Bedrock response sender

This commit is contained in:
Danny Avila 2024-09-02 21:40:38 -04:00
parent d66a35887d
commit edcc66685a
No known key found for this signature in database
GPG key ID: 2DD9CC89B9B50364
6 changed files with 31 additions and 12 deletions

View file

@ -1,5 +1,9 @@
const { createContentAggregator } = require('@librechat/agents');
const { EModelEndpoint, providerEndpointMap } = require('librechat-data-provider');
const {
EModelEndpoint,
providerEndpointMap,
getResponseSender,
} = require('librechat-data-provider');
const { getDefaultHandlers } = require('~/server/controllers/agents/callbacks');
// const { loadAgentTools } = require('~/server/services/ToolService');
const getOptions = require('~/server/services/Endpoints/bedrock/options');
@ -40,9 +44,15 @@ const initializeClient = async ({ req, res, endpointOption }) => {
agent.max_context_tokens ??
getModelMaxTokens(modelOptions.model, providerEndpointMap[agent.provider]);
const sender = getResponseSender({
...endpointOption,
model: endpointOption.model_parameters.model,
});
const client = new AgentClient({
req,
agent,
sender,
// tools,
// toolMap,
modelOptions,