@@ -58,8 +58,8 @@ function analyzeQuery(query: string): {
5858
5959 // If a course code is detected, add tehse namespaces
6060 if ( containsCourseCode ) {
61- if ( ! relevantNamespaces . includes ( "courses " ) )
62- relevantNamespaces . push ( "courses " ) ;
61+ if ( ! relevantNamespaces . includes ( "courses_v2 " ) )
62+ relevantNamespaces . push ( "courses_v2 " ) ;
6363 if ( ! relevantNamespaces . includes ( "offerings" ) )
6464 relevantNamespaces . push ( "offerings" ) ;
6565 if ( ! relevantNamespaces . includes ( "prerequisites" ) )
@@ -70,8 +70,8 @@ function analyzeQuery(query: string): {
7070 if ( DEPARTMENT_CODES . some ( ( code ) => lowerQuery . includes ( code ) ) ) {
7171 if ( ! relevantNamespaces . includes ( "departments" ) )
7272 relevantNamespaces . push ( "departments" ) ;
73- if ( ! relevantNamespaces . includes ( "courses " ) )
74- relevantNamespaces . push ( "courses " ) ;
73+ if ( ! relevantNamespaces . includes ( "courses_v2 " ) )
74+ relevantNamespaces . push ( "courses_v2 " ) ;
7575 }
7676
7777 // If search is required at all
@@ -83,11 +83,12 @@ function analyzeQuery(query: string): {
8383 // If no specific namespaces identified & search required, then search all
8484 if ( requiresSearch && relevantNamespaces . length === 0 ) {
8585 relevantNamespaces . push (
86- "courses " ,
86+ "courses_v2 " ,
8787 "offerings" ,
8888 "prerequisites" ,
8989 "corequisites" ,
9090 "departments" ,
91+ "programs" ,
9192 ) ;
9293 }
9394
@@ -153,6 +154,8 @@ async function reformulateQuery(
153154 apiKey : process . env . OPENAI_API_KEY ,
154155 } ) ;
155156
157+ // console.log("History: ", conversationHistory);
158+
156159 // Create messages array with the correct type structure
157160 const messages : OpenAI . Chat . ChatCompletionMessageParam [ ] = [
158161 {
@@ -184,7 +187,13 @@ async function reformulateQuery(
184187 Output: "What are the course names of course codes: MGTA01, CSCA08, MATA31, MATA35?"
185188
186189 User: "How are you doing today?"
187- Output: "How are you doing today?"` ,
190+ Output: "How are you doing today?"
191+
192+ User: "Give 2nd year math courses."
193+ Output: "What are some 2nd year math courses?"
194+
195+ User: "Give first year math courses."
196+ Output: "What are some 1st year math courses?"` ,
188197 } ,
189198 ] ;
190199
@@ -206,7 +215,7 @@ async function reformulateQuery(
206215 model : "gpt-4o-mini" ,
207216 messages : messages ,
208217 temperature : 0.1 , // Lower temperature for more consistent, focused queries
209- max_tokens : 150 , // Limit response length
218+ max_tokens : latestQuery . length * 3 , // Limit response length. Proportional to user input.
210219 top_p : 0.5 , // Reduced top_p for more focused outputs
211220 } ) ;
212221
0 commit comments