ursa-ai 0.6.0rc1__py3-none-any.whl → 0.6.1__py3-none-any.whl

ursa/agents/code_review_agent.py

@@ -270,7 +270,9 @@ def read_file(filename: str, state: Annotated[dict, InjectedState]):
 
 
 @tool
-def write_file(code: str, filename: str, state: Annotated[dict, InjectedState]):
+def write_file(
+    code: str, filename: str, state: Annotated[dict, InjectedState]
+) -> str:
     """
     Writes text to a file in the given workspace as requested.
 

ursa/agents/optimization_agent.py

@@ -280,7 +280,9 @@ def run_cmd(query: str, state: Annotated[dict, InjectedState]) -> str:
 
 
 @tool
-def write_code(code: str, filename: str, state: Annotated[dict, InjectedState]):
+def write_code(
+    code: str, filename: str, state: Annotated[dict, InjectedState]
+) -> str:
     """
     Writes python or Julia code to a file in the given workspace as requested.
 

ursa/tools/write_code.py

@@ -4,7 +4,7 @@ from langchain_core.tools import tool
 
 
 @tool
-def write_python(code: str, filename: str, workspace_dir: str):
+def write_python(code: str, filename: str, workspace_dir: str) -> str:
     """
     Writes code to a file in the given workspace.
 

ursa/util/parse.py

@@ -2,7 +2,7 @@ import json
 import re
 
 
-def extract_json(text: str):
+def extract_json(text: str) -> list[dict]:
     """
     Extract a JSON object or array from text that might contain markdown or other content.
 

{ursa_ai-0.6.0rc1.dist-info → ursa_ai-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.6.0rc1
+Version: 0.6.1
 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.6.0rc1.dist-info → ursa_ai-0.6.1.dist-info}/RECORD

@@ -1,12 +1,13 @@
+ursa/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ursa/agents/__init__.py,sha256=u5ClncJ-w4nIJLQJTOC-NBv6Hu4pXxALgZJJxDt3tZw,1202
 ursa/agents/arxiv_agent.py,sha256=naXZuPd3tHuyjqoldsN1gJXWcGB6GJaEdtEzfP1JO1w,14570
 ursa/agents/base.py,sha256=XZ62XhF-GdMykdUyiMuxqQwt41hrryaK9aBhgIFJMMQ,14482
-ursa/agents/code_review_agent.py,sha256=aUDq5gT-jdl9Qs-Wewj2oz1d60xov9sN-DOYRfGNTU0,11550
+ursa/agents/code_review_agent.py,sha256=F3RJmS1jBYOoWPg5Hiy-xirutLTtw_u1_hqiKIsA2eY,11563
 ursa/agents/execution_agent.py,sha256=gMQsiNoj8feG3kalcVs39AV58GXrfvq6jff3m2ecyyw,17923
 ursa/agents/hypothesizer_agent.py,sha256=2jSOHkR8MjNhS5HMeOhZCkhT4wGDLNBs-aLgQhYBHr4,23124
 ursa/agents/lammps_agent.py,sha256=z7RbZwk70aC31CyagXk3LyPhtU4IInUMMa1v8r2qji4,14774
 ursa/agents/mp_agent.py,sha256=HvD4JC9OuGeEicW3mKZBQmQ1yn0HVB03118E1urXwx4,6833
-ursa/agents/optimization_agent.py,sha256=E2Egh_xi1vwRaAAn7qZJsBRCWgUN6cH2HFJYX6LHOJA,13932
+ursa/agents/optimization_agent.py,sha256=siJKyujfy_oNQDbnIFaCUkE_DLvmxPYsBJBDms6Cgbg,13945
 ursa/agents/planning_agent.py,sha256=9XwwmcEtNHW8xd4osljawp1V0X4y9t3R6ncHrsPo1eo,6489
 ursa/agents/rag_agent.py,sha256=kc7ZYWxgWNh-LbMpREaJY90IVbs2SKPrXdCutxrEdT0,10099
 ursa/agents/recall_agent.py,sha256=MoS0FaQFAb14DeYSCo6sYKNTjvE_u-egTOvuXSTXLpk,1693
@@ -15,6 +16,7 @@ ursa/cli/__init__.py,sha256=YYh7E1Ce3fcJaYn1XP19U1d9iHG1INfnMQ6gAhtARKM,3879
 ursa/cli/hitl.py,sha256=NIKr72fIdwXE_g6KtRc6c94zSRvft2Ptyi4F0UPSBI8,14695
 ursa/observability/pricing.py,sha256=wNe51EcYjk3KDSuU8rv_vjOPlGZB3We_dQrVBcPXdWA,10551
 ursa/observability/timing.py,sha256=yQX5ZtkxRBH_1V61KjWmEBph02DOeZuP-v0-OtJNffQ,48612
+ursa/prompt_library/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ursa/prompt_library/code_review_prompts.py,sha256=-HuhwW9W_p2LDn44bXLntxLADHCOyl-2KIXxRHto66w,2444
 ursa/prompt_library/execution_prompts.py,sha256=bJpRPBW-6d41amqC5DXqcIwsoOPpMvY7ZMNGVsJWgII,2388
 ursa/prompt_library/hypothesizer_prompts.py,sha256=ieupOF5tUy_u8actOjPbK-y5Qkrgw6EYxAfw6RXBebs,762
@@ -22,18 +24,20 @@ ursa/prompt_library/literature_prompts.py,sha256=zhBiN3Q-1Z2hp-hkXXp0T8Ipc-6YUM9
 ursa/prompt_library/optimization_prompts.py,sha256=n3fNBsG0hmggAMPb5R7t2lyxBLQt1E_udXM0bAl8Jz4,7929
 ursa/prompt_library/planning_prompts.py,sha256=C8IfVc3ny_5-03bJZop2Yax7wfqS_UIdUGsTZSNQRC0,3534
 ursa/prompt_library/websearch_prompts.py,sha256=n4DJaYn_lIYAVtdy00CCJjT-dLWhn2JNipYqMJAotdY,8846
+ursa/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ursa/tools/feasibility_checker.py,sha256=Tboc_3vDx8Pb-cs56SfiXOsayLz8AkXyicfz6puKZ_U,4172
 ursa/tools/feasibility_tools.py,sha256=Drr3L238hzXieNIzQ0UNfrn66mZon7YnM_diSgU9KQw,36921
 ursa/tools/run_command.py,sha256=sQRuHtRyJYWEyL9dpW_Ukc-xQ5vmKKJK1i_6z3uKEfA,690
-ursa/tools/write_code.py,sha256=DtCsUMZegYm0mk-HMPG5Zo3Ba1gbGfnXHsv1NZTdDs8,1220
+ursa/tools/write_code.py,sha256=cDUJ6nPUfUmr7Fz2PwYUzawWm05V40cTyWNhUU5dfqo,1227
+ursa/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ursa/util/diff_renderer.py,sha256=1L1q2qWWb8gLhR532-LgJn2TrqXDx0gUpPVOWD_sqeU,4086
 ursa/util/helperFunctions.py,sha256=cs-pQEcXyOr4nb5FoH9Ssg3hktycEapW07-MKJcUNOA,4122
 ursa/util/memory_logger.py,sha256=GiKYbQBpxlNRLKyqKFJyrbSbVCkXpRB7Yr5so43tUAw,6097
 ursa/util/optimization_schema.py,sha256=b2wO0BjCIgQb15Q3adDu-ZyG3dfncme9OGQw6FmDGDc,2731
-ursa/util/parse.py,sha256=M0cjyQWmjatxX4WbVmDRUiirTLyW-t_Aemlrlrsc5nA,2811
-ursa_ai-0.6.0rc1.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
-ursa_ai-0.6.0rc1.dist-info/METADATA,sha256=eQk55Q_LYcZ-BA24TALgH9T58Aveg0yTU6m3xFnbGag,9888
-ursa_ai-0.6.0rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ursa_ai-0.6.0rc1.dist-info/entry_points.txt,sha256=B0bbgM-NcAqLujZ1lfmChJZpQMYQDYFJck4moU89Y4E,39
-ursa_ai-0.6.0rc1.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
-ursa_ai-0.6.0rc1.dist-info/RECORD,,
+ursa/util/parse.py,sha256=bTC582fAJDWozXU1XGq4WpY2KR1GnAKVC6kTDFh-kXk,2825
+ursa_ai-0.6.1.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
+ursa_ai-0.6.1.dist-info/METADATA,sha256=tv_wrElNtP6U-bER1D8fO5hpiqrzr8wDRaaD1hJAtzc,9885
+ursa_ai-0.6.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ursa_ai-0.6.1.dist-info/entry_points.txt,sha256=B0bbgM-NcAqLujZ1lfmChJZpQMYQDYFJck4moU89Y4E,39
+ursa_ai-0.6.1.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
+ursa_ai-0.6.1.dist-info/RECORD,,