chore: remove verbose prints and add agent/memory tool registration in assistant core
This commit is contained in:
@@ -47,6 +47,8 @@ class AdvancedContextManager:
|
||||
return complexity
|
||||
|
||||
def extract_key_sentences(self, text: str, top_k: int = 5) -> List[str]:
|
||||
if not text.strip():
|
||||
return []
|
||||
sentences = re.split(r'(?<=[.!?])\s+', text)
|
||||
if not sentences:
|
||||
return []
|
||||
|
||||
+18
-1
@@ -133,6 +133,18 @@ class Assistant:
|
||||
'display_edit_summary': lambda **kw: display_edit_summary(),
|
||||
'display_edit_timeline': lambda **kw: display_edit_timeline(**kw),
|
||||
'clear_edit_tracker': lambda **kw: clear_edit_tracker(),
|
||||
'create_agent': lambda **kw: create_agent(**kw),
|
||||
'list_agents': lambda **kw: list_agents(**kw),
|
||||
'execute_agent_task': lambda **kw: execute_agent_task(**kw),
|
||||
'remove_agent': lambda **kw: remove_agent(**kw),
|
||||
'collaborate_agents': lambda **kw: collaborate_agents(**kw),
|
||||
'add_knowledge_entry': lambda **kw: add_knowledge_entry(**kw),
|
||||
'get_knowledge_entry': lambda **kw: get_knowledge_entry(**kw),
|
||||
'search_knowledge': lambda **kw: search_knowledge(**kw),
|
||||
'get_knowledge_by_category': lambda **kw: get_knowledge_by_category(**kw),
|
||||
'update_knowledge_importance': lambda **kw: update_knowledge_importance(**kw),
|
||||
'delete_knowledge_entry': lambda **kw: delete_knowledge_entry(**kw),
|
||||
'get_knowledge_statistics': lambda **kw: get_knowledge_statistics(**kw),
|
||||
}
|
||||
|
||||
if func_name in func_map:
|
||||
@@ -230,6 +242,7 @@ class Assistant:
|
||||
path_options = [p + os.sep if os.path.isdir(p) else p for p in path_options]
|
||||
|
||||
combined_options = sorted(list(set(options + path_options)))
|
||||
#combined_options.extend(self.commands)
|
||||
|
||||
if state < len(combined_options):
|
||||
return combined_options[state]
|
||||
@@ -279,7 +292,8 @@ class Assistant:
|
||||
else:
|
||||
message = sys.stdin.read()
|
||||
|
||||
process_message(self, message)
|
||||
from pr.autonomous.mode import run_autonomous_mode
|
||||
run_autonomous_mode(self, message)
|
||||
|
||||
def cleanup(self):
|
||||
if hasattr(self, 'enhanced') and self.enhanced:
|
||||
@@ -299,9 +313,12 @@ class Assistant:
|
||||
|
||||
def run(self):
|
||||
try:
|
||||
print(f"DEBUG: interactive={self.args.interactive}, message={self.args.message}, isatty={sys.stdin.isatty()}")
|
||||
if self.args.interactive or (not self.args.message and sys.stdin.isatty()):
|
||||
print("DEBUG: calling run_repl")
|
||||
self.run_repl()
|
||||
else:
|
||||
print("DEBUG: calling run_single")
|
||||
self.run_single()
|
||||
finally:
|
||||
self.cleanup()
|
||||
|
||||
Reference in New Issue
Block a user