Update agent.py

agent.py

@@ -315,59 +315,14 @@ class EsiAgent:
             openai_api_key=os.environ["GPT_4o_API_KEY"],
             model_name="meta-llama/llama-4-scout-17b-16e-instruct",
             temperature=0,
-            max_tokens=128
-            # max_tokens=256
+            # max_tokens=128
+            max_tokens=256
         )
 
         llm_with_tools = llm.bind_tools(tools)
         # 3) Single system message
         self.sys_msg = SystemMessage(content=PLANNER_PROMPT)
 
-        # # 4) Assistant node
-        # def assistant(state: MessagesState) -> dict:
-        #   """
-        #   Assistant node: Validates if a final answer exists or generates the plan.
-        #   If a final answer is found, it returns it, otherwise, it generates a plan and iterates.
-        #   """
-
-        #   resutls_dict = {"tool_names":[],"tool_args":[],"tool_results":[]}
-        #   # Check if the final answer exists already in previous steps
-        #   for msg in state["messages"]:
-        #     if isinstance(msg, ToolMessage):
-        #       resutls_dict['tool_results'].append(msg.content)
-        #     elif isinstance(msg, AIMessage) and hasattr(msg, "tool_calls"):
-        #       for call in msg.tool_calls:
-        #         resutls_dict['tool_names'].append(call['name'])
-        #         resutls_dict['tool_args'].append(call['args'])
-
-
-        #   llm_input = [SystemMessage(content=PLANNER_PROMPT)]+state["messages"]
-
-        #   ai_resp = llm_with_tools.invoke(llm_input)
-        #   content = ai_resp.content.strip()
-
-        #   # Ensure that the FINAL ANSWER is correctly returned
-        #   s1=len(list(set(resutls_dict['tool_names'])))
-        #   s2=len(list(set(resutls_dict['tool_names'])))
-        #   s3=len(list(set(resutls_dict['tool_names'])))
-
-        #   if s1==s2==s3:
-        #     state['assist_iter'] += 1
-
-        #   if "FINAL ANSWER:" in content and (state['assist_iter']>=2 or state['assist_iter']<=5):
-        #       final_answer = content
-        #       is_done = True
-        #   else:
-        #       final_answer = None
-        #       is_done = False
-
-        #   return {
-        #       "messages": state["messages"] + [ai_resp],
-        #       "input_file": state.get("input_file", None),
-        #       "final_answer": final_answer,
-        #       'assist_iter': state['assist_iter'] + 1,
-        #       "is_done": is_done,
-        #   }
         # 4) Assistant node
         def assistant(state: MessagesState) -> dict:
           """