From c9d51642fb57ae3112c7407e56fd6ad7e2ff289b Mon Sep 17 00:00:00 2001 From: bridge Date: Thu, 2 Oct 2025 17:44:09 +0800 Subject: [PATCH] update llm to qwen --- src/classes/action.py | 2 +- src/utils/llm.py | 2 ++ static/config.yml | 1 + 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/classes/action.py b/src/classes/action.py index 191dda5..cff3112 100644 --- a/src/classes/action.py +++ b/src/classes/action.py @@ -279,7 +279,7 @@ class MoveToAvatar(DefineAction, ActualActionMixin): target = None if target is None: return StepStatus.COMPLETED, [] - done = self.avatar.is_in_region(target.tile.region) + done = self.avatar.tile == target.tile return (StepStatus.COMPLETED if done else StepStatus.RUNNING), [] def finish(self, avatar_name: str) -> list[Event]: diff --git a/src/utils/llm.py b/src/utils/llm.py index 575e1cb..ce7298e 100644 --- a/src/utils/llm.py +++ b/src/utils/llm.py @@ -29,11 +29,13 @@ def call_llm(prompt: str) -> str: # 从配置中获取模型信息 model_name = CONFIG.llm.model_name api_key = CONFIG.llm.key + base_url = CONFIG.llm.base_url # 调用litellm的completion函数 response = completion( model=model_name, messages=[{"role": "user", "content": prompt}], api_key=api_key, + base_url=base_url, ) # 返回生成的内容 diff --git a/static/config.yml b/static/config.yml index 66e9579..585caf2 100644 --- a/static/config.yml +++ b/static/config.yml @@ -2,6 +2,7 @@ llm: # 填入litellm支持的model name和key model_name: "your-model-name" key: "your-api-key" + base_url: "your-base-url-of-llm" paths: templates: static/templates/