snsd0805/NavGPT_explore_module
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: steps counter in llm-success

Browse Source
This commit is contained in:
Ting-Jun Wang 2024-05-06 16:41:07 +08:00
parent 68330c5163
commit 64fbce018a
Signed by: snsd0805
GPG Key ID: 48D331A3D6160354
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
nav_src/agent.py
View File

@ -100,6 +100,7 @@ class NavGPTOutputParser(AgentOutputParser):
print(f"SUCCESS = {SUCCESS}") print(f"SUCCESS = {SUCCESS}")
NOW_LOCATION = tool_input NOW_LOCATION = tool_input
TEMP_STEPS_COUNTER += 1
print(f"NOW_LOCATION = {NOW_LOCATION}") print(f"NOW_LOCATION = {NOW_LOCATION}")
@ -120,6 +121,8 @@ class NavGPTOutputParser(AgentOutputParser):
return AgentAction(action, tool_input, text) return AgentAction(action, tool_input, text)
elif includes_answer: elif includes_answer:
if NOW_LOCATION == FINAL_STOP_POINT: if NOW_LOCATION == FINAL_STOP_POINT:
STEPS_COUNTER += TEMP_STEPS_COUNTER
TEMP_STEPS_COUNTER = 0
SUCCESS += 1 SUCCESS += 1
print(f"SUCCESS = {SUCCESS}") print(f"SUCCESS = {SUCCESS}")
else: else:
@ -128,6 +131,7 @@ class NavGPTOutputParser(AgentOutputParser):
print(f"{NOW_LOCATION}_{type(NOW_LOCATION)}") print(f"{NOW_LOCATION}_{type(NOW_LOCATION)}")
print(f"{FINAL_STOP_POINT}_{type(FINAL_STOP_POINT)}") print(f"{FINAL_STOP_POINT}_{type(FINAL_STOP_POINT)}")
print(f"SUCCESS = {SUCCESS}") print(f"SUCCESS = {SUCCESS}")
print(f"STEPS_COUNTER = {STEPS_COUNTER}")
return AgentFinish( return AgentFinish(
{"output": text.split(FINAL_ANSWER_ACTION)[-1].strip()}, text {"output": text.split(FINAL_ANSWER_ACTION)[-1].strip()}, text
) )
@ -712,6 +716,7 @@ class NavGPTAgent(BaseAgent):
global FINAL_STOP_POINT global FINAL_STOP_POINT
global TEMP_STEPS_COUNTER global TEMP_STEPS_COUNTER
global STEPS_COUNTER
global NOW_LOCATION global NOW_LOCATION
FINAL_STOP_POINT = obs[0]['stop'] FINAL_STOP_POINT = obs[0]['stop']