feat: 修改一下目录结构

src/mindpilot/app/api/api_schemas.py

@@ -12,11 +12,13 @@ from openai.types.chat import (
     completion_create_params,
 )
 
+from ..utils.openai_utils import MsgType
+
 # from chatchat.configs import DEFAULT_LLM_MODEL, TEMPERATURE
 DEFAULT_LLM_MODEL = None # TODO 配置文件
 TEMPERATURE = 0.8
 from ..pydantic_v2 import AnyUrl, BaseModel, Field
-from ..utils import MsgType
+
 
 
 class OpenAIBaseInput(BaseModel):

src/mindpilot/app/chat/chat.py

@@ -16,14 +16,8 @@ from ..callback_handler.agent_callback_handler import (
     AgentStatus,
 )
 from ..chat.utils import History
-from ..utils import (
-    MsgType,
-    get_ChatOpenAI,
-    get_prompt_template,
-    get_tool,
-    wrap_done,
-)
-from app.configs import MODEL_CONFIG,TOOL_CONFIG
+from ..configs import MODEL_CONFIG, TOOL_CONFIG
+from ..utils.openai_utils import get_ChatOpenAI, get_prompt_template, get_tool, wrap_done, MsgType
 
 
 def create_models_from_config(configs, callbacks, stream):

src/mindpilot/app/utils/colorful.py

@@ -0,0 +1,61 @@
+import platform
+from sys import stdout
+
+if platform.system()=="Linux":
+    pass
+else:
+    from colorama import init
+    init()
+
+# Do you like the elegance of Chinese characters?
+def print红(*kw,**kargs):
+    print("\033[0;31m",*kw,"\033[0m",**kargs)
+def print绿(*kw,**kargs):
+    print("\033[0;32m",*kw,"\033[0m",**kargs)
+def print黄(*kw,**kargs):
+    print("\033[0;33m",*kw,"\033[0m",**kargs)
+def print蓝(*kw,**kargs):
+    print("\033[0;34m",*kw,"\033[0m",**kargs)
+def print紫(*kw,**kargs):
+    print("\033[0;35m",*kw,"\033[0m",**kargs)
+def print靛(*kw,**kargs):
+    print("\033[0;36m",*kw,"\033[0m",**kargs)
+
+def print亮红(*kw,**kargs):
+    print("\033[1;31m",*kw,"\033[0m",**kargs)
+def print亮绿(*kw,**kargs):
+    print("\033[1;32m",*kw,"\033[0m",**kargs)
+def print亮黄(*kw,**kargs):
+    print("\033[1;33m",*kw,"\033[0m",**kargs)
+def print亮蓝(*kw,**kargs):
+    print("\033[1;34m",*kw,"\033[0m",**kargs)
+def print亮紫(*kw,**kargs):
+    print("\033[1;35m",*kw,"\033[0m",**kargs)
+def print亮靛(*kw,**kargs):
+    print("\033[1;36m",*kw,"\033[0m",**kargs)
+
+# Do you like the elegance of Chinese characters?
+def sprint红(*kw):
+    return "\033[0;31m"+' '.join(kw)+"\033[0m"
+def sprint绿(*kw):
+    return "\033[0;32m"+' '.join(kw)+"\033[0m"
+def sprint黄(*kw):
+    return "\033[0;33m"+' '.join(kw)+"\033[0m"
+def sprint蓝(*kw):
+    return "\033[0;34m"+' '.join(kw)+"\033[0m"
+def sprint紫(*kw):
+    return "\033[0;35m"+' '.join(kw)+"\033[0m"
+def sprint靛(*kw):
+    return "\033[0;36m"+' '.join(kw)+"\033[0m"
+def sprint亮红(*kw):
+    return "\033[1;31m"+' '.join(kw)+"\033[0m"
+def sprint亮绿(*kw):
+    return "\033[1;32m"+' '.join(kw)+"\033[0m"
+def sprint亮黄(*kw):
+    return "\033[1;33m"+' '.join(kw)+"\033[0m"
+def sprint亮蓝(*kw):
+    return "\033[1;34m"+' '.join(kw)+"\033[0m"
+def sprint亮紫(*kw):
+    return "\033[1;35m"+' '.join(kw)+"\033[0m"
+def sprint亮靛(*kw):
+    return "\033[1;36m"+' '.join(kw)+"\033[0m"

src/mindpilot/main.py

@@ -8,7 +8,9 @@ from contextlib import asynccontextmanager
 from multiprocessing import Process
 import argparse
 from fastapi import FastAPI
-from app.configs import HOST,PORT
+from app.configs import HOST, PORT
+from src.mindpilot.app.utils.colorful import print亮蓝
+
 
 logger = logging.getLogger()
 
@@ -37,7 +39,7 @@ def run_api_server(
 ):
     import uvicorn
     from app.api.api_server import create_app
-    from app.utils import set_httpx_config
+    from src.mindpilot.app.utils.openai_utils import set_httpx_config
 
     set_httpx_config()
     app = create_app(run_mode=run_mode)
@@ -115,7 +117,8 @@ def main():
     cwd = os.getcwd()
     sys.path.append(cwd)
     multiprocessing.freeze_support()
-    print("cwd:" + cwd)
+    print亮蓝(f"当前工作目录：{cwd}")
+    print亮蓝(f"OpenAPI 文档地址：http://{HOST}:{PORT}/docs")
 
     if sys.version_info < (3, 10):
         loop = asyncio.get_event_loop()