Skip to content

Commit

Permalink
Add support for importing model configurations from YAML files
Browse files Browse the repository at this point in the history
  • Loading branch information
Cufeyue committed Nov 20, 2024
1 parent 3918f07 commit 29ab673
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions src/agentscope/manager/_model.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
# -*- coding: utf-8 -*-
"""The model manager for AgentScope."""
import json
import yaml
from typing import Any, Union, Sequence

from loguru import logger
Expand Down Expand Up @@ -80,8 +81,12 @@ def load_model_configs(
cfgs = None

if isinstance(model_configs, str):
with open(model_configs, "r", encoding="utf-8") as f:
cfgs = json.load(f)
if model_configs.endswith(".yaml"):
with open(model_configs, "r", encoding="utf-8") as f:
cfgs = yaml.safe_load(f)
else:
with open(model_configs, "r", encoding="utf-8") as f:
cfgs = json.load(f)

if isinstance(model_configs, dict):
cfgs = [model_configs]
Expand Down

0 comments on commit 29ab673

Please sign in to comment.