Andreas/LibreChat
1
0
Fork 0
mirror of https://github.com/danny-avila/LibreChat.git synced 2025-09-22 06:00:56 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

ci: add test similar to example from the langchain PR

Browse source
This commit is contained in:
Dustin Healy 2025-08-19 19:11:37 -07:00
parent a2fed40fbb
commit cf0f9e8f30
1 changed files with 58 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

59
packages/api/src/format/addBedrockCacheControl.spec.ts
View file

@ -2,7 +2,7 @@ import { addBedrockCacheControl } from './addBedrockCacheControl';
import { ContentTypes, Agents } from 'librechat-data-provider'; import { ContentTypes, Agents } from 'librechat-data-provider';
type TestMsg = { type TestMsg = {
role?: 'user' | 'assistant'; role?: 'user' | 'assistant' | 'system';
content?: string | Agents.MessageContentComplex[]; content?: string | Agents.MessageContentComplex[];
}; };
@ -105,4 +105,61 @@ describe('addBedrockCacheControl (Bedrock cache checkpoints)', () => {
expect(first[0]).toEqual({ type: ContentTypes.TEXT, text: 'Will be modified' }); expect(first[0]).toEqual({ type: ContentTypes.TEXT, text: 'Will be modified' });
expect(first[1]).toEqual({ cachePoint: { type: 'default' } }); expect(first[1]).toEqual({ cachePoint: { type: 'default' } });
}); });
it('works with the example from the langchain pr', () => {
const messages: TestMsg[] = [
{
role: 'system',
content: [{ type: ContentTypes.TEXT, text: "You're an advanced AI assistant." }],
},
{
role: 'user',
content: [{ type: ContentTypes.TEXT, text: 'What is the capital of France?' }],
},
];
const result = addBedrockCacheControl(messages);
let system = result[0].content as Agents.MessageContentComplex[];
let user = result[1].content as Agents.MessageContentComplex[];
expect(system[0]).toEqual({
type: ContentTypes.TEXT,
text: "You're an advanced AI assistant.",
});
expect(system[1]).toEqual({ cachePoint: { type: 'default' } });
expect(user[0]).toEqual({
type: ContentTypes.TEXT,
text: 'What is the capital of France?',
});
expect(user[1]).toEqual({ cachePoint: { type: 'default' } });
result.push({
role: 'assistant',
content: [{ type: ContentTypes.TEXT, text: 'Sure! The capital of France is Paris.' }],
});
const result2 = addBedrockCacheControl(result);
system = result2[0].content as Agents.MessageContentComplex[];
user = result2[1].content as Agents.MessageContentComplex[];
const assistant = result2[2].content as Agents.MessageContentComplex[];
expect(system[0]).toEqual({
type: ContentTypes.TEXT,
text: "You're an advanced AI assistant.",
});
expect(system[1]).toEqual({ cachePoint: { type: 'default' } });
expect(user[0]).toEqual({
type: ContentTypes.TEXT,
text: 'What is the capital of France?',
});
expect(user[1]).toEqual({ cachePoint: { type: 'default' } });
expect(assistant[0]).toEqual({
type: ContentTypes.TEXT,
text: 'Sure! The capital of France is Paris.',
});
expect(assistant[1]).toEqual({ cachePoint: { type: 'default' } });
});
}); });