xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-07-20 19:36:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

style: introduce ruff as linter and formatter (#1356)

Browse Source 
* style: remove necessary imports

* style: fix F841

* style: fix F401

* style: fix F811

* style: fix E402

* style: fix E721

* style: fix E722

* style: fix E722

* style: fix F541

* style: ruff format

* style: all passed

* style: add ruff in deps

* style: more ignores in ruff.toml

* style: add pre-commit
This commit is contained in:
Junyan Qin (Chin)
2025-04-29 17:24:07 +08:00
committed by GitHub
parent 09e70d70e9
commit 209f16af76
240 changed files with 5307 additions and 4689 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/provider/runners/localagent.py
+21 -13
View File
@@ -4,24 +4,28 @@ import json
import typing
from .. import runner
from ...core import app, entities as core_entities
from ...core import entities as core_entities
from .. import entities as llm_entities
@runner.runner_class("local-agent")
@runner.runner_class('local-agent')
class LocalAgentRunner(runner.RequestRunner):
"""本地Agent请求运行器
"""
"""本地Agent请求运行器"""
async def run(self, query: core_entities.Query) -> typing.AsyncGenerator[llm_entities.Message, None]:
"""运行请求
"""
async def run(
self, query: core_entities.Query
) -> typing.AsyncGenerator[llm_entities.Message, None]:
"""运行请求"""
pending_tool_calls = []
req_messages = query.prompt.messages.copy() + query.messages.copy() + [query.user_message]
req_messages = (
query.prompt.messages.copy() + query.messages.copy() + [query.user_message]
)
# 首次请求
msg = await query.use_llm_model.requester.invoke_llm(query, query.use_llm_model, req_messages, query.use_funcs)
msg = await query.use_llm_model.requester.invoke_llm(
query, query.use_llm_model, req_messages, query.use_funcs
)
yield msg
@@ -34,7 +38,7 @@ class LocalAgentRunner(runner.RequestRunner):
for tool_call in pending_tool_calls:
try:
func = tool_call.function
parameters = json.loads(func.arguments)
func_ret = await self.ap.tool_mgr.execute_func_call(
@@ -42,7 +46,9 @@ class LocalAgentRunner(runner.RequestRunner):
)
msg = llm_entities.Message(
role="tool", content=json.dumps(func_ret, ensure_ascii=False), tool_call_id=tool_call.id
role='tool',
content=json.dumps(func_ret, ensure_ascii=False),
tool_call_id=tool_call.id,
)
yield msg
@@ -51,7 +57,7 @@ class LocalAgentRunner(runner.RequestRunner):
except Exception as e:
# 工具调用出错，添加一个报错信息到 req_messages
err_msg = llm_entities.Message(
role="tool", content=f"err: {e}", tool_call_id=tool_call.id
role='tool', content=f'err: {e}', tool_call_id=tool_call.id
)
yield err_msg
@@ -59,7 +65,9 @@ class LocalAgentRunner(runner.RequestRunner):
req_messages.append(err_msg)
# 处理完所有调用，再次请求
msg = await query.use_llm_model.requester.invoke_llm(query, query.use_llm_model, req_messages, query.use_funcs)
msg = await query.use_llm_model.requester.invoke_llm(
query, query.use_llm_model, req_messages, query.use_funcs
)
yield msg