ursa-ai 0.2.12__py3-none-any.whl → 0.2.13__py3-none-any.whl

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/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.dist-info → ursa_ai-0.2.13.dist-info}/METADATA

@@ -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.dist-info → ursa_ai-0.2.13.dist-info}/RECORD

@@ -2,10 +2,10 @@ ursa/agents/__init__.py,sha256=IdwVBLlz7QydHiFSok06McIfEJed6rxrT01hZgpQig0,1008
 ursa/agents/arxiv_agent.py,sha256=6pycpm1hkUNjwXF6lJ4DCF5CxOiSnS5dvpW2Qes3ZD8,14125
 ursa/agents/base.py,sha256=uFhRLVzqhFbTZVA7IePKbUi03ATCXuvga7rzwaHy1B0,1321
 ursa/agents/code_review_agent.py,sha256=aUDq5gT-jdl9Qs-Wewj2oz1d60xov9sN-DOYRfGNTU0,11550
-ursa/agents/execution_agent.py,sha256=-At1EcKRHP9lYQ80jpqTPtQyPuQV-sIpf9J8LlEfWdA,16618
+ursa/agents/execution_agent.py,sha256=nvKjGEj3tcYs26YzEUyZTubxzzTe_XGfzyuab-livZM,16583
 ursa/agents/hypothesizer_agent.py,sha256=KrOoyM5znZVAIpcflGLO1sybMbyx7S7HERAXvct5zig,23295
 ursa/agents/mp_agent.py,sha256=UyJSheMGHZpWQJL3EgYgPPqArfv6F8sndN05q4CPtyo,6015
-ursa/agents/planning_agent.py,sha256=AKWQJ848RLPiwQGrvDNdN9lBlf3YI5qWmt2hqXnRGj8,5426
+ursa/agents/planning_agent.py,sha256=ayyNDQifPvYtQ-JYnFk3TaXWZcd_6k8qUheJGariqG8,5574
 ursa/agents/recall_agent.py,sha256=bQk7ZJtiO5pj89A50OBDzAJ4G2F7ZdsMwmKnp1WWR7g,813
 ursa/agents/websearch_agent.py,sha256=rCv4AWbqe5Us4FmuypM6jptri21nKoNg044ncsu9u3E,8014
 ursa/prompt_library/code_review_prompts.py,sha256=-HuhwW9W_p2LDn44bXLntxLADHCOyl-2KIXxRHto66w,2444
@@ -19,8 +19,8 @@ ursa/tools/write_code.py,sha256=DtCsUMZegYm0mk-HMPG5Zo3Ba1gbGfnXHsv1NZTdDs8,1220
 ursa/util/diff_renderer.py,sha256=1L1q2qWWb8gLhR532-LgJn2TrqXDx0gUpPVOWD_sqeU,4086
 ursa/util/memory_logger.py,sha256=9tNPF3Fgth4ImpNbs7Qg5Ci6Hb1ziMWsDTsqjkDi1iY,6096
 ursa/util/parse.py,sha256=M0cjyQWmjatxX4WbVmDRUiirTLyW-t_Aemlrlrsc5nA,2811
-ursa_ai-0.2.12.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
-ursa_ai-0.2.12.dist-info/METADATA,sha256=Yx4oeE3N7uIaQemsgzHldvOC818Hmo0UNyTnxcHchcQ,6833
-ursa_ai-0.2.12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ursa_ai-0.2.12.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
-ursa_ai-0.2.12.dist-info/RECORD,,
+ursa_ai-0.2.13.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
+ursa_ai-0.2.13.dist-info/METADATA,sha256=3F0m31Cwq1SYV6GNxkBgXcclvV359UtH-h-GoNwQ6h0,6833
+ursa_ai-0.2.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ursa_ai-0.2.13.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
+ursa_ai-0.2.13.dist-info/RECORD,,