@pinecone-database/pinecone/dist/integration/assistant-data/context.test.js

This is the official Node.js SDK for [Pinecone](https://www.pinecone.io), written in TypeScript.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
const pinecone_1 = require("../../pinecone");
let pinecone;
let assistant;
let assistantName;
if (!process.env.ASSISTANT_NAME) {
    throw new Error('ASSISTANT_NAME environment variable is not set');
}
else {
    assistantName = process.env.ASSISTANT_NAME;
}
beforeAll(async () => {
    pinecone = new pinecone_1.Pinecone();
    assistant = pinecone.Assistant(assistantName);
});
describe('Context happy path', () => {
    test('Get context with valid query', async () => {
        const response = await assistant.context({
            query: 'What is in the uploaded file?',
        });
        expect(response).toBeDefined();
        expect(response.snippets).toBeDefined();
        expect(Array.isArray(response.snippets)).toBe(true);
    });
    test('Get context with query and filter', async () => {
        const response = await assistant.context({
            query: 'What is in the uploaded file?',
            filter: { key: 'valueOne' },
        });
        expect(response).toBeDefined();
        expect(response.snippets).toBeDefined();
        expect(Array.isArray(response.snippets)).toBe(true);
    });
});
describe('Context error paths', () => {
    test('Context with empty query', async () => {
        await expect(assistant.context({ query: '' })).rejects.toThrow('You must pass an object with required properties (`query`, or `messages`) to retrieve context snippets.');
    });
    test('Context with nonexistent assistant', async () => {
        await expect(pinecone.Assistant('nonexistent').context({
            query: 'What is in the file?',
        })).rejects.toThrow(/404/);
    });
});
//# sourceMappingURL=context.test.js.map