Browse Source

feat:模型配置后端实现

main
guojialiang 1 year ago
parent
commit
004afeafca
11 changed files with 11 additions and 11 deletions
  1. +1
    -1
      src/mindpilot/app/api/api_schemas.py
  2. +3
    -3
      src/mindpilot/app/chat/utils.py
  3. +1
    -1
      src/mindpilot/app/tools/arxiv.py
  4. +1
    -1
      src/mindpilot/app/tools/calculate.py
  5. +1
    -1
      src/mindpilot/app/tools/search_internet.py
  6. +1
    -1
      src/mindpilot/app/tools/shell.py
  7. +1
    -1
      src/mindpilot/app/tools/tools_registry.py
  8. +1
    -1
      src/mindpilot/app/tools/weather_check.py
  9. +1
    -1
      src/mindpilot/app/tools/wolfram.py
  10. +0
    -0
      src/mindpilot/app/utils/pydantic_v1.py
  11. +0
    -0
      src/mindpilot/app/utils/pydantic_v2.py

+ 1
- 1
src/mindpilot/app/api/api_schemas.py View File

@@ -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):


+ 3
- 3
src/mindpilot/app/chat/utils.py View File

@@ -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()



+ 1
- 1
src/mindpilot/app/tools/arxiv.py View File

@@ -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



+ 1
- 1
src/mindpilot/app/tools/calculate.py View File

@@ -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



+ 1
- 1
src/mindpilot/app/tools/search_internet.py View File

@@ -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


+ 1
- 1
src/mindpilot/app/tools/shell.py View File

@@ -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



+ 1
- 1
src/mindpilot/app/tools/tools_registry.py View File

@@ -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"]



+ 1
- 1
src/mindpilot/app/tools/weather_check.py View File

@@ -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


+ 1
- 1
src/mindpilot/app/tools/wolfram.py View File

@@ -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


src/mindpilot/app/pydantic_v1.py → src/mindpilot/app/utils/pydantic_v1.py View File


src/mindpilot/app/pydantic_v2.py → src/mindpilot/app/utils/pydantic_v2.py View File


Loading…
Cancel
Save