autoflake import fix

binary-husky · binary-husky · commit dae31658ee86 · 2026-01-16T18:07:18.000+08:00
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -10,3 +10,15 @@ repos:
       - id: check-json
       - id: check-merge-conflict
       - id: detect-private-key
+
+
+
+  - repo: https://github.com/myint/autoflake
+    rev: v2.2.0
+    hooks:
+      - id: autoflake
+        args: [
+          --in-place,
+          --remove-all-unused-imports,
+          --ignore-init-module-imports
+        ]
diff --git a/ajet/backbone/main_vllm.py b/ajet/backbone/main_vllm.py
@@ -1,4 +1,3 @@
-import atexit
 import os
 import sys
 from types import SimpleNamespace
@@ -83,7 +82,7 @@ def submit_chat_completions(self, messages, sampling_params, request_id, tools=[
                 "content": message["content"],
                 "tool_calls": message.get("tool_calls", None),
                 "tokens": [
-                    TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content # type: ignore
+                    TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content  # type: ignore
                 ],
             }
         )
@@ -131,7 +130,7 @@ async def submit_chat_completions_async(self, messages, sampling_params, request
                 "content": message["content"],
                 "tool_calls": message.get("tool_calls", None),
                 "tokens": [
-                    TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content # type: ignore
+                    TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content  # type: ignore
                 ],
             }
         )
diff --git a/ajet/backbone/trainer_trinity.py b/ajet/backbone/trainer_trinity.py
@@ -65,7 +65,6 @@ def __init__(
         )
 
     def convert_task(self, task: TrinityTask):
-        from ajet.schema.task import Task
         assert isinstance(task.raw_task, dict)
         return dict_to_ajet_task(task.raw_task)
 
diff --git a/ajet/context_tracker/base_tracker.py b/ajet/context_tracker/base_tracker.py
@@ -1,5 +1,5 @@
 from typing import List, Tuple, Union
-from typing import List, Union, Tuple, Dict, Optional, Any
+from typing import List, Union, Tuple, Dict, Optional
 from ajet.schema.task import WorkflowTask
 
 from ajet.schema.extended_msg import (
diff --git a/ajet/context_tracker/timeline_merging/timeline_merging.py b/ajet/context_tracker/timeline_merging/timeline_merging.py
@@ -1,6 +1,5 @@
 from typing import List
 
-from beast_logger import print_listofdict
 
 from ajet.context_tracker.basic_tracker import ExtendedMessage
 
diff --git a/ajet/schema/convertion.py b/ajet/schema/convertion.py
@@ -4,11 +4,10 @@
 from openai.types.chat.chat_completion_message import ChatCompletionMessage
 from agentscope.model import ChatResponse as AgentScopeChatResponse
 from openai.types.completion_usage import CompletionUsage
-from typing import Any, Callable, Dict, List, Literal, Type, Union
+from typing import List, Type
 from agentscope.message import TextBlock, ToolUseBlock
 from agentscope._utils._common import _json_loads_with_repair
 from pydantic import BaseModel
-from agentscope.model import ChatResponse
 
 
 def convert_llm_proxy_response_to_oai_response(llm_proxy_response):
diff --git a/ajet/task_rollout/async_llm_bridge.py b/ajet/task_rollout/async_llm_bridge.py
@@ -3,14 +3,13 @@
 import json
 import time
 import uuid
-from typing import Any, Callable, Dict, List, Literal, Type, Union
+from typing import Any, Callable, Dict, List, Literal, Union
 
 
 
 from loguru import logger
 from omegaconf import DictConfig
 from pydantic import BaseModel
-from transformers.tokenization_utils import PreTrainedTokenizer
 from vllm.entrypoints.openai.tool_parsers.hermes_tool_parser import Hermes2ProToolParser
 from vllm.outputs import RequestOutput as VerlVllmRequestOutput
 
diff --git a/ajet/task_runner/base_runner.py b/ajet/task_runner/base_runner.py
@@ -3,7 +3,6 @@
 from threading import Lock
 from typing import Any, Callable, Union, Type
 from multiprocessing import Process, Queue
-from unittest import result
 
 from ajet.context_tracker.basic_tracker import BaseContextTracker
 from ajet.schema.task import WorkflowOutput, WorkflowTask
diff --git a/ajet/task_runner/general_runner.py b/ajet/task_runner/general_runner.py
@@ -1,7 +1,6 @@
-from venv import logger
 
 from ajet import AjetTuner
-from ajet import Workflow, WorkflowOutput
+from ajet import WorkflowOutput
 from ajet.context_tracker.multiagent_tracking import (
     MultiAgentContextTracker,
 )
diff --git a/ajet/tuner.py b/ajet/tuner.py
@@ -1,4 +1,4 @@
-from typing import TYPE_CHECKING, Any, Literal, Callable, Union, Type
+from typing import TYPE_CHECKING, Callable, Union, Type
 
 from ajet.context_tracker.multiagent_tracking import (
     MultiAgentContextTracker,
diff --git a/ajet/tuner_lib/weight_tuner/as_agentscope_model.py b/ajet/tuner_lib/weight_tuner/as_agentscope_model.py
@@ -1,7 +1,7 @@
-from typing import TYPE_CHECKING, Any, Literal, Type
+from typing import Any, Literal, Type
 
 from agentscope._utils._common import _create_tool_from_base_model
-from agentscope.model import ChatModelBase, ChatResponse, DashScopeChatModel
+from agentscope.model import ChatResponse, DashScopeChatModel
 from loguru import logger
 from pydantic import BaseModel
 
@@ -10,9 +10,6 @@
 )
 from ajet.task_rollout.async_llm_bridge import AgentScopeLlmProxyWithTracker
 
-if TYPE_CHECKING:
-    from ajet import Workflow
-
 
 class AgentScopeModelTuner(DashScopeChatModel):
     """
diff --git a/ajet/tuner_lib/weight_tuner/as_oai_baseurl_apikey.py b/ajet/tuner_lib/weight_tuner/as_oai_baseurl_apikey.py
@@ -1,22 +1,13 @@
 import os
-import asyncio
-from typing import TYPE_CHECKING, Any, List, Callable, Literal, Type, Union
-from loguru import logger
+from typing import Any
 from pydantic import BaseModel, Field
 from ajet.context_tracker.multiagent_tracking import (
     MultiAgentContextTracker,
 )
-from ajet.task_rollout.async_llm_bridge import OpenaiLlmProxyWithTracker
-from ajet.utils.magic_mock import SpecialMagicMock
-from openai.types.chat.chat_completion import ChatCompletion
-from openai.resources.chat.chat import Chat, AsyncChat
+from openai.resources.chat.chat import AsyncChat
 from openai.resources.completions import AsyncCompletions
-from openai import OpenAI, AsyncOpenAI
-from ajet.utils.networking import find_free_port
 from .experimental.as_oai_model_client import generate_auth_token
 
-if TYPE_CHECKING:
-    from ajet import Workflow
 
 class MockAsyncCompletions(AsyncCompletions):
     async def create(self, *args, **kwargs) -> Any: # type: ignore
diff --git a/ajet/tuner_lib/weight_tuner/as_oai_sdk_model.py b/ajet/tuner_lib/weight_tuner/as_oai_sdk_model.py
@@ -1,19 +1,12 @@
-import asyncio
-from typing import TYPE_CHECKING, Any, List, Callable, Literal, Type, Union
-from loguru import logger
-from pydantic import BaseModel
+from typing import Any, List, Callable
 from ajet.context_tracker.multiagent_tracking import (
     MultiAgentContextTracker,
 )
 from ajet.task_rollout.async_llm_bridge import OpenaiLlmProxyWithTracker
-from ajet.utils.magic_mock import SpecialMagicMock
 from openai.types.chat.chat_completion import ChatCompletion
-from openai.resources.chat.chat import Chat, AsyncChat
+from openai.resources.chat.chat import AsyncChat
 from openai.resources.completions import AsyncCompletions
-from openai import OpenAI, AsyncOpenAI
-
-if TYPE_CHECKING:
-    from ajet import Workflow
+from openai import AsyncOpenAI
 
 
 class MockAsyncCompletions(AsyncCompletions):
diff --git a/ajet/tuner_lib/weight_tuner/experimental/as_oai_model_server.py b/ajet/tuner_lib/weight_tuner/experimental/as_oai_model_server.py
@@ -24,7 +24,7 @@
 
 from loguru import logger
 from pydantic import BaseModel
-from fastapi import FastAPI, Header, HTTPException, Request, Body
+from fastapi import FastAPI, Header, HTTPException, Request
 from contextlib import asynccontextmanager
 from multiprocessing import Process
 from concurrent.futures import ThreadPoolExecutor
diff --git a/ajet/utils/metric_helper/reward_metric_helper.py b/ajet/utils/metric_helper/reward_metric_helper.py
@@ -11,7 +11,7 @@
 - judge_time/                 Judge time consumption statistics
 """
 
-from typing import List, Dict, Any, Optional
+from typing import List, Dict, Any
 import numpy as np
 
 
diff --git a/ajet/utils/msg_converter.py b/ajet/utils/msg_converter.py
@@ -21,8 +21,7 @@
   {"role": "user/assistant/system", "content": "..."}
 """
 
-import json
-from typing import List, Dict, Any, Union
+from typing import List, Dict, Any
 
 
 
diff --git a/ajet/utils/testing_utils.py b/ajet/utils/testing_utils.py
@@ -11,23 +11,20 @@
 from loguru import logger
 
 from ajet.utils.dynamic_import import dynamic_import
-from ajet.utils.sington import singleton
 
 
 class TestSuccessException(Exception):
     """
     All test is done, end the program early with exception.
     """
 
-    pass
 
 
 class TestFailException(Exception):
     """
     Test has failed, end the program early with exception.
     """
 
-    pass
 
 
 class BaseProbe(object):
diff --git a/scripts/display_dataset.py b/scripts/display_dataset.py
@@ -1,10 +1,5 @@
 import argparse
-import glob
-import os
-import time
 
-from beast_logger import print_list
-from huggingface_hub import snapshot_download
 
 parser = argparse.ArgumentParser(description="download Hugging Face dataset")
 parser.add_argument("--target", default="openai/gsm8k", type=str, help="HuggingFace dataset name")
diff --git a/tests/bench/benchmark_math/benchmark_math.py b/tests/bench/benchmark_math/benchmark_math.py
@@ -1,5 +1,4 @@
 # flake8: noqa
-import os
 import time
 
 from ajet.utils.testing_utils import BenchmarkProbe, singleton
diff --git a/tests/test_networking.py b/tests/test_networking.py
@@ -1,6 +1,5 @@
 import socket
 import unittest
-import sys
 import os
 import importlib.util
 
diff --git a/tutorial/example_appworld/appworld.py b/tutorial/example_appworld/appworld.py
@@ -1,5 +1,4 @@
 from agentscope.message import Msg
-from pydantic import Field
 
 from ajet import AjetTuner, Workflow, WorkflowOutput, WorkflowTask
 
diff --git a/tutorial/example_appworld/appworld_oai_sdk.py b/tutorial/example_appworld/appworld_oai_sdk.py
@@ -1,5 +1,4 @@
 from agentscope.message import Msg
-from pydantic import Field
 
 from ajet import Workflow, WorkflowOutput, WorkflowTask
 from ajet import AjetTuner
diff --git a/tutorial/example_learn2ask/learn2ask_langchain.py b/tutorial/example_learn2ask/learn2ask_langchain.py
@@ -4,7 +4,6 @@
 import asyncio
 import threading
 
-from agentscope.message import Msg
 from loguru import logger
 
 from ajet import AjetTuner, Workflow, WorkflowOutput, WorkflowTask
diff --git a/tutorial/example_ma_deepresearch/ma_deepresearch.py b/tutorial/example_ma_deepresearch/ma_deepresearch.py
@@ -2,13 +2,8 @@
 from loguru import logger
 from pydantic import BaseModel, Field
 from ajet import AjetTuner, Workflow, WorkflowOutput, WorkflowTask
-from openai.types.chat.chat_completion import ChatCompletion
-from openai.types.chat import ChatCompletionMessageToolCall
-from textwrap import dedent
 
-import json
 import os
-import asyncio
 import requests
 
 
diff --git a/tutorial/example_math_agent/math_agent_langchain.py b/tutorial/example_math_agent/math_agent_langchain.py
@@ -1,12 +1,6 @@
-from loguru import logger
 from ajet import AjetTuner, Workflow, WorkflowOutput, WorkflowTask
-from openai.types.chat.chat_completion import ChatCompletion
-from openai.types.chat import ChatCompletionMessageToolCall
 from textwrap import dedent
 
-import json
-import asyncio
-import requests
 from langchain.agents import create_agent
 
 
diff --git a/tutorial/example_math_agent/math_agent_oai_sdk.py b/tutorial/example_math_agent/math_agent_oai_sdk.py
@@ -1,4 +1,3 @@
-from loguru import logger
 from ajet import AjetTuner, Workflow, WorkflowOutput, WorkflowTask
 from openai.types.chat.chat_completion import ChatCompletion
 from openai.types.chat import ChatCompletionMessageToolCall
diff --git a/tutorial/example_math_agent/math_agent_raw_http.py b/tutorial/example_math_agent/math_agent_raw_http.py
@@ -1,11 +1,6 @@
-from loguru import logger
 from ajet import AjetTuner, Workflow, WorkflowOutput, WorkflowTask
-from openai.types.chat.chat_completion import ChatCompletion
-from openai.types.chat import ChatCompletionMessageToolCall
 from textwrap import dedent
 
-import json
-import asyncio
 import requests
 
 
diff --git a/tutorial/example_werewolves/start.py b/tutorial/example_werewolves/start.py
@@ -12,7 +12,7 @@
 
 from agentscope.agent import ReActAgent
 from agentscope.formatter import DashScopeMultiAgentFormatter, OpenAIMultiAgentFormatter
-from agentscope.model import DashScopeChatModel, OpenAIChatModel
+from agentscope.model import OpenAIChatModel
 from loguru import logger
 from pydantic import Field
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,3 @@`
`1`		`-import atexit`
`2`	`1`	`import os`
`3`	`2`	`import sys`
`4`	`3`	`from types import SimpleNamespace`
`@@ -83,7 +82,7 @@ def submit_chat_completions(self, messages, sampling_params, request_id, tools=[`
`83`	`82`	`"content": message["content"],`
`84`	`83`	`"tool_calls": message.get("tool_calls", None),`
`85`	`84`	`"tokens": [`
`86`		`- TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content # type: ignore`
	`85`	`+ TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content # type: ignore`
`87`	`86`	`],`
`88`	`87`	`}`
`89`	`88`	`)`
`@@ -131,7 +130,7 @@ async def submit_chat_completions_async(self, messages, sampling_params, request`
`131`	`130`	`"content": message["content"],`
`132`	`131`	`"tool_calls": message.get("tool_calls", None),`
`133`	`132`	`"tokens": [`
`134`		`- TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content # type: ignore`
	`133`	`+ TokenAndProbVllmDebug(t) for t in completion.choices[0].logprobs.content # type: ignore`
`135`	`134`	`],`
`136`	`135`	`}`
`137`	`136`	`)`
Original file line number	Diff line number	Diff line change
`@@ -65,7 +65,6 @@ def __init__(`
`65`	`65`	`)`
`66`	`66`
`67`	`67`	`def convert_task(self, task: TrinityTask):`
`68`		`- from ajet.schema.task import Task`
`69`	`68`	`assert isinstance(task.raw_task, dict)`
`70`	`69`	`return dict_to_ajet_task(task.raw_task)`
`71`	`70`
Original file line number	Diff line number	Diff line change
`@@ -1,6 +1,5 @@`
`1`	`1`	`from typing import List`
`2`	`2`
`3`		`-from beast_logger import print_listofdict`
`4`	`3`
`5`	`4`	`from ajet.context_tracker.basic_tracker import ExtendedMessage`
`6`	`5`
Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,6 @@`
`1`		`-from venv import logger`
`2`	`1`
`3`	`2`	`from ajet import AjetTuner`
`4`		`-from ajet import Workflow, WorkflowOutput`
	`3`	`+from ajet import WorkflowOutput`
`5`	`4`	`from ajet.context_tracker.multiagent_tracking import (`
`6`	`5`	`MultiAgentContextTracker,`
`7`	`6`	`)`
Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from typing import TYPE_CHECKING, Any, Literal, Callable, Union, Type`
	`1`	`+from typing import TYPE_CHECKING, Callable, Union, Type`
`2`	`2`
`3`	`3`	`from ajet.context_tracker.multiagent_tracking import (`
`4`	`4`	`MultiAgentContextTracker,`