diff --git a/arkitect/core/component/llm/llm.py b/arkitect/core/component/llm/llm.py
index cb764564..165b493a 100644
--- a/arkitect/core/component/llm/llm.py
+++ b/arkitect/core/component/llm/llm.py
@@ -14,8 +14,8 @@
 
 from typing import Any, Callable, Dict, List, Optional, Union
 
-from langchain.prompts.chat import BaseChatPromptTemplate
-from langchain.schema.output_parser import BaseTransformOutputParser
+from langchain_core.output_parsers import BaseTransformOutputParser
+from langchain_core.prompts.chat import BaseChatPromptTemplate
 from volcenginesdkarkruntime import Ark, AsyncArk
 from volcenginesdkarkruntime._streaming import AsyncStream, Stream
 from volcenginesdkarkruntime.types.chat import (
diff --git a/arkitect/core/component/llm/utils.py b/arkitect/core/component/llm/utils.py
index 86874df1..89177c54 100644
--- a/arkitect/core/component/llm/utils.py
+++ b/arkitect/core/component/llm/utils.py
@@ -16,7 +16,6 @@
 import logging
 from typing import Any, Dict, List, Set, Union
 
-from langchain.prompts.chat import BaseChatPromptTemplate
 from langchain_core.messages import (
     AIMessage,
     BaseMessage,
@@ -25,18 +24,19 @@
     SystemMessage,
 )
 from langchain_core.messages.tool import ToolCall
+from langchain_core.prompts.chat import BaseChatPromptTemplate
 from typing_extensions import Literal
 from volcenginesdkarkruntime.types.chat import ChatCompletionMessage
 from volcenginesdkarkruntime.types.chat.chat_completion_chunk import ChoiceDelta
 
 from arkitect.core.errors import InvalidParameter
+from arkitect.core.utils.converter import to_dict
 from arkitect.telemetry.trace import task
 from arkitect.types.llm.model import (
     ArkMessage,
     ChatCompletionMessageToolCallParam,
     Function,
 )
-from arkitect.core.utils.converter import to_dict
 
 
 def _convert_message_role_to_ark_role(  # type: ignore
diff --git a/arkitect/core/component/prompts/custom_prompt.py b/arkitect/core/component/prompts/custom_prompt.py
index 22aeacbf..614d76fd 100644
--- a/arkitect/core/component/prompts/custom_prompt.py
+++ b/arkitect/core/component/prompts/custom_prompt.py
@@ -21,14 +21,14 @@
 import jinja2
 import pytz
 from jinja2 import Template
-from langchain.prompts.chat import BaseChatPromptTemplate
-from langchain.schema.messages import (
+from langchain_core.messages import (
     AIMessage,
     BaseMessage,
     HumanMessage,
     SystemMessage,
     get_buffer_string,
 )
+from langchain_core.prompts.chat import BaseChatPromptTemplate
 from pydantic.v1 import validator
 
 from arkitect.types.llm.model import ArkChatRequest
diff --git a/demohouse/live_voice_call/backend/prompt.py b/demohouse/live_voice_call/backend/prompt.py
index c8f3f8c4..6e2f042a 100644
--- a/demohouse/live_voice_call/backend/prompt.py
+++ b/demohouse/live_voice_call/backend/prompt.py
@@ -11,8 +11,8 @@
 
 from typing import Any, List
 
-from langchain.prompts.chat import BaseChatPromptTemplate
 from langchain_core.messages import AnyMessage, BaseMessage, SystemMessage
+from langchain_core.prompts.chat import BaseChatPromptTemplate
 
 SYSTEM_PROMPT = """
 # 角色任务