Merge pull request #83 from streetycat/main

fix bugs
This commit is contained in:
Liu Zhicong
2023-10-16 19:54:23 -07:00
committed by GitHub
2 changed files with 5 additions and 3 deletions
+2 -1
View File
@@ -544,8 +544,9 @@ class Workflow:
async def _get_prompt_from_session(self,the_role:AIRole,chatsession:AIChatSession) -> AgentPrompt: async def _get_prompt_from_session(self,the_role:AIRole,chatsession:AIChatSession) -> AgentPrompt:
messages = chatsession.read_history(the_role.history_len) # read last 10 message messages = chatsession.read_history(the_role.history_len) # read last 10 message
result_prompt = AgentPrompt() result_prompt = AgentPrompt()
for msg in reversed(messages): for msg in reversed(messages):
if msg.sender == chatsession.owner_id: if msg.sender == the_role.role_id:
result_prompt.messages.append({"role":"assistant","content":msg.body}) result_prompt.messages.append({"role":"assistant","content":msg.body})
else: else:
result_prompt.messages.append({"role":"user","content":f"{msg.body}"}) result_prompt.messages.append({"role":"user","content":f"{msg.body}"})
+3 -2
View File
@@ -414,8 +414,8 @@ class AIOS_Shell:
if len(args) < 4: if len(args) < 4:
return show_text return show_text
model_name = args[3] model_name = args[2]
url = args[4] url = args[3]
ComputeNodeConfig.get_instance().remove_node("llama", url, model_name) ComputeNodeConfig.get_instance().remove_node("llama", url, model_name)
ComputeNodeConfig.get_instance().save() ComputeNodeConfig.get_instance().save()
else: else:
@@ -747,6 +747,7 @@ async def main():
'/disable $feature', '/disable $feature',
'/node add llama $model_name $url', '/node add llama $model_name $url',
'/node rm llama $model_name $url', '/node rm llama $model_name $url',
'/node list',
'/show', '/show',
'/exit', '/exit',
'/help'], ignore_case=True) '/help'], ignore_case=True)