perf: minor fixes

This commit is contained in:
Junyan Qin
2025-08-03 15:30:11 +08:00
parent 47ff883fc7
commit b9f768af25
6 changed files with 14 additions and 23 deletions

View File

@@ -148,8 +148,6 @@ class MessageChunk(pydantic.BaseModel):
tool_call_id: typing.Optional[str] = None
# tool_calls: typing.Optional[list[ToolCallChunk]] = None
is_final: bool = False
def readable_str(self) -> str:

View File

@@ -98,7 +98,6 @@ class ProviderAPIRequester(metaclass=abc.ABCMeta):
"""
pass
@abc.abstractmethod
async def invoke_llm_stream(
self,
query: core_entities.Query,

View File

@@ -116,7 +116,6 @@ class LocalAgentRunner(runner.RequestRunner):
query.use_llm_model,
req_messages,
query.use_funcs,
stream=is_stream,
extra_args=query.use_llm_model.model_entity.extra_args,
):
assert isinstance(msg, llm_entities.MessageChunk)
@@ -178,10 +177,8 @@ class LocalAgentRunner(runner.RequestRunner):
query.use_llm_model,
req_messages,
query.use_funcs,
stream=is_stream,
extra_args=query.use_llm_model.model_entity.extra_args,
):
assert isinstance(msg, llm_entities.MessageChunk)
yield msg
if msg.tool_calls:
for tool_call in msg.tool_calls: