diff --git a/nonebot_plugin_marshoai/handler.py b/nonebot_plugin_marshoai/handler.py
index cf9ec97..4afc9fc 100644
--- a/nonebot_plugin_marshoai/handler.py
+++ b/nonebot_plugin_marshoai/handler.py
@@ -20,7 +20,6 @@ from nonebot.matcher import (
 from nonebot_plugin_alconna.uniseg import UniMessage, UniMsg
 from openai import AsyncOpenAI, AsyncStream
 from openai.types.chat import ChatCompletion, ChatCompletionChunk, ChatCompletionMessage
-from openai.types.chat.chat_completion import Choice
 
 from .config import config
 from .constants import SUPPORT_IMAGE_MODELS
diff --git a/nonebot_plugin_marshoai/marsho.py b/nonebot_plugin_marshoai/marsho.py
index f32a18b..6219bdc 100644
--- a/nonebot_plugin_marshoai/marsho.py
+++ b/nonebot_plugin_marshoai/marsho.py
@@ -30,7 +30,6 @@ from .handler import MarshoHandler
 from .hooks import *
 from .instances import client, context, model_name, target_list, tools
 from .metadata import metadata
-from .models import MarshoContext
 from .plugin.func_call.caller import get_function_calls
 from .util import *
 from .utils.request import process_chat_stream