pycoze 0.1.80__py3-none-any.whl → 0.1.81__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- pycoze/bot/bot.py +12 -3
- {pycoze-0.1.80.dist-info → pycoze-0.1.81.dist-info}/METADATA +1 -1
- {pycoze-0.1.80.dist-info → pycoze-0.1.81.dist-info}/RECORD +6 -6
- {pycoze-0.1.80.dist-info → pycoze-0.1.81.dist-info}/LICENSE +0 -0
- {pycoze-0.1.80.dist-info → pycoze-0.1.81.dist-info}/WHEEL +0 -0
- {pycoze-0.1.80.dist-info → pycoze-0.1.81.dist-info}/top_level.txt +0 -0
pycoze/bot/bot.py
CHANGED
@@ -5,6 +5,7 @@ import asyncio
|
|
5
5
|
from langchain_core.messages import HumanMessage
|
6
6
|
from pycoze import utils
|
7
7
|
from pycoze.access.tool_for_bot import import_tools
|
8
|
+
from langchain_core.utils.function_calling import convert_to_openai_tool
|
8
9
|
|
9
10
|
params = utils.arg.read_params_file()
|
10
11
|
llm_file = params["appPath"] + "/JsonStorage/llm.json"
|
@@ -36,7 +37,7 @@ def agent_chat(bot_setting_file, history):
|
|
36
37
|
model=cfg["model"],
|
37
38
|
temperature=(
|
38
39
|
role_setting["temperature"] * 2
|
39
|
-
if cfg["model"].startswith("deepseek")
|
40
|
+
if cfg["model"].startswith("deepseek") or cfg["model"].startswith("yi-")
|
40
41
|
else role_setting["temperature"]
|
41
42
|
),
|
42
43
|
stop_sequences=[
|
@@ -45,15 +46,23 @@ def agent_chat(bot_setting_file, history):
|
|
45
46
|
], # 停用deepseek的工具调用标记,不然会虚构工具调用过程和结果
|
46
47
|
)
|
47
48
|
prompt = role_setting["prompt"]
|
48
|
-
if
|
49
|
+
if (
|
50
|
+
cfg["model"].startswith("deepseek")
|
51
|
+
or cfg["model"].startswith("yi-")
|
52
|
+
and len(tools) > 0
|
53
|
+
):
|
49
54
|
prompt += """
|
50
55
|
如果需要调用工具,请使用以正确的json格式进行结尾(务必保证json格式正确,不要出现反斜杠未转义等问题):
|
51
56
|
```json
|
52
57
|
{"name": 函数名, "parameters": 参数词典}
|
53
58
|
```
|
54
59
|
"""
|
60
|
+
if cfg["model"].startswith("yi-"):
|
61
|
+
prompt += "\nAvailable functions:\n"
|
62
|
+
for t in tools:
|
63
|
+
prompt += f"\n```json\n{json.dumps(convert_to_openai_tool(t))}\n```"
|
55
64
|
agent = Runnable(
|
56
|
-
"FuncCall",
|
65
|
+
agent_execution_mode="FuncCall",
|
57
66
|
tools=tools,
|
58
67
|
llm=chat,
|
59
68
|
assistant_message=prompt,
|
@@ -6,7 +6,7 @@ pycoze/ai/__init__.py,sha256=Smivpb8qbRnzWkzKRe2IxsmKP5Dh8EvngDFdkD_DVLo,73
|
|
6
6
|
pycoze/ai/comfyui.py,sha256=u75tZywkuXiOdm7XST2kBAaveJKpPvY_qTQr_TN9sXk,795
|
7
7
|
pycoze/ai/vram_reserve.py,sha256=s55Cy-Q5mTq-k5oIPbAFwCfrjatjN0QTjQxW7WBTPZI,5738
|
8
8
|
pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
|
9
|
-
pycoze/bot/bot.py,sha256=
|
9
|
+
pycoze/bot/bot.py,sha256=t5wtxigZO23qVhVrXBNVu1WYc8jBAvbWWBH1QNwmjXU,3006
|
10
10
|
pycoze/bot/agent/__init__.py,sha256=YR9vpkEQn1e4937r_xFPJXUCPBEJ0SFzEQDBe2x3-YA,157
|
11
11
|
pycoze/bot/agent/agent.py,sha256=3504Q6IlgNirLma_bLcIxkBjTvmAWSjKzv6da7A6c2Y,3307
|
12
12
|
pycoze/bot/agent/assistant.py,sha256=eEcu8aS0m-Kmkv5QttnVeirJrsjqqUJ6iwLZsvwlSMY,1049
|
@@ -21,8 +21,8 @@ pycoze/ui/ui_def.py,sha256=UhhU_yB3GV9ISbvTWT48hsHPHI250BhMILh6bu5Uioo,4206
|
|
21
21
|
pycoze/utils/__init__.py,sha256=TNJhFfY7JYdLlzuP9GvgxfNXUtbgH_NUUJSqHXCxJn4,78
|
22
22
|
pycoze/utils/arg.py,sha256=kA3KBQzXc2WlH5XbF8kfikfpqljiKaW7oY_GE4Qyffc,753
|
23
23
|
pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
|
24
|
-
pycoze-0.1.
|
25
|
-
pycoze-0.1.
|
26
|
-
pycoze-0.1.
|
27
|
-
pycoze-0.1.
|
28
|
-
pycoze-0.1.
|
24
|
+
pycoze-0.1.81.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
|
25
|
+
pycoze-0.1.81.dist-info/METADATA,sha256=qccbO0_mIWODm1vfnkbWLNY2qC0_gdRJ5Kqs1j3_XEY,719
|
26
|
+
pycoze-0.1.81.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
|
27
|
+
pycoze-0.1.81.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
|
28
|
+
pycoze-0.1.81.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|