mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
88 lines
2.6 KiB
Python
88 lines
2.6 KiB
Python
import re
|
|
from datetime import datetime
|
|
from importlib import import_module
|
|
|
|
from memory_scope.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_class_path: str = "memory_scope", suffix_name: str = "", **kwargs):
|
|
class_name = config.pop("class")
|
|
if not class_name:
|
|
raise RuntimeError("empty class_name!")
|
|
|
|
class_name_split = class_name.split(".")
|
|
class_name: str = class_name_split[-1]
|
|
if suffix_name and not class_name.lower().endswith(suffix_name.lower()):
|
|
class_name = f"{class_name}_{suffix_name}"
|
|
class_name_split[-1] = class_name
|
|
|
|
class_paths = []
|
|
if default_class_path:
|
|
class_paths.append(default_class_path)
|
|
class_paths.extend(class_name_split)
|
|
module = import_module(".".join(class_paths))
|
|
|
|
cls_name = under_line_to_hump(class_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
|