Update main_process/moe_router.py
Browse files
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 |
-
|
| 1493 |
|
| 1494 |
-
|
|
|
|
|
|
|
| 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
|