weaviate-agents/dist/query/agent.test.js

JS/TS client for Weaviate Agents

var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
import { QueryAgent } from "./agent.js";
it("runs the query agent", () => __awaiter(void 0, void 0, void 0, function* () {
    const mockClient = {
        getConnectionDetails: jest.fn().mockResolvedValue({
            host: "test-cluster",
            bearerToken: "test-token",
            headers: { "X-Provider": "test-key" },
        }),
    };
    global.fetch = jest.fn(() => Promise.resolve({
        ok: true,
        json: () => Promise.resolve({
            original_query: "Test query",
            collection_names: ["test-collection"],
            searches: [
                [
                    {
                        collection: "test-collection",
                        queries: ["Test search"],
                        filters: [],
                        filter_operators: "AND",
                    },
                ],
            ],
            aggregations: [],
            usage: {
                requests: 1,
                request_tokens: 128,
                response_tokens: 256,
                total_tokens: 384,
            },
            total_time: 10,
            is_partial_answer: false,
            missing_information: [],
            final_answer: "Test answer",
            sources: [
                {
                    object_id: "123",
                    collection: "test-collection",
                },
            ],
        }),
    }));
    const agent = new QueryAgent(mockClient, {
        systemPrompt: "test system prompt",
    });
    const response = yield agent.run("What is the capital of France?", {
        collections: ["test-collection"],
    });
    expect(response).toEqual({
        outputType: "finalState",
        originalQuery: "Test query",
        collectionNames: ["test-collection"],
        searches: [
            [
                {
                    collection: "test-collection",
                    queries: ["Test search"],
                    filters: [],
                    filterOperators: "AND",
                },
            ],
        ],
        aggregations: [],
        usage: {
            requests: 1,
            requestTokens: 128,
            responseTokens: 256,
            totalTokens: 384,
            details: undefined,
        },
        totalTime: 10,
        isPartialAnswer: false,
        missingInformation: [],
        finalAnswer: "Test answer",
        sources: [
            {
                objectId: "123",
                collection: "test-collection",
            },
        ],
        display: expect.any(Function),
    });
}));