exact goal loading in the tool-use-agent

Author: amanjaiswal73892

src/agentlab/agents/tool_use_agent/tool_use_agent.py

@@ -325,12 +325,17 @@ def _init(self):
                 embedder_server=self.embedder_server,
             )
 
-    def apply(self, llm, discussion: StructuredDiscussion, task_name: str) -> dict:
+    def apply(self, llm, discussion: StructuredDiscussion, obs: dict, task_name: str) -> dict:
         if not self.use_task_hint:
             return {}
 
-        goal = "\n".join([c.get("text", "") for c in discussion.groups[0].messages[1].content])
-        task_hints = self.hints_source.choose_hints(llm, task_name, goal)
+        # goal = "\n".join([c.get("text", "") for c in discussion.groups[0].messages[1].content])
+        try:
+            goal_text = obs["goal_object"][0]["text"]
+        except (KeyError, IndexError):
+            Warning("Goal text not found in observation")
+            goal_text = ""
+        task_hints = self.hints_source.choose_hints(llm, task_name, goal_text)
 
         hints = []
         for hint in task_hints:
@@ -472,7 +477,7 @@ def get_action(self, obs: Any) -> float:
 
             self.config.summarizer.apply_init(self.llm, self.discussion)
             self.config.general_hints.apply(self.llm, self.discussion)
-            self.task_hint.apply(self.llm, self.discussion, self.task_name)
+            self.task_hint.apply(self.llm, self.discussion, obs=obs, task_name=self.task_name)
 
             self.discussion.new_group()
 

src/agentlab/utils/hinting.py

@@ -52,7 +52,6 @@ def __init__(
             self.hint_db_path,
             header=0,
             index_col=None,
-            dtype=str,
             converters={
                 "trace_paths_json": lambda x: json.loads(x) if pd.notna(x) else [],
                 "source_trace_goals": lambda x: json.loads(x) if pd.notna(x) else [],