Skip to content

Commit 07221e5

Browse files
committed
Merge branch 'mt/scrum-138-child-139-backendTesting-timetablesController' of https://github.com/UTSC-CSCC01-Software-Engineering-I/term-group-project-c01w25-project-course-matrix into mt/scrum-138-child-139-backendTesting-timetablesController
2 parents 329b653 + dd15efc commit 07221e5

File tree

3 files changed

+21
-21
lines changed

3 files changed

+21
-21
lines changed

course-matrix/backend/__tests__/timetablesController.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ jest.mock("@ai-sdk/openai", () => ({
1515

1616
jest.mock("ai", () => ({
1717
streamText: jest.fn(() =>
18-
Promise.resolve({ pipeDataStreamToResponse: jest.fn() })
18+
Promise.resolve({ pipeDataStreamToResponse: jest.fn() }),
1919
),
2020
}));
2121

@@ -179,7 +179,7 @@ describe("GET /api/timetables", () => {
179179
(
180180
authHandler as jest.MockedFunction<typeof authHandler>
181181
).mockImplementationOnce(
182-
mockAuthHandler("testuser01-ab9e6877-f603-4c6a-9832-864e520e4d01")
182+
mockAuthHandler("testuser01-ab9e6877-f603-4c6a-9832-864e520e4d01"),
183183
);
184184

185185
const response = await request(app).get("/api/timetables");
@@ -194,7 +194,7 @@ describe("GET /api/timetables", () => {
194194
(
195195
authHandler as jest.MockedFunction<typeof authHandler>
196196
).mockImplementationOnce(
197-
mockAuthHandler("testuser02-1d3f02df-f926-4c1f-9f41-58ca50816a33")
197+
mockAuthHandler("testuser02-1d3f02df-f926-4c1f-9f41-58ca50816a33"),
198198
);
199199

200200
const response = await request(app).get("/api/timetables");

course-matrix/backend/src/controllers/aiController.ts

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ const pinecone = new Pinecone({
3636
});
3737

3838
const index: Index<RecordMetadata> = pinecone.Index(
39-
process.env.PINECONE_INDEX_NAME!
39+
process.env.PINECONE_INDEX_NAME!,
4040
);
4141

4242
console.log("Connected to OpenAI API");
@@ -93,7 +93,7 @@ function analyzeQuery(query: string): {
9393
"prerequisites",
9494
"corequisites",
9595
"departments",
96-
"programs"
96+
"programs",
9797
);
9898
}
9999

@@ -111,7 +111,7 @@ async function searchSelectedNamespaces(
111111
query: string,
112112
k: number,
113113
namespaces: string[],
114-
filters?: Object
114+
filters?: Object,
115115
): Promise<Document[]> {
116116
let allResults: Document[] = [];
117117

@@ -133,7 +133,7 @@ async function searchSelectedNamespaces(
133133
const results = await namespaceStore.similaritySearch(
134134
query,
135135
Math.max(k, namespaceToMinResults.get(namespace)),
136-
namespace === "courses_v3" ? filters : undefined
136+
namespace === "courses_v3" ? filters : undefined,
137137
);
138138
console.log(`Found ${results.length} results in namespace: ${namespace}`);
139139
allResults = [...allResults, ...results];
@@ -154,7 +154,7 @@ async function searchSelectedNamespaces(
154154
// Reformulate user query to make more concise query to database, taking into consideration context
155155
async function reformulateQuery(
156156
latestQuery: string,
157-
conversationHistory: any[]
157+
conversationHistory: any[],
158158
): Promise<string> {
159159
try {
160160
const openai = new OpenAI({
@@ -252,7 +252,7 @@ function includeFilters(query: string) {
252252
if (keywords.some((keyword) => lowerQuery.includes(keyword))) {
253253
relaventBreadthRequirements.push(convertBreadthRequirement(namespace));
254254
}
255-
}
255+
},
256256
);
257257

258258
Object.entries(YEAR_LEVEL_KEYWORDS).forEach(([namespace, keywords]) => {
@@ -317,7 +317,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
317317
// Use GPT-4o to reformulate the query based on conversation history
318318
const reformulatedQuery = await reformulateQuery(
319319
latestMessage,
320-
conversationHistory.slice(-CHATBOT_MEMORY_THRESHOLD) // last K messages
320+
conversationHistory.slice(-CHATBOT_MEMORY_THRESHOLD), // last K messages
321321
);
322322
console.log(">>>> Original query:", latestMessage);
323323
console.log(">>>> Reformulated query:", reformulatedQuery);
@@ -331,8 +331,8 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
331331
if (requiresSearch) {
332332
console.log(
333333
`Query requires knowledge retrieval, searching namespaces: ${relevantNamespaces.join(
334-
", "
335-
)}`
334+
", ",
335+
)}`,
336336
);
337337

338338
const filters = includeFilters(reformulatedQuery);
@@ -343,7 +343,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
343343
reformulatedQuery,
344344
3,
345345
relevantNamespaces,
346-
Object.keys(filters).length === 0 ? undefined : filters
346+
Object.keys(filters).length === 0 ? undefined : filters,
347347
);
348348
// console.log("Search Results: ", searchResults);
349349

@@ -411,15 +411,15 @@ export const testSimilaritySearch = asyncHandler(
411411
if (requiresSearch) {
412412
console.log(
413413
`Query requires knowledge retrieval, searching namespaces: ${relevantNamespaces.join(
414-
", "
415-
)}`
414+
", ",
415+
)}`,
416416
);
417417

418418
// Search only the relevant namespaces
419419
const searchResults = await searchSelectedNamespaces(
420420
message,
421421
3,
422-
relevantNamespaces
422+
relevantNamespaces,
423423
);
424424
console.log("Search Results: ", searchResults);
425425

@@ -429,11 +429,11 @@ export const testSimilaritySearch = asyncHandler(
429429
}
430430
} else {
431431
console.log(
432-
"Query does not require knowledge retrieval, skipping search"
432+
"Query does not require knowledge retrieval, skipping search",
433433
);
434434
}
435435

436436
console.log("CONTEXT: ", context);
437437
res.status(200).send(context);
438-
}
438+
},
439439
);

course-matrix/backend/src/index.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,8 @@ app.get(
4949
asyncHandler(async (_, response) =>
5050
response.json({
5151
info: "Testing course matrix backend server",
52-
})
53-
)
52+
}),
53+
),
5454
);
5555

5656
/**
@@ -67,7 +67,7 @@ app.get(
6767
} catch (err) {
6868
return res.status(500).send({ err });
6969
}
70-
})
70+
}),
7171
);
7272

7373
server = app.listen(config.PORT, () => {

0 commit comments

Comments
 (0)