gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge remote-tracking branch 'origin/main' into MVP

Browse Source 
# Conflicts:
#	src/service/aios_shell/aios_shell.py
This commit is contained in:
Liu Zhicong
2023-11-08 22:17:50 -08:00
parent de685da38b e27c995afe
commit fb0b88d44a
5 changed files with 45 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/service/aios_shell/aios_shell.py
+10 -6
View File
@@ -356,10 +356,13 @@ class AIOS_Shell:
pipelines = KnowledgePipelineManager.get_instance().get_pipelines()
print_formatted_text("\r\n".join(pipeline.get_name() for pipeline in pipelines))
if sub_cmd == "journal":
name = args[1]
topn = 10 if len(args) == 2 else int(args[2])
journals = [str(journal) for journal in KnowledgePipelineManager.get_instance().get_pipeline(name).get_journal().latest_journals(topn)]
print_formatted_text("\r\n".join(str(journal) for journal in journals))
try:
name = args[1]
topn = 10 if len(args) == 2 else int(args[2])
journals = [str(journal) for journal in KnowledgePipelineManager.get_instance().get_pipeline(name).get_journal().latest_journals(topn)]
print_formatted_text("\r\n".join(str(journal) for journal in journals))
except ValueError:
return FormattedText([("class:title", f"/knowledge journal failed: {args[1]} is not a valid integer.\n")])
if sub_cmd == "query":
if len(args) < 2:
return show_text
@@ -404,8 +407,8 @@ class AIOS_Shell:
if len(args) < 4:
return show_text
model_name = args[3]
url = args[4]
model_name = args[2]
url = args[3]
ComputeNodeConfig.get_instance().remove_node("llama", url, model_name)
ComputeNodeConfig.get_instance().save()
else:
@@ -760,6 +763,7 @@ async def main():
'/disable $feature',
'/node add llama $model_name $url',
'/node rm llama $model_name $url',
'/node list',
'/show',
'/exit',
'/help'], ignore_case=True)