Skip to content

Commit dfe3fd3

Browse files
kevin-lanngithub-actions[bot]
authored andcommitted
Auto-formatted the code using Prettier
1 parent 7acb45d commit dfe3fd3

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

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

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ const pinecone = new Pinecone({
4242
});
4343

4444
const index: Index<RecordMetadata> = pinecone.Index(
45-
process.env.PINECONE_INDEX_NAME!
45+
process.env.PINECONE_INDEX_NAME!,
4646
);
4747

4848
console.log("Connected to OpenAI API");
@@ -99,7 +99,7 @@ function analyzeQuery(query: string): {
9999
"prerequisites",
100100
"corequisites",
101101
"departments",
102-
"programs"
102+
"programs",
103103
);
104104
}
105105

@@ -117,7 +117,7 @@ async function searchSelectedNamespaces(
117117
query: string,
118118
k: number,
119119
namespaces: string[],
120-
filters?: Object
120+
filters?: Object,
121121
): Promise<Document[]> {
122122
let allResults: Document[] = [];
123123

@@ -139,7 +139,7 @@ async function searchSelectedNamespaces(
139139
const results = await namespaceStore.similaritySearch(
140140
query,
141141
Math.max(k, namespaceToMinResults.get(namespace)),
142-
namespace === "courses_v3" ? filters : undefined
142+
namespace === "courses_v3" ? filters : undefined,
143143
);
144144
console.log(`Found ${results.length} results in namespace: ${namespace}`);
145145
allResults = [...allResults, ...results];
@@ -160,7 +160,7 @@ async function searchSelectedNamespaces(
160160
// Reformulate user query to make more concise query to database, taking into consideration context
161161
async function reformulateQuery(
162162
latestQuery: string,
163-
conversationHistory: any[]
163+
conversationHistory: any[],
164164
): Promise<string> {
165165
try {
166166
const openai2 = new OpenAI({
@@ -258,7 +258,7 @@ function includeFilters(query: string) {
258258
if (keywords.some((keyword) => lowerQuery.includes(keyword))) {
259259
relaventBreadthRequirements.push(convertBreadthRequirement(namespace));
260260
}
261-
}
261+
},
262262
);
263263

264264
Object.entries(YEAR_LEVEL_KEYWORDS).forEach(([namespace, keywords]) => {
@@ -415,7 +415,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
415415
// Use GPT-4o to reformulate the query based on conversation history
416416
const reformulatedQuery = await reformulateQuery(
417417
latestMessage,
418-
conversationHistory.slice(-CHATBOT_MEMORY_THRESHOLD) // last K messages
418+
conversationHistory.slice(-CHATBOT_MEMORY_THRESHOLD), // last K messages
419419
);
420420
console.log(">>>> Original query:", latestMessage);
421421
console.log(">>>> Reformulated query:", reformulatedQuery);
@@ -429,8 +429,8 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
429429
if (requiresSearch) {
430430
console.log(
431431
`Query requires knowledge retrieval, searching namespaces: ${relevantNamespaces.join(
432-
", "
433-
)}`
432+
", ",
433+
)}`,
434434
);
435435

436436
const filters = includeFilters(reformulatedQuery);
@@ -441,7 +441,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
441441
reformulatedQuery,
442442
3,
443443
relevantNamespaces,
444-
Object.keys(filters).length === 0 ? undefined : filters
444+
Object.keys(filters).length === 0 ? undefined : filters,
445445
);
446446
// console.log("Search Results: ", searchResults);
447447

@@ -451,7 +451,7 @@ export const chat = asyncHandler(async (req: Request, res: Response) => {
451451
}
452452
} else {
453453
console.log(
454-
"Query does not require knowledge retrieval, skipping search"
454+
"Query does not require knowledge retrieval, skipping search",
455455
);
456456
}
457457

@@ -516,15 +516,15 @@ export const testSimilaritySearch = asyncHandler(
516516
if (requiresSearch) {
517517
console.log(
518518
`Query requires knowledge retrieval, searching namespaces: ${relevantNamespaces.join(
519-
", "
520-
)}`
519+
", ",
520+
)}`,
521521
);
522522

523523
// Search only the relevant namespaces
524524
const searchResults = await searchSelectedNamespaces(
525525
message,
526526
3,
527-
relevantNamespaces
527+
relevantNamespaces,
528528
);
529529
console.log("Search Results: ", searchResults);
530530

@@ -534,11 +534,11 @@ export const testSimilaritySearch = asyncHandler(
534534
}
535535
} else {
536536
console.log(
537-
"Query does not require knowledge retrieval, skipping search"
537+
"Query does not require knowledge retrieval, skipping search",
538538
);
539539
}
540540

541541
console.log("CONTEXT: ", context);
542542
res.status(200).send(context);
543-
}
543+
},
544544
);

0 commit comments

Comments
 (0)