ursa-ai 0.6.0rc2__tar.gz → 0.6.2__tar.gz

{ursa_ai-0.6.0rc2 → ursa_ai-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.6.0rc2
+Version: 0.6.2
 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.0rc2 → ursa_ai-0.6.2}/pyproject.toml

@@ -86,10 +86,16 @@ dev = [
     "pytest>=8.4.2",
     "scikit-optimize>=0.10.2",
 ]
+docs = [
+    "mkdocs>=1.6.1",
+    "mkdocs-autorefs>=1.4.3",
+    "mkdocs-material>=9.6.21",
+    "mkdocstrings-python>=1.18.2",
+]
 lammps = [
     "atomman>=1.5.2",
     "trafilatura>=1.6.1",
 ]
 opt = [
     "ortools>=9.14,<9.15",
-]
+]

{ursa_ai-0.6.0rc2 → ursa_ai-0.6.2}/src/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_ai-0.6.0rc2 → ursa_ai-0.6.2}/src/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_ai-0.6.0rc2 → ursa_ai-0.6.2}/src/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_ai-0.6.0rc2 → ursa_ai-0.6.2}/src/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.0rc2 → ursa_ai-0.6.2}/src/ursa_ai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.6.0rc2
+Version: 0.6.2
 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.0rc2 → ursa_ai-0.6.2}/src/ursa_ai.egg-info/SOURCES.txt

@@ -1,6 +1,7 @@
 LICENSE
 README.md
 pyproject.toml
+src/ursa/__init__.py
 src/ursa/agents/__init__.py
 src/ursa/agents/arxiv_agent.py
 src/ursa/agents/base.py
@@ -18,6 +19,7 @@ src/ursa/cli/__init__.py
 src/ursa/cli/hitl.py
 src/ursa/observability/pricing.py
 src/ursa/observability/timing.py
+src/ursa/prompt_library/__init__.py
 src/ursa/prompt_library/code_review_prompts.py
 src/ursa/prompt_library/execution_prompts.py
 src/ursa/prompt_library/hypothesizer_prompts.py
@@ -25,10 +27,12 @@ src/ursa/prompt_library/literature_prompts.py
 src/ursa/prompt_library/optimization_prompts.py
 src/ursa/prompt_library/planning_prompts.py
 src/ursa/prompt_library/websearch_prompts.py
+src/ursa/tools/__init__.py
 src/ursa/tools/feasibility_checker.py
 src/ursa/tools/feasibility_tools.py
 src/ursa/tools/run_command.py
 src/ursa/tools/write_code.py
+src/ursa/util/__init__.py
 src/ursa/util/diff_renderer.py
 src/ursa/util/helperFunctions.py
 src/ursa/util/memory_logger.py