refactor llm
This commit is contained in:
@@ -10,7 +10,7 @@ if TYPE_CHECKING:
|
||||
|
||||
from src.classes.event import Event
|
||||
from src.utils.config import CONFIG
|
||||
from src.utils.llm import get_prompt_and_call_llm_async
|
||||
from src.utils.llm import call_llm_with_template, LLMMode
|
||||
from src.run.log import get_logger
|
||||
|
||||
logger = get_logger().logger
|
||||
@@ -73,7 +73,7 @@ async def generate_nickname(avatar: "Avatar") -> Optional[str]:
|
||||
}
|
||||
|
||||
# 调用LLM并自动解析JSON
|
||||
response_data = await get_prompt_and_call_llm_async(template_path, infos, mode="fast")
|
||||
response_data = await call_llm_with_template(template_path, infos, LLMMode.FAST)
|
||||
|
||||
nickname = response_data.get("nickname", "").strip()
|
||||
thinking = response_data.get("thinking", "")
|
||||
|
||||
Reference in New Issue
Block a user