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 +3 -2
- pycoze/bot/bot.py +30 -40
- {pycoze-0.1.152.dist-info → pycoze-0.1.154.dist-info}/METADATA +1 -1
- {pycoze-0.1.152.dist-info → pycoze-0.1.154.dist-info}/RECORD +7 -7
- {pycoze-0.1.152.dist-info → pycoze-0.1.154.dist-info}/LICENSE +0 -0
- {pycoze-0.1.152.dist-info → pycoze-0.1.154.dist-info}/WHEEL +0 -0
- {pycoze-0.1.152.dist-info → pycoze-0.1.154.dist-info}/top_level.txt +0 -0
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
|
-
|
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
|
6
|
-
|
7
|
-
|
8
|
-
|
9
|
-
|
10
|
-
|
11
|
-
|
12
|
-
|
13
|
-
|
14
|
-
|
15
|
-
|
16
|
-
|
17
|
-
|
18
|
-
|
19
|
-
|
20
|
-
|
21
|
-
|
22
|
-
|
23
|
-
|
24
|
-
|
25
|
-
|
26
|
-
|
27
|
-
|
28
|
-
|
29
|
-
|
30
|
-
|
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
|
@@ -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=
|
6
|
-
pycoze/bot/bot.py,sha256=
|
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.
|
27
|
-
pycoze-0.1.
|
28
|
-
pycoze-0.1.
|
29
|
-
pycoze-0.1.
|
30
|
-
pycoze-0.1.
|
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,,
|
File without changes
|
File without changes
|
File without changes
|