diff --git a/src/mindpilot/app/api/api_schemas.py b/src/mindpilot/app/api/api_schemas.py index 18e565a..39b0e1b 100644 --- a/src/mindpilot/app/api/api_schemas.py +++ b/src/mindpilot/app/api/api_schemas.py @@ -15,7 +15,7 @@ from openai.types.chat import ( from ..utils.system_utils import MsgType TEMPERATURE = 0.8 -from ..pydantic_v2 import AnyUrl, BaseModel, Field +from src.mindpilot.app.utils.pydantic_v2 import AnyUrl, BaseModel, Field class OpenAIBaseInput(BaseModel): diff --git a/src/mindpilot/app/chat/utils.py b/src/mindpilot/app/chat/utils.py index 7476b0a..4f52e6f 100644 --- a/src/mindpilot/app/chat/utils.py +++ b/src/mindpilot/app/chat/utils.py @@ -1,8 +1,8 @@ -from typing import Dict, List, Tuple, Union +from typing import List, Union from langchain.prompts.chat import ChatMessagePromptTemplate -from ..pydantic_v2 import BaseModel, Field +from src.mindpilot.app.utils.pydantic_v2 import BaseModel, Field import logging -from typing import AsyncGenerator, Dict, Iterable, Tuple +from typing import Dict, Tuple logger = logging.getLogger() diff --git a/src/mindpilot/app/tools/arxiv.py b/src/mindpilot/app/tools/arxiv.py index f8ff68b..c4ef2fa 100644 --- a/src/mindpilot/app/tools/arxiv.py +++ b/src/mindpilot/app/tools/arxiv.py @@ -1,5 +1,5 @@ # LangChain 的 ArxivQueryRun 工具 -from ..pydantic_v1 import Field +from src.mindpilot.app.utils.pydantic_v1 import Field from .tools_registry import BaseToolOutput, regist_tool diff --git a/src/mindpilot/app/tools/calculate.py b/src/mindpilot/app/tools/calculate.py index 46cd400..d40ee8b 100644 --- a/src/mindpilot/app/tools/calculate.py +++ b/src/mindpilot/app/tools/calculate.py @@ -1,4 +1,4 @@ -from ..pydantic_v1 import Field +from src.mindpilot.app.utils.pydantic_v1 import Field from .tools_registry import BaseToolOutput, regist_tool diff --git a/src/mindpilot/app/tools/search_internet.py b/src/mindpilot/app/tools/search_internet.py index ae581ce..2f79ab5 100644 --- a/src/mindpilot/app/tools/search_internet.py +++ b/src/mindpilot/app/tools/search_internet.py @@ -8,7 +8,7 @@ from markdownify import markdownify from strsimpy.normalized_levenshtein import NormalizedLevenshtein -from ..pydantic_v1 import Field +from src.mindpilot.app.utils.pydantic_v1 import Field # from chatchat.server.utils import get_tool_config from .tools_registry import BaseToolOutput, regist_tool diff --git a/src/mindpilot/app/tools/shell.py b/src/mindpilot/app/tools/shell.py index c709d96..b5abbfa 100644 --- a/src/mindpilot/app/tools/shell.py +++ b/src/mindpilot/app/tools/shell.py @@ -1,7 +1,7 @@ # LangChain 的 Shell 工具 from langchain_community.tools import ShellTool -from ..pydantic_v1 import Field +from src.mindpilot.app.utils.pydantic_v1 import Field from .tools_registry import BaseToolOutput, regist_tool diff --git a/src/mindpilot/app/tools/tools_registry.py b/src/mindpilot/app/tools/tools_registry.py index b4ba0fa..d43f97e 100644 --- a/src/mindpilot/app/tools/tools_registry.py +++ b/src/mindpilot/app/tools/tools_registry.py @@ -5,7 +5,7 @@ from typing import Any, Callable, Dict, Optional, Tuple, Type, Union from langchain.agents import tool from langchain_core.tools import BaseTool -from ..pydantic_v1 import BaseModel, Extra +from src.mindpilot.app.utils.pydantic_v1 import BaseModel, Extra __all__ = ["regist_tool", "BaseToolOutput"] diff --git a/src/mindpilot/app/tools/weather_check.py b/src/mindpilot/app/tools/weather_check.py index 1eef750..1ed1dd1 100644 --- a/src/mindpilot/app/tools/weather_check.py +++ b/src/mindpilot/app/tools/weather_check.py @@ -3,7 +3,7 @@ """ import requests -from ..pydantic_v1 import Field +from src.mindpilot.app.utils.pydantic_v1 import Field from .tools_registry import BaseToolOutput, regist_tool from ..utils.system_utils import get_tool_config diff --git a/src/mindpilot/app/tools/wolfram.py b/src/mindpilot/app/tools/wolfram.py index 8461ae4..7f43dd9 100644 --- a/src/mindpilot/app/tools/wolfram.py +++ b/src/mindpilot/app/tools/wolfram.py @@ -1,6 +1,6 @@ # Langchain 自带的 Wolfram Alpha API 封装 -from ..pydantic_v1 import Field +from src.mindpilot.app.utils.pydantic_v1 import Field from .tools_registry import BaseToolOutput, regist_tool from ..utils.system_utils import get_tool_config diff --git a/src/mindpilot/app/pydantic_v1.py b/src/mindpilot/app/utils/pydantic_v1.py similarity index 100% rename from src/mindpilot/app/pydantic_v1.py rename to src/mindpilot/app/utils/pydantic_v1.py diff --git a/src/mindpilot/app/pydantic_v2.py b/src/mindpilot/app/utils/pydantic_v2.py similarity index 100% rename from src/mindpilot/app/pydantic_v2.py rename to src/mindpilot/app/utils/pydantic_v2.py