ursa-ai 0.0.3__py3-none-any.whl → 0.2.2__py3-none-any.whl

ursa/util/parse.py

@@ -0,0 +1,89 @@
+import json
+import re
+
+
+def extract_json(text: str):
+    """
+    Extract a JSON object or array from text that might contain markdown or other content.
+
+    The function attempts three strategies:
+        1. Extract JSON from a markdown code block labeled as JSON.
+        2. Extract JSON from any markdown code block.
+        3. Use bracket matching to extract a JSON substring starting with '{' or '['.
+
+    Returns:
+        A Python object parsed from the JSON string (dict or list).
+
+    Raises:
+        ValueError: If no valid JSON is found.
+    """
+    # Approach 1: Look for a markdown code block specifically labeled as JSON.
+    labeled_block = re.search(
+        r"```json\s*([\[{].*?[\]}])\s*```", text, re.DOTALL
+    )
+    if labeled_block:
+        json_str = labeled_block.group(1).strip()
+        try:
+            return json.loads(json_str)
+        except json.JSONDecodeError:
+            # Fall back to the next approach if parsing fails.
+            pass
+
+    # Approach 2: Look for any code block delimited by triple backticks.
+    generic_block = re.search(r"```(.*?)```", text, re.DOTALL)
+    if generic_block:
+        json_str = generic_block.group(1).strip()
+        if json_str.startswith("{") or json_str.startswith("["):
+            try:
+                return json.loads(json_str)
+            except json.JSONDecodeError:
+                pass
+
+    # Approach 3: Attempt to extract JSON using bracket matching.
+    # Find the first occurrence of either '{' or '['.
+    first_obj = text.find("{")
+    first_arr = text.find("[")
+    if first_obj == -1 and first_arr == -1:
+        raise ValueError("No JSON object or array found in the text.")
+
+    # Determine which bracket comes first.
+    if first_obj == -1:
+        start = first_arr
+        open_bracket = "["
+        close_bracket = "]"
+    elif first_arr == -1:
+        start = first_obj
+        open_bracket = "{"
+        close_bracket = "}"
+    else:
+        if first_obj < first_arr:
+            start = first_obj
+            open_bracket = "{"
+            close_bracket = "}"
+        else:
+            start = first_arr
+            open_bracket = "["
+            close_bracket = "]"
+
+    # Bracket matching: find the matching closing bracket.
+    depth = 0
+    end = None
+    for i in range(start, len(text)):
+        if text[i] == open_bracket:
+            depth += 1
+        elif text[i] == close_bracket:
+            depth -= 1
+            if depth == 0:
+                end = i
+                break
+
+    if end is None:
+        raise ValueError(
+            "Could not find matching closing bracket for JSON content."
+        )
+
+    json_str = text[start : end + 1]
+    try:
+        return json.loads(json_str)
+    except json.JSONDecodeError as e:
+        raise ValueError("Extracted content is not valid JSON.") from e

ursa_ai-0.2.2.dist-info/METADATA

@@ -0,0 +1,130 @@
+Metadata-Version: 2.4
+Name: ursa-ai
+Version: 0.2.2
+Summary: Agents for science at LANL
+Author-email: Mike Grosskopf <mikegros@lanl.gov>, Rahul Somasundaram <rsomasundaram@lanl.gov>, Arthur Lui <alui@lanl.gov>
+Project-URL: Homepage, https://github.com/lanl/ursa
+Project-URL: Documentation, https://github.com/lanl/ursa/tree/main/docs
+Project-URL: Repository, https://github.com/lanl/ursa
+Project-URL: Issues, https://github.com/lanl/ursa/issues
+Classifier: Operating System :: OS Independent
+Classifier: License :: OSI Approved :: BSD License
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: arxiv<3.0,>=2.2.0
+Requires-Dist: beautifulsoup4<5.0,>=4.13.4
+Requires-Dist: coolname<3.0,>=2.2.0
+Requires-Dist: langchain<0.4,>=0.3.22
+Requires-Dist: langchain-community<0.4,>=0.3.20
+Requires-Dist: langchain-litellm>=0.2.2
+Requires-Dist: langchain-openai<0.4,>=0.3.12
+Requires-Dist: langgraph>=0.5
+Requires-Dist: pandas<3.0,>=2.2.3
+Requires-Dist: pillow>=11.2.1
+Requires-Dist: pymupdf<2.0,>=1.26.0
+Requires-Dist: pypdf<6.0,>=5.4.0
+Requires-Dist: rich<14.0,>=13.7.0
+Requires-Dist: langchain-chroma>=0.2.4
+Requires-Dist: chromadb>=1.0.15
+Requires-Dist: mp-api>=0.45.8
+Requires-Dist: langchain-google-genai>=2.1.9
+Requires-Dist: langchain-anthropic>=0.3.18
+Requires-Dist: langgraph-checkpoint-sqlite>=2.0.10
+Requires-Dist: duckduckgo-search>=8.1.1
+Requires-Dist: langchain-ollama>=0.3.6
+Dynamic: license-file
+
+# URSA - The Universal Research and Scientific Agent
+
+<img src="./logos/logo.png" alt="URSA Logo" width="200" height="200">
+
+[![PyPI Version][pypi-version]](https://pypi.org/project/ursa-ai/)
+[![PyPI Downloads][total-downloads]](https://pepy.tech/projects/ursa-ai)
+
+The flexible agentic workflow for accelerating scientific tasks. 
+Composes information flow between agents for planning, code writing and execution, and online research to solve complex problems.
+
+## Installation
+You can install `ursa` via `pip` or `uv`.
+
+**pip**
+```bash
+pip install ursa-ai
+```
+
+**uv**
+```bash
+uv add ursa-ai
+```
+
+## How to use this code
+Better documentation will be incoming, but for now there are examples in the examples folder that should give
+a decent idea for how to set up some basic problems. They also should give some idea of how to pass results from
+one agent to another. I will look to add things with multi-agent graphs, etc. in the future. 
+
+Documentation for each URSA agent:
+- [Planning Agent](docs/planning_agent.md)
+- [Execution Agent](docs/execution_agent.md)
+- [ArXiv Agent](docs/arxiv_agent.md)
+- [Web Search Agent](docs/web_search_agent.md)
+- [Hypothesizer Agent](docs/hypothesizer_agent.md)
+
+Documentation for combining agents:
+- [ArXiv -> Execution for Materials](docs/combining_arxiv_and_execution.md)
+- [ArXiv -> Execution for Neutron Star Properties](docs/combining_arxiv_and_execution_neutronStar.md)
+
+# Sandboxing
+The Execution Agent is allowed to run system commands and write/run code. Being able to execute arbitrary system commands or write
+and execute code has the potential to cause problems like:
+- Damage code or data on the computer
+- Damage the computer
+- Transmit your local data
+
+The Web Search Agent scrapes data from urls, so has the potential to attempt to pull information from questionable sources.
+
+Some suggestions for sandboxing the agent:
+- Creating a specific environment such that limits URSA's access to only what you want. Examples:
+    - Creating/using a virtual machine that is sandboxed from the rest of your machine
+    - Creating a new account on your machine specifically for URSA 
+- Creating a network blacklist/whitelist to ensure that network commands and webscraping are contained to safe sources
+
+You have a duty for ensuring that you use URSA responsibly.
+
+## Development Dependencies
+
+* [`uv`](https://docs.astral.sh/uv/)
+    * `uv` is an extremely fast python package and project manager, written in Rust.
+      Follow installation instructions
+      [here](https://docs.astral.sh/uv/getting-started/installation/)
+
+* [`ruff`](https://docs.astral.sh/ruff/)
+    * An extremely fast Python linter and code formatter, written in Rust.
+    * After installing `uv`, you can install just ruff `uv tool install ruff`
+
+* [`just`](https://github.com/casey/just)
+    * A modern way to save and run project-specific commands
+    * After installing `uv`, you can install just with `uv tool install rust-just`
+
+## Development Team
+
+URSA has been developed at Los Alamos National Laboratory as part of the ArtIMis project.
+
+<img src="./logos/artimis.png" alt="ArtIMis Logo" width="200" height="200">
+
+### Notice of Copyright Assertion (O4958):
+*This program is Open-Source under the BSD-3 License.
+Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:*
+- *Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.*
+- *Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.*
+- *Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.*
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+[pypi-version]: https://img.shields.io/pypi/v/ursa-ai?style=flat-square&label=PyPI
+[total-downloads]: https://img.shields.io/pepy/dt/ursa-ai?style=flat-square&label=downloads&color=blue

ursa_ai-0.2.2.dist-info/RECORD

@@ -0,0 +1,26 @@
+ursa/agents/__init__.py,sha256=J2GVc4GpQwNHvzxePgjtAHgz806ZGrSPJs3QyOPHFlA,616
+ursa/agents/arxiv_agent.py,sha256=1w3Mu8iK3vhP1zWCtqnuMQ5CQ74H6oy6RPWJPBkLC_o,13151
+ursa/agents/base.py,sha256=4wHwXPAcmH_Ml3SG-E4aJwcMVkNPAG9MBbKTWpCQVsc,1337
+ursa/agents/code_review_agent.py,sha256=0qTTeTgYSM0n1Z7nPtDKP7790Ea9bm9Jg2E4w8jLDC8,11376
+ursa/agents/execution_agent.py,sha256=NoEsT4Y9wpMU0rceQP9RUhmBFBjPfjy6zp0hgiQXnQA,16468
+ursa/agents/hypothesizer_agent.py,sha256=FQwLKUKsoV-yw9dS8Tync1fy1Tw84VCx4zThX_SJH8I,23210
+ursa/agents/mp_agent.py,sha256=qMosIZncfCqsNNNHsShc0paq2bH_gzrHZiaEf_a6j-8,9220
+ursa/agents/planning_agent.py,sha256=CdXNgH4VDvdcrgtwtydvvZOeV5DPLDkX24I_iWYBAew,5192
+ursa/agents/recall_agent.py,sha256=wtUf3tkcyUAdhQ8gNV_mLeXBoAhFV-RDZ20EMTweXSw,836
+ursa/agents/websearch_agent.py,sha256=_JlTPSeA_UY2nwMaiiYBDWzvcgoys_RWBfNIUWMfbr8,7117
+ursa/prompt_library/code_review_prompts.py,sha256=6hI158aIJ4LPxuwuGHSsqj3AIngpJVez05twPQSCFaw,2442
+ursa/prompt_library/execution_prompts.py,sha256=hDmvSZzGcmza4DP1Z-yDVAJCTcvQ-sgz8bQxBAVqaeg,2153
+ursa/prompt_library/hypothesizer_prompts.py,sha256=is1SpCbsUtsYsyWOFz3H6M4nCnfyOMPj2p0mOcaEucc,763
+ursa/prompt_library/literature_prompts.py,sha256=xUteIC6Cf8CvKFNKy_YxViiJBK1ldYu2Y18BNOWOZ68,420
+ursa/prompt_library/planning_prompts.py,sha256=SvvKbPU29yHnM4dLyNu12Z8OrBIjFWc_jUPAtsEtJOA,3536
+ursa/prompt_library/websearch_prompts.py,sha256=Dyu26-o-QpUjo_m4Uj7mE_SXzLxPueT8P-yXI1s3RZU,8858
+ursa/tools/run_command.py,sha256=sQRuHtRyJYWEyL9dpW_Ukc-xQ5vmKKJK1i_6z3uKEfA,690
+ursa/tools/write_code.py,sha256=DtCsUMZegYm0mk-HMPG5Zo3Ba1gbGfnXHsv1NZTdDs8,1220
+ursa/util/diff_renderer.py,sha256=efiXAd6Lz1-4QSNNX4_yYSK4xEInnmWUENXqg7emOPg,4014
+ursa/util/memory_logger.py,sha256=Prnqs-ZJDiCVGzsKAxBK5NUUOGCrzE9mGPKjz7adRFY,5674
+ursa/util/parse.py,sha256=M0cjyQWmjatxX4WbVmDRUiirTLyW-t_Aemlrlrsc5nA,2811
+ursa_ai-0.2.2.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
+ursa_ai-0.2.2.dist-info/METADATA,sha256=01crOx6RAtm-YcYSvGmXJGrafsgeGD5P-R4VI57Bmy0,6557
+ursa_ai-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ursa_ai-0.2.2.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
+ursa_ai-0.2.2.dist-info/RECORD,,

ursa_ai-0.2.2.dist-info/licenses/LICENSE

@@ -0,0 +1,8 @@
+This program is Open-Source under the BSD-3 License.
+
+Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
+    Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
+    Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
+    Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

ursa/__init__.py

@@ -1,2 +0,0 @@
-def hello() -> str:
-    return "Hello from ursa!"

ursa_ai-0.0.3.dist-info/METADATA

@@ -1,7 +0,0 @@
-Metadata-Version: 2.4
-Name: ursa-ai
-Version: 0.0.3
-Summary: Add your description here
-Author-email: Mike Grosskopf <mikegros@lanl.gov>, Arthur Lui <alui@lanl.gov>
-Requires-Python: >=3.11
-Description-Content-Type: text/markdown

ursa_ai-0.0.3.dist-info/RECORD

@@ -1,6 +0,0 @@
-ursa/__init__.py,sha256=axGrqZf_bAtq20gqwmzW5O2nu4qKHfvXIpH2y5ihh18,50
-ursa/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ursa_ai-0.0.3.dist-info/METADATA,sha256=bfGgg-JCXWuu80P0nK3NtNiLJKikBVvqYzA8yIg1Okw,227
-ursa_ai-0.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ursa_ai-0.0.3.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
-ursa_ai-0.0.3.dist-info/RECORD,,