aient 1.2.31__py3-none-any.whl → 1.2.33__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.
- aient/models/chatgpt.py +7 -3
- {aient-1.2.31.dist-info → aient-1.2.33.dist-info}/METADATA +1 -1
- {aient-1.2.31.dist-info → aient-1.2.33.dist-info}/RECORD +6 -6
- {aient-1.2.31.dist-info → aient-1.2.33.dist-info}/WHEEL +0 -0
- {aient-1.2.31.dist-info → aient-1.2.33.dist-info}/licenses/LICENSE +0 -0
- {aient-1.2.31.dist-info → aient-1.2.33.dist-info}/top_level.txt +0 -0
aient/models/chatgpt.py
CHANGED
@@ -125,8 +125,12 @@ class chatgpt(BaseLLM):
|
|
125
125
|
else:
|
126
126
|
# 如果没有提供 logger,创建一个默认的,它只会打印到控制台
|
127
127
|
self.logger = logging.getLogger("chatgpt_default")
|
128
|
+
self.logger.propagate = False
|
128
129
|
if not self.logger.handlers: # 防止重复添加 handler
|
129
|
-
|
130
|
+
handler = logging.StreamHandler()
|
131
|
+
formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
|
132
|
+
handler.setFormatter(formatter)
|
133
|
+
self.logger.addHandler(handler)
|
130
134
|
self.logger.setLevel(logging.INFO if print_log else logging.WARNING)
|
131
135
|
|
132
136
|
# 注册和处理传入的工具
|
@@ -688,7 +692,7 @@ class chatgpt(BaseLLM):
|
|
688
692
|
# 准备会话
|
689
693
|
self.system_prompt = system_prompt or self.system_prompt
|
690
694
|
if convo_id not in self.conversation or pass_history <= 2:
|
691
|
-
self.reset(convo_id=convo_id, system_prompt=system_prompt)
|
695
|
+
self.reset(convo_id=convo_id, system_prompt=self.system_prompt)
|
692
696
|
self.add_to_conversation(prompt, role, convo_id=convo_id, function_name=function_name, total_tokens=total_tokens, function_arguments=function_arguments, pass_history=pass_history, function_call_id=function_call_id)
|
693
697
|
|
694
698
|
# 获取请求体
|
@@ -945,7 +949,7 @@ class chatgpt(BaseLLM):
|
|
945
949
|
"""
|
946
950
|
self.system_prompt = system_prompt or self.system_prompt
|
947
951
|
self.conversation[convo_id] = Messages(
|
948
|
-
SystemMessage(
|
952
|
+
SystemMessage(self.system_prompt, self.conversation[convo_id].provider("files")),
|
949
953
|
)
|
950
954
|
self.tokens_usage[convo_id] = 0
|
951
955
|
self.current_tokens[convo_id] = 0
|
@@ -17,7 +17,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
|
|
17
17
|
aient/models/__init__.py,sha256=ZTiZgbfBPTjIPSKURE7t6hlFBVLRS9lluGbmqc1WjxQ,43
|
18
18
|
aient/models/audio.py,sha256=FNW4lxG1IhxOU7L8mvcbaeC1nXk_lpUZQlg9ijQ0h_Q,1937
|
19
19
|
aient/models/base.py,sha256=HWIGfa2A7OTccvHK0wG1-UlHB-yaWRC7hbi4oR1Mu1Y,7228
|
20
|
-
aient/models/chatgpt.py,sha256=
|
20
|
+
aient/models/chatgpt.py,sha256=eeZ-hK0U50MyXnan-pRXADvMnydgNtKOCQdSqKfHC6M,43565
|
21
21
|
aient/plugins/__init__.py,sha256=p3KO6Aa3Lupos4i2SjzLQw1hzQTigOAfEHngsldrsyk,986
|
22
22
|
aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
|
23
23
|
aient/plugins/config.py,sha256=TGgZ5SnNKZ8MmdznrZ-TEq7s2ulhAAwTSKH89bci3dA,7079
|
@@ -35,8 +35,8 @@ aient/plugins/write_file.py,sha256=Jt8fOEwqhYiSWpCbwfAr1xoi_BmFnx3076GMhuL06uI,3
|
|
35
35
|
aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
|
36
36
|
aient/utils/prompt.py,sha256=ZvGAt_ImJ_CGbDnWgpsWskfSV5fCkpFKRpNQjYL7M7s,11100
|
37
37
|
aient/utils/scripts.py,sha256=Q0tS7E9AmdikO7GeDBd_3Ii5opXHCvKjDGqHsXen6_A,40622
|
38
|
-
aient-1.2.
|
39
|
-
aient-1.2.
|
40
|
-
aient-1.2.
|
41
|
-
aient-1.2.
|
42
|
-
aient-1.2.
|
38
|
+
aient-1.2.33.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
|
39
|
+
aient-1.2.33.dist-info/METADATA,sha256=Sq5S-JuLRFvQ_Qv69V24o7hAokFm4nldoMXx8WeuOcA,4842
|
40
|
+
aient-1.2.33.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
|
41
|
+
aient-1.2.33.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
|
42
|
+
aient-1.2.33.dist-info/RECORD,,
|
File without changes
|
File without changes
|
File without changes
|