alex/LocalAGI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Avoid repetitions

Browse Source
This commit is contained in:
mudler
2023-08-05 00:14:49 +02:00
parent 8d35592024
commit cef94cb52d
1 changed files with 16 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
main.py
View File

@@ -418,9 +418,10 @@ def search_memory(query, agent_actions={}):
for doc in docs: for doc in docs:
text_res+="- "+doc.page_content+"\n" text_res+="- "+doc.page_content+"\n"
if args.postprocess: #if args.postprocess:
return post_process(text_res) # return post_process(text_res)
return text_res #return text_res
return post_process(text_res)
def generate_plan(user_input, agent_actions={}): def generate_plan(user_input, agent_actions={}):
res = json.loads(user_input) res = json.loads(user_input)
@@ -437,10 +438,8 @@ def generate_plan(user_input, agent_actions={}):
"content": f"""Transcript of AI assistant responding to user requests. "content": f"""Transcript of AI assistant responding to user requests.
{descriptions} {descriptions}
Request: {res["description"]} Request: {plan_message}
Thought: {res["description"]}
The assistant replies with a plan of 3 steps to answer the request with a list of subtasks with logical steps. The reasoning includes a self-contained, detailed and descriptive instruction to fullfill the task.
Function call: """ Function call: """
} }
] ]
@@ -640,28 +639,28 @@ def evaluate(user_input, conversation_history = [],re_evaluate=False, agent_acti
subtask_result="" subtask_result=""
for subtask in function_results["subtasks"]: for subtask in function_results["subtasks"]:
#ctr="Context: "+user_input+"\nThought: "+action["reasoning"]+ "\nRequest: "+subtask["reasoning"] #ctr="Context: "+user_input+"\nThought: "+action["reasoning"]+ "\nRequest: "+subtask["reasoning"]
#cr="Context: "+user_input+"\n" #cr="Request: "+user_input+"\n"
cr="Reasoning: "+action["reasoning"]+ "\n" cr=""
#cr=""
if subtask_result != "" and subtaskContext: if subtask_result != "" and subtaskContext:
# Include cumulative results of previous subtasks # Include cumulative results of previous subtasks
# TODO: this grows context, maybe we should use a different approach or summarize # TODO: this grows context, maybe we should use a different approach or summarize
##if postprocess: ##if postprocess:
## cr+= "Subtask results: "+post_process(subtask_result)+"\n" ## cr+= "Subtask results: "+post_process(subtask_result)+"\n"
##else: ##else:
cr+="Subtask results: "+subtask_result+"\n" cr+="\n"+subtask_result+"\n"
subtask_reasoning = subtask["reasoning"] subtask_reasoning = subtask["reasoning"]
cr+="Reasoning: "+action["reasoning"]+ "\n"
cr+="\nFunction to call:" +subtask["function"]+"\n"
logger.info("==> subtask '{subtask}' ({reasoning})", subtask=subtask["function"], reasoning=subtask_reasoning) logger.info("==> subtask '{subtask}' ({reasoning})", subtask=subtask["function"], reasoning=subtask_reasoning)
if postprocess: if postprocess:
cr+= "Request: "+post_process(subtask_reasoning) cr+= "Assistant: "+post_process(subtask_reasoning)
else: else:
cr+= "Request: "+subtask_reasoning cr+= "Assistant: "+subtask_reasoning
subtask_response, function_results = process_functions(cr, subtask["function"],agent_actions=agent_actions) subtask_response, function_results = process_functions(cr, subtask["function"],agent_actions=agent_actions)
subtask_result+=process_history(subtask_response[1:]) subtask_result+=str(function_results)+"\n"
# if postprocess: # if postprocess:
# subtask_result=post_process(subtask_result) # subtask_result=post_process(subtask_result)
responses.extend(subtask_response) responses.append(subtask_response[-1])
if re_evaluate: if re_evaluate:
## Better output or this infinite loops.. ## Better output or this infinite loops..
logger.info("-> Re-evaluate if another action is needed") logger.info("-> Re-evaluate if another action is needed")
@@ -681,7 +680,7 @@ def evaluate(user_input, conversation_history = [],re_evaluate=False, agent_acti
return conversation_history return conversation_history
# TODO: this needs to be optimized # TODO: this needs to be optimized
responses = analyze(responses, prefix=f"Return an appropriate answer to the user input '{user_input}' given the context below and summarizing the actions taken\n") responses = analyze(responses, prefix=f"You are an AI assistant. Return an appropriate answer to the user input '{user_input}' given the context below and summarizing the actions taken\n")
# add responses to conversation history by extending the list # add responses to conversation history by extending the list
conversation_history.append( conversation_history.append(