jarvis-ai-assistant 0.1.16__py3-none-any.whl → 0.1.17__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.
Potentially problematic release.
This version of jarvis-ai-assistant might be problematic. Click here for more details.
- jarvis/__init__.py +1 -1
- jarvis/__pycache__/__init__.cpython-313.pyc +0 -0
- jarvis/__pycache__/agent.cpython-313.pyc +0 -0
- jarvis/__pycache__/main.cpython-313.pyc +0 -0
- jarvis/agent.py +8 -6
- jarvis/main.py +5 -2
- jarvis/models/__pycache__/kimi.cpython-313.pyc +0 -0
- jarvis/models/kimi.py +1 -3
- {jarvis_ai_assistant-0.1.16.dist-info → jarvis_ai_assistant-0.1.17.dist-info}/METADATA +1 -1
- {jarvis_ai_assistant-0.1.16.dist-info → jarvis_ai_assistant-0.1.17.dist-info}/RECORD +14 -14
- {jarvis_ai_assistant-0.1.16.dist-info → jarvis_ai_assistant-0.1.17.dist-info}/LICENSE +0 -0
- {jarvis_ai_assistant-0.1.16.dist-info → jarvis_ai_assistant-0.1.17.dist-info}/WHEEL +0 -0
- {jarvis_ai_assistant-0.1.16.dist-info → jarvis_ai_assistant-0.1.17.dist-info}/entry_points.txt +0 -0
- {jarvis_ai_assistant-0.1.16.dist-info → jarvis_ai_assistant-0.1.17.dist-info}/top_level.txt +0 -0
jarvis/__init__.py
CHANGED
|
Binary file
|
|
Binary file
|
|
Binary file
|
jarvis/agent.py
CHANGED
|
@@ -77,12 +77,13 @@ class Agent:
|
|
|
77
77
|
except Exception as e:
|
|
78
78
|
raise Exception(f"{self.name}: 模型调用失败: {str(e)}")
|
|
79
79
|
|
|
80
|
-
def run(self, user_input: str, file_list: Optional[List[str]] = None):
|
|
80
|
+
def run(self, user_input: str, file_list: Optional[List[str]] = None, keep_history: bool = False):
|
|
81
81
|
"""处理用户输入并返回响应,返回任务总结报告
|
|
82
82
|
|
|
83
83
|
Args:
|
|
84
84
|
user_input: 用户输入的任务描述
|
|
85
85
|
file_list: 可选的文件列表,默认为None
|
|
86
|
+
keep_history: 是否保留对话历史,默认为False
|
|
86
87
|
|
|
87
88
|
Returns:
|
|
88
89
|
str: 任务总结报告
|
|
@@ -245,11 +246,12 @@ arguments:
|
|
|
245
246
|
return f"Task failed: {str(e)}"
|
|
246
247
|
|
|
247
248
|
finally:
|
|
248
|
-
#
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
249
|
+
# 只在不保留历史时删除会话
|
|
250
|
+
if not keep_history:
|
|
251
|
+
try:
|
|
252
|
+
self.model.delete_chat()
|
|
253
|
+
except Exception as e:
|
|
254
|
+
PrettyOutput.print(f"清理会话时发生错误: {str(e)}", OutputType.ERROR)
|
|
253
255
|
|
|
254
256
|
def clear_history(self):
|
|
255
257
|
"""清除对话历史,只保留系统提示"""
|
jarvis/main.py
CHANGED
|
@@ -90,6 +90,7 @@ def main():
|
|
|
90
90
|
# Add argument parser
|
|
91
91
|
parser = argparse.ArgumentParser(description='Jarvis AI Assistant')
|
|
92
92
|
parser.add_argument('-f', '--files', nargs='*', help='List of files to process')
|
|
93
|
+
parser.add_argument('--keep-history', action='store_true', help='Keep chat history (do not delete chat session)')
|
|
93
94
|
args = parser.parse_args()
|
|
94
95
|
|
|
95
96
|
load_env_from_file()
|
|
@@ -125,6 +126,8 @@ def main():
|
|
|
125
126
|
|
|
126
127
|
# 欢迎信息
|
|
127
128
|
PrettyOutput.print(f"Jarvis 已初始化 - With Kimi", OutputType.SYSTEM)
|
|
129
|
+
if args.keep_history:
|
|
130
|
+
PrettyOutput.print("已启用历史保留模式", OutputType.INFO)
|
|
128
131
|
|
|
129
132
|
# 加载预定义任务
|
|
130
133
|
tasks = load_tasks()
|
|
@@ -132,7 +135,7 @@ def main():
|
|
|
132
135
|
selected_task = select_task(tasks)
|
|
133
136
|
if selected_task:
|
|
134
137
|
PrettyOutput.print(f"\n执行任务: {selected_task}", OutputType.INFO)
|
|
135
|
-
agent.run(selected_task, args.files)
|
|
138
|
+
agent.run(selected_task, args.files, keep_history=args.keep_history)
|
|
136
139
|
return 0
|
|
137
140
|
|
|
138
141
|
# 如果没有选择预定义任务,进入交互模式
|
|
@@ -141,7 +144,7 @@ def main():
|
|
|
141
144
|
user_input = get_multiline_input("请输入您的任务(输入空行退出):")
|
|
142
145
|
if not user_input or user_input == "__interrupt__":
|
|
143
146
|
break
|
|
144
|
-
agent.run(user_input, args.files)
|
|
147
|
+
agent.run(user_input, args.files, keep_history=args.keep_history)
|
|
145
148
|
except Exception as e:
|
|
146
149
|
PrettyOutput.print(f"错误: {str(e)}", OutputType.ERROR)
|
|
147
150
|
|
|
Binary file
|
jarvis/models/kimi.py
CHANGED
|
@@ -261,7 +261,7 @@ class KimiModel(BaseModel):
|
|
|
261
261
|
response = requests.delete(url, headers=headers)
|
|
262
262
|
if response.status_code == 200:
|
|
263
263
|
PrettyOutput.print("会话已删除", OutputType.SUCCESS)
|
|
264
|
-
self.
|
|
264
|
+
self.reset()
|
|
265
265
|
return True
|
|
266
266
|
else:
|
|
267
267
|
PrettyOutput.print(f"删除会话失败: HTTP {response.status_code}", OutputType.ERROR)
|
|
@@ -272,8 +272,6 @@ class KimiModel(BaseModel):
|
|
|
272
272
|
|
|
273
273
|
def reset(self):
|
|
274
274
|
"""重置对话"""
|
|
275
|
-
if self.chat_id:
|
|
276
|
-
self.delete_chat() # 删除现有会话
|
|
277
275
|
self.chat_id = ""
|
|
278
276
|
self.uploaded_files = []
|
|
279
277
|
self.first_chat = True # 重置first_chat标记
|
|
@@ -1,20 +1,20 @@
|
|
|
1
|
-
jarvis/__init__.py,sha256=
|
|
2
|
-
jarvis/agent.py,sha256=
|
|
3
|
-
jarvis/main.py,sha256=
|
|
1
|
+
jarvis/__init__.py,sha256=xxLxgf6AkpQC_oecmvL8lRlgdfb5m-km2HjhnO8aW1U,50
|
|
2
|
+
jarvis/agent.py,sha256=f8RIOFIiwRAuql4O32P_nDXByzBKbrNM-ergf_F8FfU,9409
|
|
3
|
+
jarvis/main.py,sha256=Llbg-3oV5GkGxergaYbAByAunNol9cvBRitYURvjy4s,7051
|
|
4
4
|
jarvis/utils.py,sha256=YipQpEuIRwFE3y3YrgGlSVLEPDrlbBNY1gRiOJix9DU,6602
|
|
5
|
-
jarvis/__pycache__/__init__.cpython-313.pyc,sha256=
|
|
6
|
-
jarvis/__pycache__/agent.cpython-313.pyc,sha256=
|
|
7
|
-
jarvis/__pycache__/main.cpython-313.pyc,sha256=
|
|
5
|
+
jarvis/__pycache__/__init__.cpython-313.pyc,sha256=TpUmCVKoqlVLwQ1cPv_nR5XhD2vW505asH_ltcKmAko,209
|
|
6
|
+
jarvis/__pycache__/agent.cpython-313.pyc,sha256=anFGVa4pfpsOCDbstPAPSsUmglCZUrRkuDN-jnvhSyI,12152
|
|
7
|
+
jarvis/__pycache__/main.cpython-313.pyc,sha256=1Dc3AKvb_cgJnl-aZOkV1_dXxhrcZnxtczzo0uthRtY,9674
|
|
8
8
|
jarvis/__pycache__/models.cpython-313.pyc,sha256=uWuRIjGrY4YDB3dGW5PGDLWaS03et8g11O725TjY_eU,5960
|
|
9
9
|
jarvis/__pycache__/tools.cpython-313.pyc,sha256=lAD4LrnnWzNZQmHXGfZ_2l7oskOpr2_2OC-gdFhxQY8,33933
|
|
10
10
|
jarvis/__pycache__/utils.cpython-313.pyc,sha256=Jpkf2GOA1d_fgrRftqZch1yiO0mB75LWHzQk8IfhHRU,8977
|
|
11
11
|
jarvis/__pycache__/zte_llm.cpython-313.pyc,sha256=kMm9IGundGmOPqjsgrm9oIaWLDagYGCPRAaE3ipkc-0,5662
|
|
12
12
|
jarvis/models/__init__.py,sha256=B_IJFvKTaxdg19FAD1ea288tYp3-bRYRpkeGI0_OcBI,262
|
|
13
13
|
jarvis/models/base.py,sha256=GgVl5N0qDqn-yqRcX_PX3wHjogouE6GPFAWktG40cXg,403
|
|
14
|
-
jarvis/models/kimi.py,sha256=
|
|
14
|
+
jarvis/models/kimi.py,sha256=CR3rU4wPSUQjmYOP516NXLU6h3YtcF6taB-4da3-rA4,10690
|
|
15
15
|
jarvis/models/__pycache__/__init__.cpython-313.pyc,sha256=jAwySX4diR7EWM_alK75tiIb_J8bVfs4Bh_U3bdjDLo,534
|
|
16
16
|
jarvis/models/__pycache__/base.cpython-313.pyc,sha256=4I9KZlXHvTB7vENA9YWK4Fx0sns_KvIOtWqzE9y_-Co,1094
|
|
17
|
-
jarvis/models/__pycache__/kimi.cpython-313.pyc,sha256
|
|
17
|
+
jarvis/models/__pycache__/kimi.cpython-313.pyc,sha256=-hOwhZrfvqfd5K_c34Tm-uR3gaZ9r6EV1NtwV0-FDmg,13647
|
|
18
18
|
jarvis/tools/__init__.py,sha256=000SvX1Z0yzKtuVR9NS45r41DHUVpE6EiSEzzE99uR0,196
|
|
19
19
|
jarvis/tools/base.py,sha256=qm73kMOkqAWHjUisUWFye-x69LJuZdwm0U3Tct-mmMc,3890
|
|
20
20
|
jarvis/tools/file_ops.py,sha256=zTksx45NZm3iz9itN5iQGZ8DoxnSeTHdrnF08_ix7PU,3770
|
|
@@ -32,9 +32,9 @@ jarvis/tools/__pycache__/user_confirmation.cpython-313.pyc,sha256=wK3Ev10lHSUSRv
|
|
|
32
32
|
jarvis/tools/__pycache__/user_input.cpython-313.pyc,sha256=JjTFOhObKsKF4Pn8KBRuKfV1_Ssj083fjU7Mfc_5z7c,2531
|
|
33
33
|
jarvis/tools/__pycache__/user_interaction.cpython-313.pyc,sha256=RuVZ-pmiPBDywY3efgXSfohMAciC1avMGPmBK5qlnew,3305
|
|
34
34
|
jarvis/tools/__pycache__/webpage.cpython-313.pyc,sha256=BjzSfnNzsKCrLETCcWjt32lNDLzwnjqcVGg4JfWd9OM,3008
|
|
35
|
-
jarvis_ai_assistant-0.1.
|
|
36
|
-
jarvis_ai_assistant-0.1.
|
|
37
|
-
jarvis_ai_assistant-0.1.
|
|
38
|
-
jarvis_ai_assistant-0.1.
|
|
39
|
-
jarvis_ai_assistant-0.1.
|
|
40
|
-
jarvis_ai_assistant-0.1.
|
|
35
|
+
jarvis_ai_assistant-0.1.17.dist-info/LICENSE,sha256=AGgVgQmTqFvaztRtCAXsAMryUymB18gZif7_l2e1XOg,1063
|
|
36
|
+
jarvis_ai_assistant-0.1.17.dist-info/METADATA,sha256=qrsMqKWDAyI_fvhoS18eKxn5TQ_28vIuzgIkb1-90yU,6525
|
|
37
|
+
jarvis_ai_assistant-0.1.17.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
|
|
38
|
+
jarvis_ai_assistant-0.1.17.dist-info/entry_points.txt,sha256=iKu7OMfew9dtfGhW71gIMTg4wvafuPqKb4wyQOnMAGU,44
|
|
39
|
+
jarvis_ai_assistant-0.1.17.dist-info/top_level.txt,sha256=1BOxyWfzOP_ZXj8rVTDnNCJ92bBGB0rwq8N1PCpoMIs,7
|
|
40
|
+
jarvis_ai_assistant-0.1.17.dist-info/RECORD,,
|
|
File without changes
|
|
File without changes
|
{jarvis_ai_assistant-0.1.16.dist-info → jarvis_ai_assistant-0.1.17.dist-info}/entry_points.txt
RENAMED
|
File without changes
|
|
File without changes
|