ursa-ai 0.2.12__tar.gz → 0.2.13__tar.gz

{ursa_ai-0.2.12/src/ursa_ai.egg-info → ursa_ai-0.2.13}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.2.12
+Version: 0.2.13
 Summary: Agents for science at LANL
 Author-email: Mike Grosskopf <mikegros@lanl.gov>, Nathan Debardeleben <ndebard@lanl.gov>, Rahul Somasundaram <rsomasundaram@lanl.gov>, Isaac Michaud <imichaud@lanl.gov>, Avanish Mishra <avanish@lanl.gov>, Arthur Lui <alui@lanl.gov>, Russell Bent <rbent@lanl.gov>, Earl Lawrence <earl@lanl.gov>
 License-Expression: BSD-3-Clause

{ursa_ai-0.2.12 → ursa_ai-0.2.13}/src/ursa/agents/execution_agent.py

@@ -82,10 +82,8 @@ class ExecutionAgent(BaseAgent):
         os.makedirs(new_state["workspace"], exist_ok=True)
 
         # code related to symlink
-        if (
-            "symlinkdir" in new_state.keys()
-            and "is_linked" not in new_state["symlinkdir"].keys()
-        ):
+        sd = new_state.get("symlinkdir")
+        if isinstance(sd, dict) and "is_linked" not in sd:
             # symlinkdir = {"source": "foo", "dest": "bar"}
             symlinkdir = new_state["symlinkdir"]
             # user provided a symlinkdir key - let's do the linking!

{ursa_ai-0.2.12 → ursa_ai-0.2.13}/src/ursa/agents/planning_agent.py

@@ -39,6 +39,7 @@ class PlanningAgent(BaseAgent):
         self._initialize_agent()
 
     def generation_node(self, state: PlanningState) -> PlanningState:
+        print("PlanningAgent: generating . . .")
         messages = state["messages"]
         if isinstance(messages[0], SystemMessage):
             messages[0] = SystemMessage(content=self.planner_prompt)
@@ -53,6 +54,7 @@ class PlanningAgent(BaseAgent):
         }
 
     def formalize_node(self, state: PlanningState) -> PlanningState:
+        print("PlanningAgent: formalizing . . .")
         cls_map = {"ai": HumanMessage, "human": AIMessage}
         translated = [state["messages"][0]] + [
             cls_map[msg.type](content=msg.content)
@@ -78,6 +80,7 @@ class PlanningAgent(BaseAgent):
         }
 
     def reflection_node(self, state: PlanningState) -> PlanningState:
+        print("PlanningAgent: reflecting . . .")
         cls_map = {"ai": HumanMessage, "human": AIMessage}
         translated = [state["messages"][0]] + [
             cls_map[msg.type](content=msg.content)

{ursa_ai-0.2.12 → ursa_ai-0.2.13/src/ursa_ai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.2.12
+Version: 0.2.13
 Summary: Agents for science at LANL
 Author-email: Mike Grosskopf <mikegros@lanl.gov>, Nathan Debardeleben <ndebard@lanl.gov>, Rahul Somasundaram <rsomasundaram@lanl.gov>, Isaac Michaud <imichaud@lanl.gov>, Avanish Mishra <avanish@lanl.gov>, Arthur Lui <alui@lanl.gov>, Russell Bent <rbent@lanl.gov>, Earl Lawrence <earl@lanl.gov>
 License-Expression: BSD-3-Clause