pycoze 0.1.152__py3-none-any.whl → 0.1.154__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/agent_chat.py CHANGED
@@ -32,7 +32,7 @@ def load_abilities(bot_setting_file: str):
32
32
  return abilities
33
33
 
34
34
 
35
- def agent_chat(bot_setting_file, history):
35
+ def agent_chat(bot_setting_file, history, result_queue):
36
36
  role_setting = load_role_setting(bot_setting_file)
37
37
  abilities = load_abilities(bot_setting_file)
38
38
 
@@ -75,4 +75,5 @@ def agent_chat(bot_setting_file, history):
75
75
  assistant_message=prompt,
76
76
  tool_compatibility_mode=cfg["toolCompatibilityMode"],
77
77
  )
78
- return asyncio.run(run_agent(agent, history, cfg["toolCompatibilityMode"]))
78
+ result = asyncio.run(run_agent(agent, history, cfg["toolCompatibilityMode"]))
79
+ result_queue.put(result_queue)
pycoze/bot/bot.py CHANGED
@@ -1,40 +1,30 @@
1
- from langchain_core.messages import HumanMessage
2
- from .agent import INPUT_MESSAGE, output
3
- from .agent_chat import agent_chat
4
- import json
5
- import threading
6
-
7
-
8
- def chat(bot_setting_file: str):
9
- history = []
10
- stop_event = threading.Event()
11
-
12
- def user_input_listener():
13
- while not stop_event.is_set():
14
- input_text = input()
15
- if input_text:
16
- stop_event.set()
17
- if not input_text.startswith(INPUT_MESSAGE):
18
- raise ValueError("Invalid message")
19
- message = json.loads(input_text[len(INPUT_MESSAGE):])
20
- history.append(HumanMessage(message["content"]))
21
- # 中断当前的机器人输出
22
- print("User interrupted the bot.")
23
- # 重新开始对话
24
- chat(bot_setting_file)
25
-
26
- input_thread = threading.Thread(target=user_input_listener)
27
- input_thread.start()
28
-
29
- while not stop_event.is_set():
30
- result = agent_chat(bot_setting_file, history)
31
- history.append(output("assistant", result))
32
- print(result) # 输出机器人回复
33
-
34
- input_thread.join()
35
-
36
-
37
- def get_chat_response(bot_setting_file: str, input_text: str):
38
- history = [HumanMessage(input_text)]
39
- result = agent_chat(bot_setting_file, history)
40
- return result
1
+ from langchain_core.messages import HumanMessage
2
+ from .agent import INPUT_MESSAGE, output
3
+ from .agent_chat import agent_chat
4
+ import json
5
+ import queue
6
+ import threading
7
+
8
+
9
+ def chat(bot_setting_file: str):
10
+ history = []
11
+ while True:
12
+ input_text = input()
13
+ if not input_text.startswith(INPUT_MESSAGE):
14
+ raise ValueError("Invalid message")
15
+ message = json.loads(input_text[len(INPUT_MESSAGE) :])
16
+ history.append(HumanMessage(message["content"]))
17
+ result_queue = queue.Queue()
18
+ agent_chat_thread = threading.Thread(target=agent_chat, args=(bot_setting_file, history, result_queue))
19
+ agent_chat_thread.start()
20
+ agent_chat_thread.join()
21
+ result = result_queue.get()
22
+ history.append(output("assistant", result))
23
+
24
+
25
+
26
+ def get_chat_response(bot_setting_file: str, input_text: str):
27
+ history = [HumanMessage(input_text)]
28
+ result_queue = queue.Queue()
29
+ result = agent_chat(bot_setting_file, history, result_queue)
30
+ return result
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: pycoze
3
- Version: 0.1.152
3
+ Version: 0.1.154
4
4
  Summary: Package for pycoze only!
5
5
  Author: Yuan Jie Xiong
6
6
  Author-email: aiqqqqqqq@qq.com
@@ -2,8 +2,8 @@ pycoze/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
2
  pycoze/ai/__init__.py,sha256=odM2lgYSnApxw4AzLV_5AyaxL5_MLfydOSB1VmLGFyA,75
3
3
  pycoze/ai/vram_reserve.py,sha256=QbqaA8qv87cnEpOVDMygi0BNMxuhLYwj1UKfR_D5BD4,4340
4
4
  pycoze/bot/__init__.py,sha256=6HHMxDQVOyZM9dtSjQm9tjGnhj4h7CixD0JOvEwTi48,41
5
- pycoze/bot/agent_chat.py,sha256=Ey45GpBHmzILsMee-iXc90mpiQK3VAm_gomd1tSScOI,2808
6
- pycoze/bot/bot.py,sha256=kQrq5Inx6IQr14V00-KyHKWPyNPdJ32-LTt1ylbShp0,1313
5
+ pycoze/bot/agent_chat.py,sha256=QtACf0MizZVv6PTzDreuZzQ6ktjddzqI3gAcZSwfWfs,2860
6
+ pycoze/bot/bot.py,sha256=XIQZpGob0iH-UgVWcnisUsEqHfqpdf-n63ZDzaH4Sy0,1042
7
7
  pycoze/bot/agent/__init__.py,sha256=YR9vpkEQn1e4937r_xFPJXUCPBEJ0SFzEQDBe2x3-YA,157
8
8
  pycoze/bot/agent/agent.py,sha256=Aue8nWeW_I7e1jo4o7cUjFFjrsV9NtVUiTX3EQYHmbA,3507
9
9
  pycoze/bot/agent/assistant.py,sha256=3iLxnRvf_ia0cP-FHK5Fv4ylltlnzPq1KscRCFYqjkc,1147
@@ -23,8 +23,8 @@ pycoze/ui/ui_def.py,sha256=UhhU_yB3GV9ISbvTWT48hsHPHI250BhMILh6bu5Uioo,4206
23
23
  pycoze/utils/__init__.py,sha256=ET0W5wzq4zlY3dr1wHVbbeRKlKdC_zqHt9b0jZyHohw,94
24
24
  pycoze/utils/arg.py,sha256=orLVEGw3x2f3l7bZDbBPkkdDPSdqXo-_Rs-4ZhzmrEw,984
25
25
  pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
26
- pycoze-0.1.152.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
27
- pycoze-0.1.152.dist-info/METADATA,sha256=GABqnA8bFV7XDdbDRhfBZ5PdekFZQcxaV7cHqw-Re_o,726
28
- pycoze-0.1.152.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
29
- pycoze-0.1.152.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
30
- pycoze-0.1.152.dist-info/RECORD,,
26
+ pycoze-0.1.154.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
27
+ pycoze-0.1.154.dist-info/METADATA,sha256=8F6oekNK8HK747zL9c5DdGGzdWAWkUK-4tf24eLAhZg,726
28
+ pycoze-0.1.154.dist-info/WHEEL,sha256=bFJAMchF8aTQGUgMZzHJyDDMPTO3ToJ7x23SLJa1SVo,92
29
+ pycoze-0.1.154.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
30
+ pycoze-0.1.154.dist-info/RECORD,,