update import

demo.py

@@ -688,7 +688,7 @@ Keep the description concise and informative. Respond with just the task descrip
         print(f"Warning: Failed to generate task description via API: {str(e)}")
         return "General query processing task requiring analysis and response generation."
 
-def get_routed_llm(query: str, config_path: str = "GraphRouter_eval/configs/config.yaml") -> Tuple[str, str, str]:
+def get_routed_llm(query: str, config_path: str = None) -> Tuple[str, str, str]:
     """
     Use graph router to get the best LLM for a query.
     
@@ -716,8 +716,12 @@ def get_routed_llm(query: str, config_path: str = "GraphRouter_eval/configs/conf
         os.chdir(graph_router_dir)
         
         try:
+            # Use default config path if none provided
+            if config_path is None:
+                config_path = 'configs/config.yaml'
+            
             # Load configuration
-            with open('configs/config.yaml', 'r', encoding='utf-8') as file:
+            with open(config_path, 'r', encoding='utf-8') as file:
                 config = yaml.safe_load(file)
             
             # Load training data