添加豆包思考模式的引导配置

.gitignore

@@ -1,4 +1,6 @@
 __pycache__
 build
 dist
-*.spec
+*.spec
+.venv
+.log

.python-version

@@ -0,0 +1 @@
+3.12

llm/doubao.py

@@ -27,6 +27,7 @@ def message(self, message: list[dict]) -> str:
             top_p=self.model_config.get("top_p"),
             frequency_penalty=self.model_config.get("frequency_penalty"),
             presence_penalty=self.model_config.get("presence_penalty"),
+            reasoning_effort=self.model_config.get("reasoning_effort"),
         )
         return response.choices[0].message.content
 
@@ -39,6 +40,7 @@ def stream_message(self, message: list[dict]) -> Generator:
             top_p=self.model_config.get("top_p"),
             frequency_penalty=self.model_config.get("frequency_penalty"),
             presence_penalty=self.model_config.get("presence_penalty"),
+            reasoning_effort=self.model_config.get("reasoning_effort"),
             stream=True,
         )
         for chunk in response:
@@ -62,6 +64,7 @@ def call_tool(self, tools: dict, message: list[dict]) -> tuple[str, str, dict, d
             top_p=self.model_config.get("top_p"),
             frequency_penalty=self.model_config.get("frequency_penalty"),
             presence_penalty=self.model_config.get("presence_penalty"),
+            reasoning_effort=self.model_config.get("reasoning_effort"),
             tools=tools,
         )
         response = response.choices[0].message

main.py

@@ -0,0 +1,6 @@
+def main():
+    print("Hello from baize!")
+
+
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -0,0 +1,64 @@
+[project]
+name = "baize-0-2"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "altgraph==0.17.4",
+    "annotated-types==0.7.0",
+    "anyio==4.4.0",
+    "asciinet==0.3.1",
+    "astor==0.8.1",
+    "cachetools==5.3.3",
+    "certifi==2024.6.2",
+    "cffi==1.16.0",
+    "charset-normalizer==3.3.2",
+    "colorama==0.4.6",
+    "cryptography==42.0.8",
+    "distro==1.9.0",
+    "h11==0.14.0",
+    "httpcore==1.0.5",
+    "httpx==0.27.0",
+    "idna==3.7",
+    "macholib==1.16.3",
+    "markdown-it-py==3.0.0",
+    "mdurl==0.1.2",
+    "msgpack-python==0.5.6",
+    "natsort==8.4.0",
+    "networkx==3.3",
+    "ollama==0.2.1",
+    "openai==1.34.0",
+    "packaging==24.1",
+    "pdfminer-six==20231228",
+    "pdfplumber==0.11.1",
+    "pefile==2023.2.7",
+    "pexpect==4.9.0",
+    "pillow==10.3.0",
+    "pretty-errors==1.2.25",
+    "ptyprocess==0.7.0",
+    "pyboxmaker==1.0",
+    "pycparser==2.22",
+    "pydantic==2.7.4",
+    "pydantic-core==2.18.4",
+    "pygments==2.18.0",
+    "pyinstaller==6.8.0",
+    "pyinstaller-hooks-contrib==2024.7",
+    "pyjwt==2.8.0",
+    "pypdfium2==4.30.0",
+    "pyperclip==1.8.2",
+    "python-dateutil==2.9.0.post0",
+    "pywin32-ctypes==0.2.2",
+    "pyyaml==6.0.1",
+    "requests==2.32.3",
+    "rich==13.7.1",
+    "setuptools==69.5.1",
+    "six==1.16.0",
+    "sniffio==1.3.1",
+    "tqdm==4.66.4",
+    "typing-extensions==4.12.2",
+    "urllib3==2.2.1",
+    "volcengine-python-sdk>=4.0.36",
+    "wheel==0.43.0",
+    "zhipuai==2.1.0.20240521",
+]

utils/setup.py

@@ -160,6 +160,8 @@ def setup():
         api_key = input_param('API_KEY')
         endpoint_id = input_param('推理接入点 ID')
         new_config['endpoint_id'] = endpoint_id
+        reasoning_effort = input_param('思考等级')
+        new_config['reasoning_effort'] = reasoning_effort
         model_name = 'doubao'
     elif model_family == 'ollama':
         base_url = input_param('BASE_URL', True)