style: format imports and add language_code to request

src/agent.ts

@@ -393,7 +393,8 @@ export async function getResponse(question?: string,
   badHostnames: string[] = [],
   onlyHostnames: string[] = [],
   maxRef: number = 10,
-                                  minRelScore: number = 0.75
+  minRelScore: number = 0.75,
+  languageCode: string | undefined = 'zh-CN'
 ): Promise<{ result: StepAction; context: TrackerContext; visitedURLs: string[], readURLs: string[], allURLs: string[] }> {
 
   let step = 0;
@@ -417,7 +418,7 @@ export async function getResponse(question?: string,
   }
 
   const SchemaGen = new Schemas();
-  await SchemaGen.setLanguage(question)
+  await SchemaGen.setLanguage(languageCode || question)
   const context: TrackerContext = {
     tokenTracker: existingContext?.tokenTracker || new TokenTracker(tokenBudget),
     actionTracker: existingContext?.actionTracker || new ActionTracker()

src/app.ts

@@ -566,7 +566,8 @@ app.post('/v1/chat/completions', (async (req: Request, res: Response) => {
       body.bad_hostnames?.map(i => normalizeHostName(i)),
       body.only_hostnames?.map(i => normalizeHostName(i)),
       body.max_annotations,
-      body.min_annotation_relevance
+      body.min_annotation_relevance,
+      body.language_code
       )
     let finalAnswer = (finalStep as AnswerAction).mdAnswer;
 

src/cli.ts

@@ -26,7 +26,7 @@ program
       const { result } = await getResponse(
         query,
         parseInt(options.tokenBudget),
-        parseInt(options.maxAttempts)
+        parseInt(options.maxAttempts),
       );
 
       if (result.action === 'answer') {

src/types.ts

@@ -245,6 +245,7 @@ export interface ChatCompletionRequest {
 
   max_annotations?: number;
   min_annotation_relevance?: number;
+  language_code?: string;
 }
 
 export interface URLAnnotation {

src/utils/schemas.ts

@@ -59,12 +59,51 @@ Evaluation: {
   };
 }
 
+const languageISO6391Map: Record<string, string> = {
+  'en': 'English',
+  'zh': 'Chinese',
+  'zh-CN': 'Simplified Chinese',
+  'zh-TW': 'Traditional Chinese',
+  'de': 'German',
+  'fr': 'French',
+  'es': 'Spanish',
+  'it': 'Italian',
+  'ja': 'Japanese',
+  'ko': 'Korean',
+  'pt': 'Portuguese',
+  'ru': 'Russian',
+  'ar': 'Arabic',
+  'hi': 'Hindi',
+  'bn': 'Bengali',
+  'tr': 'Turkish',
+  'nl': 'Dutch',
+  'pl': 'Polish',
+  'sv': 'Swedish',
+  'no': 'Norwegian',
+  'da': 'Danish',
+  'fi': 'Finnish',
+  'el': 'Greek',
+  'he': 'Hebrew',
+  'hu': 'Hungarian',
+  'id': 'Indonesian',
+  'ms': 'Malay',
+  'th': 'Thai',
+  'vi': 'Vietnamese',
+  'ro': 'Romanian',
+  'bg': 'Bulgarian',
+}
+
 export class Schemas {
   public languageStyle: string = 'formal English';
   public languageCode: string = 'en';
 
 
   async setLanguage(query: string) {
+    if (languageISO6391Map[query]) {
+      this.languageCode = query;
+      this.languageStyle = `formal ${languageISO6391Map[query]}`;
+      return;
+    }
     const generator = new ObjectGeneratorSafe();
     const prompt = getLanguagePrompt(query.slice(0, 100))