diff --git a/llmkira/receiver/function.py b/llmkira/receiver/function.py index 9898ad2f1..45f65a7cf 100644 --- a/llmkira/receiver/function.py +++ b/llmkira/receiver/function.py @@ -208,8 +208,8 @@ async def run_pending_task(task: TaskHeader, pending_task: TaskBatch): _env_dict = await EnvManager.from_uid(uid=task.receiver.uid).get_env_list(name_list=_tool_obj.env_list) assert isinstance(_env_dict, dict), "unexpected env dict? it should be dict..." # Resign Chain - if (task.task_meta.resign_next_step or task.task_meta.is_complete( - num_end=1)) and not _tool_obj.repeatable: + if ((task.task_meta.resign_next_step or task.task_meta.is_complete(num_end=1)) + and not _tool_obj.repeatable): logger.debug(f"Function {pending_task.get_batch_name()} need resign chain") await chain_func.resign_chain( task=task, @@ -248,7 +248,7 @@ async def process_function_call(self, message: AbstractIncomingMessage logger.trace("No Function Call") return None pending_task: TaskBatch - logger.debug(f"Received A Batch FunctionRequest") + logger.debug("Received A Batch FunctionRequest") try: await self.run_pending_task(task=task, pending_task=pending_task) except Exception as e: diff --git a/llmkira/receiver/receiver_client.py b/llmkira/receiver/receiver_client.py index f56f243cf..d6c055c44 100644 --- a/llmkira/receiver/receiver_client.py +++ b/llmkira/receiver/receiver_client.py @@ -202,7 +202,7 @@ async def llm_request( raise Exception(e) except httpx.RemoteProtocolError as e: logger.error(f"[Network RemoteProtocolError] {e}") - raise ReplyNeededError(message=f"Server disconnected without sending a response.") + raise ReplyNeededError(message="Server disconnected without sending a response.") except ServiceUnavailableError as e: logger.error(f"[Service Unavailable Error] {e}") raise ReplyNeededError(message=f"[551721]Service Unavailable {e}") @@ -214,7 +214,7 @@ async def llm_request( raise ReplyNeededError(message=f"[551183]Service Provider Error {e}") except PydanticValidationError as e: logger.exception(e) - raise ReplyNeededError(message=f"[551684]Request Data ValidationError") + raise ReplyNeededError(message="[551684]Request Data ValidationError") except Exception as e: logger.exception(e) raise e