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 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 cfg["model"].startswith("deepseek") and len(tools) > 0:
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,
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: pycoze
3
- Version: 0.1.80
3
+ Version: 0.1.81
4
4
  Summary: Package for pycoze only!
5
5
  Author: Yuan Jie Xiong
6
6
  Author-email: aiqqqqqqq@qq.com
@@ -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=_oWYL0ay9kcFOfXVYgiL83YT7QvGighG6SyuFGaS4wg,2607
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.80.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
25
- pycoze-0.1.80.dist-info/METADATA,sha256=m9ZMjtlfWH5bd9cD3dewBIM50XVnNiLn59iKrpbr2IQ,719
26
- pycoze-0.1.80.dist-info/WHEEL,sha256=eOLhNAGa2EW3wWl_TU484h7q1UNgy0JXjjoqKoxAAQc,92
27
- pycoze-0.1.80.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
28
- pycoze-0.1.80.dist-info/RECORD,,
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,,