VeuReu commited on
Commit
d73d0bb
·
verified ·
1 Parent(s): c54fb9f

Update main_process/moe_router.py

Browse files
Files changed (1) hide show
  1. main_process/moe_router.py +6 -3
main_process/moe_router.py CHANGED
@@ -6,6 +6,7 @@ import json
6
  import tempfile
7
  from pathlib import Path
8
  from typing import List, Dict, Counter
 
9
 
10
  # --- Third-Party Libraries ---
11
  import cv2
@@ -1489,12 +1490,14 @@ async def generate_salamadra_result(
1489
 
1490
  messages_registro = [{"role": "system", "content": "Eres un agente que puede ejecutar herramientas Python usando las herramientas disponibles."}]
1491
 
1492
- import yaml
1493
 
1494
- with open("config.yaml", "r") as file:
 
 
1495
  config = yaml.safe_load(file)
1496
 
1497
- stools_enabled = config.get("stools", False)
1498
 
1499
  if stools_enabled:
1500
  count = 1
 
6
  import tempfile
7
  from pathlib import Path
8
  from typing import List, Dict, Counter
9
+ import yaml
10
 
11
  # --- Third-Party Libraries ---
12
  import cv2
 
1490
 
1491
  messages_registro = [{"role": "system", "content": "Eres un agente que puede ejecutar herramientas Python usando las herramientas disponibles."}]
1492
 
1493
+ BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
1494
 
1495
+ config_path = os.path.join(BASE_DIR, "config.yaml")
1496
+
1497
+ with open(config_path, "r") as file:
1498
  config = yaml.safe_load(file)
1499
 
1500
+ stools_enabled = config.get("stools", False)
1501
 
1502
  if stools_enabled:
1503
  count = 1