mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
import re
|
|
from importlib import import_module
|
|
from datetime import datetime
|
|
|
|
from enumeration.message_role_enum import MessageRoleEnum
|
|
|
|
|
|
def under_line_to_hump(underline_str):
|
|
sub = re.sub(r"(_\w)", lambda x: x.group(1)[1].upper(), underline_str)
|
|
return sub[0:1].upper() + sub[1:]
|
|
|
|
|
|
def init_instance_by_config(
|
|
config: dict, default_clazz_path: str = "", suffix_name: str = "", **kwargs
|
|
):
|
|
clazz_path = config.pop("clazz")
|
|
if not clazz_path:
|
|
raise RuntimeError("empty clazz_path!")
|
|
clazz_name_split = clazz_path.split(".")
|
|
clazz_name: str = clazz_name_split[-1]
|
|
if suffix_name and not clazz_name.endswith(suffix_name):
|
|
clazz_name = f"{clazz_name}_{suffix_name}"
|
|
|
|
# 构造path
|
|
clazz_paths = []
|
|
if default_clazz_path:
|
|
clazz_paths.append(default_clazz_path)
|
|
clazz_paths.extend(clazz_name_split[:-1])
|
|
clazz_paths.append(clazz_name)
|
|
module = import_module(".".join(clazz_paths))
|
|
|
|
cls_name = under_line_to_hump(clazz_name)
|
|
return getattr(module, cls_name)(**config, **kwargs)
|
|
|
|
|
|
def complete_config_name(config_name: str, suffix: str = ".json"):
|
|
if not config_name.endswith(suffix):
|
|
config_name += suffix
|
|
return config_name
|
|
|
|
|
|
def prompt_to_msg(system_prompt: str, few_shot: str, user_query: str):
|
|
return [
|
|
{
|
|
"role": MessageRoleEnum.SYSTEM.value,
|
|
"content": system_prompt.strip(),
|
|
},
|
|
{
|
|
"role": MessageRoleEnum.USER.value,
|
|
"content": "\n".join(
|
|
[x.strip() for x in [few_shot, system_prompt, user_query]]
|
|
),
|
|
},
|
|
]
|
|
|
|
|
|
def get_datetime_info_dict(parse_dt: datetime):
|
|
return {
|
|
"year": parse_dt.year,
|
|
"month": parse_dt.month,
|
|
"day": parse_dt.day,
|
|
"hour": parse_dt.hour,
|
|
"minute": parse_dt.minute,
|
|
"second": parse_dt.second,
|
|
"week": parse_dt.isocalendar().week,
|
|
"weekday": WEEKDAYS[parse_dt.isocalendar().weekday - 1],
|
|
}
|
|
|
|
|
|
def time_to_formatted_str(
|
|
time: datetime | str | int | float = None,
|
|
date_format: str = "%Y%m%d", # e.g. %Y%m%d -> "20240528", add %H:%M:%S
|
|
string_format: str = "",
|
|
) -> str:
|
|
if isinstance(time, str | int | float):
|
|
if isinstance(time, str):
|
|
time = float(time)
|
|
current_dt = datetime.fromtimestamp(time)
|
|
elif isinstance(time, datetime):
|
|
current_dt = time
|
|
else:
|
|
current_dt = datetime.now()
|
|
|
|
return_str = ""
|
|
if date_format:
|
|
return_str = current_dt.strftime(date_format)
|
|
elif string_format:
|
|
return_str = string_format.format(**get_datetime_info_dict(current_dt))
|
|
|
|
return return_str
|