jarvis-ai-assistant 0.1.220__py3-none-any.whl → 0.1.221__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.
jarvis/__init__.py CHANGED
@@ -1,4 +1,4 @@
1
1
  # -*- coding: utf-8 -*-
2
2
  """Jarvis AI Assistant"""
3
3
 
4
- __version__ = "0.1.220"
4
+ __version__ = "0.1.221"
@@ -9,6 +9,16 @@ from typing import Any, Callable, Dict, List, Optional, Protocol, Tuple, Union
9
9
 
10
10
  # 本地库导入
11
11
  # jarvis_agent 相关
12
+ from jarvis.jarvis_agent.prompt_builder import build_action_prompt
13
+ from jarvis.jarvis_agent.protocols import OutputHandlerProtocol
14
+ from jarvis.jarvis_agent.session_manager import SessionManager
15
+ from jarvis.jarvis_agent.tool_executor import execute_tool_call
16
+ from jarvis.jarvis_agent.prompts import (
17
+ DEFAULT_SUMMARY_PROMPT,
18
+ SUMMARY_REQUEST_PROMPT,
19
+ TASK_ANALYSIS_PROMPT,
20
+ )
21
+
12
22
  # jarvis_platform 相关
13
23
  from jarvis.jarvis_platform.base import BasePlatform
14
24
  from jarvis.jarvis_platform.registry import PlatformRegistry
@@ -67,32 +77,12 @@ origin_agent_system_prompt = f"""
67
77
  """
68
78
 
69
79
 
70
- class OutputHandlerProtocol(Protocol):
71
- def name(self) -> str:
72
- ...
73
-
74
- def can_handle(self, response: str) -> bool:
75
- ...
76
-
77
- def prompt(self) -> str:
78
- ...
79
-
80
- def handle(self, response: str, agent: Any) -> Tuple[bool, Any]:
81
- ...
82
-
83
-
84
80
  class Agent:
85
81
  def clear(self):
86
- """清除当前对话历史,保留系统消息。
87
-
88
- 该方法将:
89
- 1. 调用模型的delete_chat方法清除对话历史
90
- 2. 重置对话长度计数器
91
- 3. 清空当前提示
92
82
  """
93
- self.model.reset() # type: ignore
94
- self.conversation_length = 0
95
- self.prompt = ""
83
+ Clears the current conversation history by delegating to the session manager.
84
+ """
85
+ self.session.clear()
96
86
 
97
87
  def __del__(self):
98
88
  # 只有在记录启动时才停止记录
@@ -144,7 +134,9 @@ class Agent:
144
134
  if isinstance(platform, str):
145
135
  self.model = PlatformRegistry().create_platform(platform)
146
136
  if self.model is None:
147
- PrettyOutput.print(f"平台 {platform} 不存在,将使用普通模型", OutputType.WARNING)
137
+ PrettyOutput.print(
138
+ f"平台 {platform} 不存在,将使用普通模型", OutputType.WARNING
139
+ )
148
140
  self.model = PlatformRegistry().get_normal_platform()
149
141
  else:
150
142
  self.model = platform
@@ -160,6 +152,9 @@ class Agent:
160
152
 
161
153
  self.model.set_suppress_output(False)
162
154
 
155
+ # Initialize the session manager
156
+ self.session = SessionManager(model=self.model, agent_name=self.name)
157
+
163
158
  from jarvis.jarvis_tools.registry import ToolRegistry
164
159
 
165
160
  self.output_handler = output_handler if output_handler else [ToolRegistry()]
@@ -180,13 +175,10 @@ class Agent:
180
175
  self.use_analysis = (
181
176
  use_analysis if use_analysis is not None else is_use_analysis()
182
177
  )
183
- self.prompt = ""
184
- self.conversation_length = 0 # Use length counter instead
185
178
  self.system_prompt = system_prompt
186
179
  self.input_handler = input_handler if input_handler is not None else []
187
180
  self.need_summary = need_summary
188
181
  # Load configuration from environment variables
189
- self.addon_prompt = ""
190
182
 
191
183
  self.after_tool_call_cb: Optional[Callable[[Agent], None]] = None
192
184
 
@@ -197,24 +189,7 @@ class Agent:
197
189
  )
198
190
 
199
191
  self.summary_prompt = (
200
- summary_prompt
201
- if summary_prompt
202
- else f"""<report>
203
- 请生成任务执行的简明总结报告,包括:
204
-
205
- <content>
206
- 1. 任务目标:任务重述
207
- 2. 执行结果:成功/失败
208
- 3. 关键信息:执行过程中提取的重要信息
209
- 4. 重要发现:任何值得注意的发现
210
- 5. 后续建议:如果有的话
211
- </content>
212
-
213
- <format>
214
- 请使用简洁的要点描述,突出重要信息。
215
- </format>
216
- </report>
217
- """
192
+ summary_prompt if summary_prompt else DEFAULT_SUMMARY_PROMPT
218
193
  )
219
194
 
220
195
  self.max_token_count = get_max_token_count()
@@ -223,46 +198,7 @@ class Agent:
223
198
 
224
199
  PrettyOutput.print(welcome_message, OutputType.SYSTEM)
225
200
 
226
- action_prompt = """
227
- <actions>
228
- # 🧰 可用操作
229
- 以下是您可以使用的操作:
230
- """
231
-
232
- # 添加工具列表概览
233
- action_prompt += "\n<overview>\n## Action List\n"
234
- action_prompt += (
235
- "[" + ", ".join([handler.name() for handler in self.output_handler]) + "]"
236
- )
237
- action_prompt += "\n</overview>"
238
-
239
- # 添加每个工具的详细说明
240
- action_prompt += "\n\n<details>\n# 📝 Action Details\n"
241
- for handler in self.output_handler:
242
- action_prompt += f"\n<tool>\n## {handler.name()}\n"
243
- # 获取工具的提示词并确保格式正确
244
- handler_prompt = handler.prompt().strip()
245
- # 调整缩进以保持层级结构
246
- handler_prompt = "\n".join(
247
- " " + line if line.strip() else line
248
- for line in handler_prompt.split("\n")
249
- )
250
- action_prompt += handler_prompt + "\n</tool>\n"
251
-
252
- # 添加工具使用总结
253
- action_prompt += """
254
- </details>
255
-
256
- <rules>
257
- # ❗ 重要操作使用规则
258
- 1. 一次对话只能使用一个操作,否则会出错
259
- 2. 严格按照每个操作的格式执行
260
- 3. 等待操作结果后再进行下一个操作
261
- 4. 处理完结果后再调用新的操作
262
- 5. 如果对操作使用不清楚,请请求帮助
263
- </rules>
264
- </actions>
265
- """
201
+ action_prompt = build_action_prompt(self.output_handler) # type: ignore
266
202
 
267
203
  self.model.set_system_prompt(
268
204
  f"""
@@ -274,12 +210,12 @@ class Agent:
274
210
  self.first = True
275
211
 
276
212
  def set_user_data(self, key: str, value: Any):
277
- """设置用户数据"""
278
- self.user_data[key] = value
213
+ """Sets user data in the session."""
214
+ self.session.set_user_data(key, value)
279
215
 
280
216
  def get_user_data(self, key: str) -> Optional[Any]:
281
- """获取用户数据"""
282
- return self.user_data.get(key, None)
217
+ """Gets user data from the session."""
218
+ return self.session.get_user_data(key)
283
219
 
284
220
  def set_use_tools(self, use_tools):
285
221
  """设置要使用的工具列表"""
@@ -292,12 +228,8 @@ class Agent:
292
228
  break
293
229
 
294
230
  def set_addon_prompt(self, addon_prompt: str):
295
- """设置附加提示。
296
-
297
- 参数:
298
- addon_prompt: 附加提示内容
299
- """
300
- self.addon_prompt = addon_prompt
231
+ """Sets the addon prompt in the session."""
232
+ self.session.set_addon_prompt(addon_prompt)
301
233
 
302
234
  def set_after_tool_call_cb(self, cb: Callable[[Any], None]): # type: ignore
303
235
  """设置工具调用后回调函数。
@@ -308,41 +240,12 @@ class Agent:
308
240
  self.after_tool_call_cb = cb
309
241
 
310
242
  def save_session(self) -> bool:
311
- """保存当前会话状态到文件"""
312
- if not self.model:
313
- PrettyOutput.print("没有可用的模型实例来保存会话。", OutputType.ERROR)
314
- return False
315
- session_dir = os.path.join(os.getcwd(), ".jarvis")
316
- os.makedirs(session_dir, exist_ok=True)
317
- platform_name = self.model.platform_name()
318
- model_name = self.model.name().replace("/", "_").replace("\\", "_")
319
- session_file = os.path.join(
320
- session_dir, f"saved_session_{self.name}_{platform_name}_{model_name}.json"
321
- )
322
- return self.model.save(session_file)
243
+ """Saves the current session state by delegating to the session manager."""
244
+ return self.session.save_session()
323
245
 
324
246
  def restore_session(self) -> bool:
325
- """从文件恢复会话状态"""
326
- if not self.model:
327
- return False # No model, cannot restore
328
- platform_name = self.model.platform_name()
329
- model_name = self.model.name().replace("/", "_").replace("\\", "_")
330
- session_file = os.path.join(
331
- os.getcwd(),
332
- ".jarvis",
333
- f"saved_session_{self.name}_{platform_name}_{model_name}.json",
334
- )
335
- if not os.path.exists(session_file):
336
- return False
337
-
338
- if self.model.restore(session_file):
339
- try:
340
- os.remove(session_file)
341
- PrettyOutput.print("会话已恢复,并已删除会话文件。", OutputType.SUCCESS)
342
- except OSError as e:
343
- PrettyOutput.print(f"删除会话文件失败: {e}", OutputType.ERROR)
344
- return True
345
- return False
247
+ """Restores the session state by delegating to the session manager."""
248
+ return self.session.restore_session()
346
249
 
347
250
  def get_tool_registry(self) -> Optional[Any]:
348
251
  """获取工具注册表实例"""
@@ -374,21 +277,23 @@ class Agent:
374
277
  if need_return:
375
278
  return message
376
279
 
377
- if self.addon_prompt:
378
- message += f"\n\n{self.addon_prompt}"
379
- self.addon_prompt = ""
280
+ if self.session.addon_prompt:
281
+ message += f"\n\n{self.session.addon_prompt}"
282
+ self.session.addon_prompt = ""
380
283
  else:
381
284
  message += f"\n\n{self.make_default_addon_prompt(need_complete)}"
382
285
 
383
286
  # 累加对话长度
384
- self.conversation_length += get_context_token_count(message)
287
+ self.session.conversation_length += get_context_token_count(message)
385
288
 
386
- if self.conversation_length > self.max_token_count:
289
+ if self.session.conversation_length > self.max_token_count:
387
290
  message = self._summarize_and_clear_history() + "\n\n" + message
388
- self.conversation_length += get_context_token_count(message)
291
+ self.session.conversation_length += get_context_token_count(message)
389
292
 
293
+ if not self.model:
294
+ raise RuntimeError("Model not initialized")
390
295
  response = self.model.chat_until_success(message) # type: ignore
391
- self.conversation_length += get_context_token_count(response)
296
+ self.session.conversation_length += get_context_token_count(response)
392
297
 
393
298
  return response
394
299
 
@@ -402,31 +307,12 @@ class Agent:
402
307
  仅生成摘要,不修改对话状态
403
308
  """
404
309
  print("📄 正在总结对话历史...")
405
- summary_prompt = """
406
- <summary_request>
407
- <objective>
408
- 请对当前对话历史进行简明扼要的总结,提取关键信息和重要决策点。这个总结将作为上下文继续任务,因此需要保留对后续对话至关重要的内容。
409
- </objective>
410
-
411
- <guidelines>
412
- 1. 提取关键信息:任务目标、已确定的事实、重要决策、达成的共识
413
- 2. 保留技术细节:命令、代码片段、文件路径、配置设置等技术细节
414
- 3. 记录任务进展:已完成的步骤、当前所处阶段、待解决的问题
415
- 4. 包含用户偏好:用户表达的明确偏好、限制条件或特殊要求
416
- 5. 省略冗余内容:问候语、重复信息、不相关的讨论
417
- </guidelines>
418
-
419
- <format>
420
- - 使用简洁、客观的语言
421
- - 按时间顺序或主题组织信息
422
- - 使用要点列表增强可读性
423
- - 总结应控制在500词以内
424
- </format>
425
- </summary_request>
426
- """
427
-
428
310
  try:
429
- summary = self.model.chat_until_success(self.prompt + "\n" + summary_prompt) # type: ignore
311
+ if not self.model:
312
+ raise RuntimeError("Model not initialized")
313
+ summary = self.model.chat_until_success(
314
+ self.session.prompt + "\n" + SUMMARY_REQUEST_PROMPT
315
+ ) # type: ignore
430
316
  print("✅ 总结对话历史完成")
431
317
  return summary
432
318
  except Exception as e:
@@ -486,46 +372,10 @@ class Agent:
486
372
  os.remove(tmp_file_name)
487
373
 
488
374
  def _call_tools(self, response: str) -> Tuple[bool, Any]:
489
- """调用工具执行响应
490
-
491
- 参数:
492
- response: 包含工具调用信息的响应字符串
493
-
494
- 返回:
495
- Tuple[bool, Any]:
496
- - 第一个元素表示是否需要返回结果
497
- - 第二个元素是返回结果或错误信息
498
-
499
- 注意:
500
- 1. 一次只能执行一个工具
501
- 2. 如果配置了确认选项,会在执行前请求用户确认
502
- 3. 使用spinner显示执行状态
503
375
  """
504
- tool_list = []
505
- for handler in self.output_handler:
506
- if handler.can_handle(response):
507
- tool_list.append(handler)
508
- if len(tool_list) > 1:
509
- PrettyOutput.print(
510
- f"操作失败:检测到多个操作。一次只能执行一个操作。尝试执行的操作:{', '.join([handler.name() for handler in tool_list])}",
511
- OutputType.WARNING,
512
- )
513
- return (
514
- False,
515
- f"操作失败:检测到多个操作。一次只能执行一个操作。尝试执行的操作:{', '.join([handler.name() for handler in tool_list])}",
516
- )
517
- if len(tool_list) == 0:
518
- return False, ""
519
-
520
- if not self.execute_tool_confirm or user_confirm(
521
- f"需要执行{tool_list[0].name()}确认执行?", True
522
- ):
523
- print(f"🔧 正在执行{tool_list[0].name()}...")
524
- result = tool_list[0].handle(response, self)
525
- print(f"✅ {tool_list[0].name()}执行完成")
526
-
527
- return result
528
- return False, ""
376
+ Delegates the tool execution to the external `execute_tool_call` function.
377
+ """
378
+ return execute_tool_call(response, self)
529
379
 
530
380
  def _complete_task(self) -> str:
531
381
  """完成任务并生成总结(如果需要)
@@ -542,8 +392,10 @@ class Agent:
542
392
  self._analysis_task()
543
393
  if self.need_summary:
544
394
  print("📄 正在生成总结...")
545
- self.prompt = self.summary_prompt
546
- ret = self.model.chat_until_success(self.prompt) # type: ignore
395
+ self.session.prompt = self.summary_prompt
396
+ if not self.model:
397
+ raise RuntimeError("Model not initialized")
398
+ ret = self.model.chat_until_success(self.session.prompt) # type: ignore
547
399
  print("✅ 总结生成完成")
548
400
  return ret
549
401
 
@@ -553,154 +405,12 @@ class Agent:
553
405
  print("🔍 正在分析任务...")
554
406
  try:
555
407
  # 让模型判断是否需要生成方法论
556
- analysis_prompt = f"""<task_analysis>
557
- <request>
558
- 当前任务已结束,请分析该任务的解决方案:
559
- 1. 首先检查现有工具或方法论是否已经可以完成该任务,如果可以,直接说明即可,无需生成新内容
560
- 2. 如果现有工具/方法论不足,评估当前任务是否可以通过编写新工具来自动化解决
561
- 3. 如果可以通过工具解决,请设计并提供工具代码
562
- 4. 如果无法通过编写通用工具完成,评估当前的执行流程是否可以总结为通用方法论
563
- 5. 如果以上都不可行,给出详细理由
564
- 请根据分析结果采取相应行动:说明现有工具/方法论、创建新工具、生成新方法论或说明原因。
565
- </request>
566
- <evaluation_criteria>
567
- 现有资源评估:
568
- 1. 现有工具 - 检查系统中是否已有可以完成该任务的工具
569
- 2. 现有方法论 - 检查是否已有适用于该任务的方法论
570
- 3. 组合使用 - 评估现有工具和方法论组合使用是否可以解决问题
571
- 工具评估标准:
572
- 1. 通用性 - 该工具是否可以解决一类问题,而不仅仅是当前特定问题
573
- 2. 自动化 - 该工具是否可以减少人工干预,提高效率
574
- 3. 可靠性 - 该工具是否可以在不同场景下稳定工作
575
- 4. 简单性 - 该工具是否易于使用,参数设计是否合理
576
- 方法论评估标准:
577
- 1. 方法论应聚焦于通用且可重复的解决方案流程
578
- 2. 方法论应该具备足够的通用性,可应用于同类问题
579
- 3. 特别注意用户在执行过程中提供的修正、反馈和改进建议
580
- 4. 如果用户明确指出了某个解决步骤的优化方向,这应该被纳入方法论
581
- 5. 方法论要严格按照实际的执行流程来总结,不要遗漏或增加任何步骤
582
- </evaluation_criteria>
583
- <tool_requirements>
584
- 工具代码要求:
585
- 1. 工具类名应与工具名称保持一致
586
- 2. 必须包含name、description、parameters属性
587
- 3. 必须实现execute方法处理输入参数
588
- 4. 可选实现check方法验证环境
589
- 5. 工具描述应详细说明用途、适用场景和使用示例
590
- 6. 参数定义应遵循JSON Schema格式
591
- 7. 不要包含特定任务的细节,保持通用性
592
- 工具设计关键点:
593
- 1. **使用PrettyOutput打印执行过程**:强烈建议在工具中使用PrettyOutput显示执行过程,
594
- 这样用户可以了解工具在做什么,提升用户体验。示例:
595
- ```python
596
- from jarvis.jarvis_utils.output import PrettyOutput, OutputType
597
- # 执行中打印信息
598
- PrettyOutput.print("正在处理数据...", OutputType.INFO)
599
- # 成功信息
600
- PrettyOutput.print("操作成功完成", OutputType.SUCCESS)
601
- # 警告信息
602
- PrettyOutput.print("发现潜在问题", OutputType.WARNING)
603
- # 错误信息
604
- PrettyOutput.print("操作失败", OutputType.ERROR)
605
- ```
606
- 2. **结构化返回结果**:工具应该始终返回结构化的结果字典,包含以下字段:
607
- - success: 布尔值,表示操作是否成功
608
- - stdout: 字符串,包含工具的主要输出内容
609
- - stderr: 字符串,包含错误信息(如果有)
610
- 3. **异常处理**:工具应该妥善处理可能发生的异常,并在失败时清理已创建的资源
611
- ```python
612
- try:
613
- # 执行逻辑
614
- return {{
615
- "success": True,
616
- "stdout": "成功结果",
617
- "stderr": ""
618
- }}
619
- except Exception as e:
620
- PrettyOutput.print(f"操作失败: {{str(e)}}", OutputType.ERROR)
621
- # 清理资源(如果有创建)
622
- return {{
623
- "success": False,
624
- "stdout": "",
625
- "stderr": f"操作失败: {{str(e)}}"
626
- }}
627
- ```
628
- </tool_requirements>
629
- <methodology_requirements>
630
- 方法论格式要求:
631
- 1. 问题重述: 简明扼要的问题归纳,不含特定细节
632
- 2. 最优解决方案: 经过用户验证的、最终有效的解决方案(将每个步骤要使用的工具也列举出来)
633
- 3. 注意事项: 执行中可能遇到的常见问题和注意点,尤其是用户指出的问题
634
- 4. 可选步骤: 对于有多种解决路径的问题,标注出可选步骤和适用场景
635
- </methodology_requirements>
636
- <output_requirements>
637
- 根据分析结果,输出以下三种情况之一:
638
- 1. 如果现有工具/方法论可以解决,直接输出说明:
639
- 已有工具/方法论可以解决该问题,无需创建新内容。
640
- 可用的工具/方法论:[列出工具名称或方法论名称]
641
- 使用方法:[简要说明如何使用]
642
- 2. 工具创建(如果需要创建新工具):
643
- {ot("TOOL_CALL")}
644
- want: 创建新工具来解决XXX问题
645
- name: generate_new_tool
646
- arguments:
647
- tool_name: 工具名称
648
- tool_code: |2
649
- # -*- coding: utf-8 -*-
650
- from typing import Dict, Any
651
- from jarvis.jarvis_utils.output import PrettyOutput, OutputType
652
- class 工具名称:
653
- name = "工具名称"
654
- description = "Tool for text transformation"
655
- Tool description
656
- 适用场景:1. 格式化文本; 2. 处理标题; 3. 标准化输出
657
- \"\"\"
658
- parameters = {{
659
- "type": "object",
660
- "properties": {{
661
- # 参数定义
662
- }},
663
- "required": []
664
- }}
665
- @staticmethod
666
- def check() -> bool:
667
- return True
668
- def execute(self, args: Dict[str, Any]) -> Dict[str, Any]:
669
- try:
670
- # 使用PrettyOutput显示执行过程
671
- PrettyOutput.print("开始执行操作...", OutputType.INFO)
672
- # 实现逻辑
673
- # ...
674
- PrettyOutput.print("操作已完成", OutputType.SUCCESS)
675
- return {{
676
- "success": True,
677
- "stdout": "结果输出",
678
- "stderr": ""
679
- }}
680
- except Exception as e:
681
- PrettyOutput.print(f"操作失败: {{str(e)}}", OutputType.ERROR)
682
- return {{
683
- "success": False,
684
- "stdout": "",
685
- "stderr": f"操作失败: {{str(e)}}"
686
- }}
687
- {ct("TOOL_CALL")}
688
- 3. 方法论创建(如果需要创建新方法论):
689
- {ot("TOOL_CALL")}
690
- want: 添加/更新xxxx的方法论
691
- name: methodology
692
- arguments:
693
- operation: add/update
694
- problem_type: 方法论类型,不要过于细节,也不要过于泛化
695
- content: |2
696
- 方法论内容
697
- {ct("TOOL_CALL")}
698
- 如果以上三种情况都不适用,则直接输出原因分析,不要使用工具调用格式。
699
- </output_requirements>
700
- </task_analysis>"""
701
-
702
- self.prompt = analysis_prompt
703
- response = self.model.chat_until_success(self.prompt) # type: ignore
408
+ analysis_prompt = TASK_ANALYSIS_PROMPT
409
+
410
+ self.session.prompt = analysis_prompt
411
+ if not self.model:
412
+ raise RuntimeError("Model not initialized")
413
+ response = self.model.chat_until_success(self.session.prompt) # type: ignore
704
414
  self._call_tools(response)
705
415
  print("✅ 分析完成")
706
416
  except Exception as e:
@@ -756,7 +466,7 @@ arguments:
756
466
  4. 自动加载相关方法论(如果是首次运行)
757
467
  """
758
468
 
759
- self.prompt = f"{user_input}"
469
+ self.session.prompt = f"{user_input}"
760
470
  try:
761
471
  set_agent(self.name, self)
762
472
 
@@ -764,33 +474,41 @@ arguments:
764
474
  if self.first:
765
475
  self._first_run()
766
476
  try:
767
- current_response = self._call_model(self.prompt, True)
768
- self.prompt = ""
477
+ current_response = self._call_model(self.session.prompt, True)
478
+ self.session.prompt = ""
769
479
 
770
480
  if get_interrupt():
771
481
  set_interrupt(False)
772
- user_input = self.multiline_inputer(f"模型交互期间被中断,请输入用户干预信息:")
482
+ user_input = self.multiline_inputer(
483
+ f"模型交互期间被中断,请输入用户干预信息:"
484
+ )
773
485
  if user_input:
774
486
  # 如果有工具调用且用户确认继续,则将干预信息和工具执行结果拼接为prompt
775
487
  if any(
776
488
  handler.can_handle(current_response)
777
489
  for handler in self.output_handler
778
490
  ):
779
- if user_confirm("检测到有工具调用,是否继续处理工具调用?", True):
780
- self.prompt = f"{user_input}\n\n{current_response}"
491
+ if user_confirm(
492
+ "检测到有工具调用,是否继续处理工具调用?", True
493
+ ):
494
+ self.session.prompt = (
495
+ f"{user_input}\n\n{current_response}"
496
+ )
781
497
  continue
782
- self.prompt += f"{user_input}"
498
+ self.session.prompt += f"{user_input}"
783
499
  continue
784
500
 
785
- need_return, self.prompt = self._call_tools(current_response)
501
+ need_return, self.session.prompt = self._call_tools(
502
+ current_response
503
+ )
786
504
 
787
505
  if need_return:
788
- return self.prompt
506
+ return self.session.prompt
789
507
 
790
508
  if self.after_tool_call_cb:
791
509
  self.after_tool_call_cb(self)
792
510
 
793
- if self.prompt or self.addon_prompt:
511
+ if self.session.prompt or self.session.addon_prompt:
794
512
  continue
795
513
 
796
514
  if self.auto_complete and ot("!!!COMPLETE!!!") in current_response:
@@ -802,7 +520,7 @@ arguments:
802
520
  )
803
521
 
804
522
  if user_input:
805
- self.prompt = user_input
523
+ self.session.prompt = user_input
806
524
  continue
807
525
 
808
526
  if not user_input:
@@ -822,46 +540,39 @@ arguments:
822
540
  if self.use_methodology:
823
541
  if not upload_methodology(self.model, other_files=self.files):
824
542
  if self.files:
825
- PrettyOutput.print("文件上传失败,将忽略文件列表", OutputType.WARNING)
543
+ PrettyOutput.print(
544
+ "文件上传失败,将忽略文件列表", OutputType.WARNING
545
+ )
826
546
  # 上传失败则回退到本地加载
827
- msg = self.prompt
547
+ msg = self.session.prompt
828
548
  for handler in self.input_handler:
829
549
  msg, _ = handler(msg, self)
830
- self.prompt = f"{self.prompt}\n\n以下是历史类似问题的执行经验,可参考:\n{load_methodology(msg, self.get_tool_registry())}"
550
+ self.session.prompt = f"{self.session.prompt}\n\n以下是历史类似问题的执行经验,可参考:\n{load_methodology(msg, self.get_tool_registry())}"
831
551
  else:
832
552
  if self.files:
833
- self.prompt = (
834
- f"{self.prompt}\n\n上传的文件包含历史对话信息和方法论文件,可以从中获取一些经验信息。"
835
- )
553
+ self.session.prompt = f"{self.session.prompt}\n\n上传的文件包含历史对话信息和方法论文件,可以从中获取一些经验信息。"
836
554
  else:
837
- self.prompt = f"{self.prompt}\n\n上传的文件包含历史对话信息,可以从中获取一些经验信息。"
555
+ self.session.prompt = f"{self.session.prompt}\n\n上传的文件包含历史对话信息,可以从中获取一些经验信息。"
838
556
  elif self.files:
839
557
  if not self.model.upload_files(self.files):
840
- PrettyOutput.print("文件上传失败,将忽略文件列表", OutputType.WARNING)
558
+ PrettyOutput.print(
559
+ "文件上传失败,将忽略文件列表", OutputType.WARNING
560
+ )
841
561
  else:
842
- self.prompt = f"{self.prompt}\n\n上传的文件包含历史对话信息,可以从中获取一些经验信息。"
562
+ self.session.prompt = f"{self.session.prompt}\n\n上传的文件包含历史对话信息,可以从中获取一些经验信息。"
843
563
  else:
844
564
  if self.files:
845
565
  PrettyOutput.print("不支持上传文件,将忽略文件列表", OutputType.WARNING)
846
566
  if self.use_methodology:
847
- msg = self.prompt
567
+ msg = self.session.prompt
848
568
  for handler in self.input_handler:
849
569
  msg, _ = handler(msg, self)
850
- self.prompt = f"{self.prompt}\n\n以下是历史类似问题的执行经验,可参考:\n{load_methodology(msg, self.get_tool_registry())}"
570
+ self.session.prompt = f"{self.session.prompt}\n\n以下是历史类似问题的执行经验,可参考:\n{load_methodology(msg, self.get_tool_registry())}"
851
571
 
852
572
  self.first = False
853
573
 
854
574
  def clear_history(self):
855
- """清空对话历史但保留系统提示
856
-
857
- 该方法将:
858
- 1. 清空当前提示
859
- 2. 重置模型状态
860
- 3. 重置对话长度计数器
861
-
862
- 注意:
863
- 用于重置Agent状态而不影响系统消息
864
575
  """
865
- self.prompt = ""
866
- self.model.reset() # type: ignore
867
- self.conversation_length = 0 # 重置对话长度
576
+ Clears conversation history by delegating to the session manager.
577
+ """
578
+ self.session.clear_history()