Skip to content

Commit 8faa7a4

Browse files
authored
(EAI-1190) fix eval experiments (#845)
* makeGenerateResponse args made optional * fix all eval experiments * using prebuild script to check for errors in complete package
1 parent 256b227 commit 8faa7a4

File tree

8 files changed

+9
-7
lines changed

8 files changed

+9
-7
lines changed

packages/chatbot-server-mongodb-public/package.json

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,8 @@
1414
".": "./build/lib.js"
1515
},
1616
"scripts": {
17+
"build:check": "tsc --noEmit --project tsconfig.json",
18+
"prebuild": "npm run build:check",
1719
"build": "rm -rf ./build/ && tsc -b tsconfig.build.json",
1820
"dev": "nodemon src/index.ts",
1921
"format": "prettier ./src --write",

packages/chatbot-server-mongodb-public/src/eval/bin/generateEvalCasesYamlFromCSV.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,7 @@ async function main({
124124
);
125125
const urlsNotIngested = await pageStore.getMissingPagesByUrl({
126126
expectedUrls,
127-
urlTransformer: normalizeUrl,
127+
urlTransformer: (url) => normalizeUrl({ url }),
128128
});
129129
if (urlsNotIngested.length > 0) {
130130
console.warn(

packages/chatbot-server-mongodb-public/src/eval/experiments/allScorersTest.eval.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ import fs from "fs";
1111
import path from "path";
1212
import { makeConversationEval } from "../ConversationEval";
1313
import { makeGenerateResponse } from "../../config";
14-
import { addMessageToConversationStream } from "../../processors/generateResponseWithSearchTool";
1514

1615
async function conversationEval() {
1716
// Get all the conversation eval cases from YAML

packages/chatbot-server-mongodb-public/src/eval/experiments/architectureCenter.eval.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ import fs from "fs";
1111
import path from "path";
1212
import { makeConversationEval } from "../ConversationEval";
1313
import { makeGenerateResponse } from "../../config";
14-
import { addMessageToConversationStream } from "../../processors/generateResponseWithSearchTool";
1514

1615
async function conversationEval() {
1716
// Get ONLY architecture center conversations

packages/chatbot-server-mongodb-public/src/eval/experiments/customSystemPrompt.eval.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@ import {
99
} from "../evalHelpers";
1010
import { makeConversationEval } from "../ConversationEval";
1111
import { closeDbConnections, makeGenerateResponse } from "../../config";
12-
import { responsesApiStream } from "../../processors/generateResponseWithSearchTool";
1312

1413
const conversationEvalCases: ConversationEvalCase[] = [
1514
// Test 1: Basic custom system prompt override

packages/chatbot-server-mongodb-public/src/eval/experiments/dotcomQuestionsTest.eval.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ import fs from "fs";
1111
import path from "path";
1212
import { makeConversationEval } from "../ConversationEval";
1313
import { makeGenerateResponse } from "../../config";
14-
import { addMessageToConversationStream } from "../../processors/generateResponseWithSearchTool";
1514

1615
async function conversationEval() {
1716
// Get dotcom question set eval cases from YAML

packages/chatbot-server-mongodb-public/src/tools/search.eval.ts

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@ import "dotenv/config";
22
import {
33
Eval,
44
EvalCase,
5+
EvalParameters,
56
EvalScorer,
67
EvalTask,
78
} from "mongodb-rag-core/braintrust";
@@ -62,8 +63,10 @@ const { k } = retrievalConfig.findNearestNeighborsOptions;
6263
const retrieveRelevantContentEvalTask: EvalTask<
6364
RetrievalEvalCaseInput,
6465
RetrievalTaskOutput,
65-
RetrievalEvalCaseExpected
66-
> = async function (data) {
66+
RetrievalEvalCaseExpected,
67+
void,
68+
EvalParameters
69+
> = async function (data: RetrievalEvalCaseInput) {
6770
// TODO: (EAI-991) implement retrieval task for evaluation
6871
const extractedMetadata: MongoDbSearchToolArgs = {
6972
productName: null,

packages/mongodb-rag-core/src/index.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,3 +24,4 @@ export * from "./VectorStore";
2424
export * from "./arrayFilters";
2525
export * from "./assertEnvVars";
2626
export * from "./getEnv";
27+
export * from "./mongoDbMetadata";

0 commit comments

Comments
 (0)