BinKhoaLe1812 commited on
Commit
ba92913
·
verified ·
1 Parent(s): e96295c

upd langmap

Browse files
Files changed (1) hide show
  1. search/coordinator.py +5 -2
search/coordinator.py CHANGED
@@ -216,7 +216,10 @@ class SearchCoordinator:
216
  lang_mapping = {
217
  'EN': 'en',
218
  'VI': 'vi',
219
- 'ZH': 'zh'
 
 
 
220
  }
221
  search_language = lang_mapping.get(target_language, 'en')
222
 
@@ -224,4 +227,4 @@ class SearchCoordinator:
224
  video_results = self.video_engine.search(query, num_results, search_language)
225
 
226
  logger.info(f"Video search completed: {len(video_results)} videos found")
227
- return video_results
 
216
  lang_mapping = {
217
  'EN': 'en',
218
  'VI': 'vi',
219
+ 'ZH': 'zh',
220
+ 'en': 'en',
221
+ 'vi': 'vi',
222
+ 'zh': 'zh'
223
  }
224
  search_language = lang_mapping.get(target_language, 'en')
225
 
 
227
  video_results = self.video_engine.search(query, num_results, search_language)
228
 
229
  logger.info(f"Video search completed: {len(video_results)} videos found")
230
+ return video_results