ursa-ai 0.9.1__py3-none-any.whl

ursa_ai-0.9.1.dist-info/METADATA

@@ -0,0 +1,304 @@
+Metadata-Version: 2.4
+Name: ursa-ai
+Version: 0.9.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
+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: 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: randomname<0.3,>=0.2.1
+Requires-Dist: pandas<3.0,>=2.3.1
+Requires-Dist: pillow<12.0,>=11.3.0
+Requires-Dist: pymupdf<2.0,>=1.26.0
+Requires-Dist: pypdf<6.0,>=5.9.0
+Requires-Dist: rich<14.0,>=13.9.4
+Requires-Dist: mp-api<0.46,>=0.45.8
+Requires-Dist: ddgs>=9.5.5
+Requires-Dist: typer>=0.16.1
+Requires-Dist: trafilatura<1.7,>=1.6.1
+Requires-Dist: selectolax<0.5,>=0.4.0
+Requires-Dist: langchain>=1.0.3
+Requires-Dist: langchain-chroma>=1.0.0
+Requires-Dist: langchain-community>=0.4.1
+Requires-Dist: langchain-openai>=1.0.1
+Requires-Dist: langgraph-checkpoint-sqlite>=3.0.0
+Requires-Dist: langchain-text-splitters>=1.0.0
+Requires-Dist: langchain-ollama>=1.0.0
+Requires-Dist: langchain-mcp-adapters<0.2,>=0.1.12
+Dynamic: license-file
+
+# URSA - The Universal Research and Scientific Agent
+
+<img src="https://github.com/lanl/ursa/raw/main/logos/logo.png" alt="URSA Logo" width="200" height="200">
+
+[![PyPI Version][pypi-version]](https://pypi.org/project/ursa-ai/)
+[![PyPI Downloads][monthly-downloads]](https://pypistats.org/packages/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.
+
+The original arxiv paper is [here](https://arxiv.org/abs/2506.22653).
+
+## 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)
+
+
+## Command line usage
+
+You can install `ursa` as a command line app with `pip install`; or with `uv` via
+
+```bash
+uv tool install ursa-ai
+```
+
+To use the command line app, run
+
+```
+ursa run
+```
+
+This will start a REPL in your terminal.
+
+```
+  __  ________________ _
+ / / / / ___/ ___/ __ `/
+/ /_/ / /  (__  ) /_/ /
+\__,_/_/  /____/\__,_/
+
+For help, type: ? or help. Exit with Ctrl+d.
+ursa>
+```
+
+Within the REPL, you can get help by typing `?` or `help`. 
+
+You can chat with an LLM by simply typing into the terminal.
+
+```
+ursa> How are you?
+Thanks for asking! I’m doing well. How are you today? What can I help you with?
+```
+
+You can run various agents by typing the name of the agent. For example,
+
+```
+ursa> plan
+Enter your prompt for Planning Agent: Write a python script to do linear regression using only numpy.
+```
+
+If you run subsequent agents, the last output will be appended to the prompt for the next agent.
+
+So, to run the Planning Agent followed by the Execution Agent:
+```
+ursa> plan
+Enter your prompt for Planning Agent: Write a python script to do linear regression using only numpy.
+
+...
+
+ursa> execute
+Enter your prompt for Execution Agent: Execute the plan.
+```
+
+You can get a list of available command line options via
+```
+ursa run --help
+```
+
+## MCP serving
+
+You can install `ursa` as a command line app via `pip` or `uv`:
+
+**pip**
+
+```shell
+pip install 'ursa-ai[mcp]'
+```
+
+**uv**
+
+```shell
+uv tool install 'ursa-ai[mcp]'
+```
+
+To start hosting URSA as a local MCP server, run
+
+```shell
+ursa serve
+```
+
+This will start an MCP server on localhost (127.0.0.1) on port 8000.
+
+You can test the server using curl from another terminal:
+
+```shell
+curl -X POST \
+    -H "Content-Type: application/json" \
+    -d '{"agent": "execute", "query": "Plot the first 1000 prime numbers with matplotlib"}' \
+    http://localhost:8000/run
+```
+
+The resulting code is written in the `ursa_mcp` subfolder of the serving
+location. The curl query will get the final summary of what the agent carried
+out. 
+
+When served locally, URSA can then be set up as an MCP tool that can be couple
+to other agentic workflows. The set of agents is the same as the cli (execute,
+plan, arxiv, web, recall, chat)
+
+You can get a list of available command line options via
+```
+ursa serve --help
+```
+
+For API documentation, after `ursa serve`, go to your browser at
+
+```
+http://localhost:8000/docs
+```
+
+## 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.
+
+## Container image
+
+To enable limited sandboxing insofar as containerization does this, you can run
+the following commands:
+
+### Docker
+
+```shell
+# Pull the image
+docker pull ghcr.io/lanl/ursa
+
+# Run included example
+docker run -e "OPENAI_API_KEY"=$OPENAI_API_KEY ursa \
+    bash -c "uv run python examples/single_agent_examples/execution_agnet/integer_sum.py"
+
+# Run script from host system
+mkdir -p scripts
+echo "import ursa; print('Hello from ursa')" > scripts/my_script.py
+docker run -e "OPENAI_API_KEY"=$OPENAI_API_KEY \
+    --mount type=bind,src=$PWD/scripts,dst=/mnt/workspace \
+    ursa \
+    bash -c "uv run /mnt/workspace/my_script.py"
+```
+
+### Charliecloud
+
+[Charliecloud](https://charliecloud.io/) is a rootless alternative to docker
+that is sometimes preferred on HPC. The following commands replicate the
+behaviors above for docker.
+
+```shell
+# Pull the image
+ch-image pull ghcr.io/lanl/ursa
+
+# Convert image to sqfs, for use on another system
+ch-convert ursa ursa.sqfs
+
+# Run included example (if wanted, replace ursa with /path/to/ursa.sqfs)
+ch-run -W ursa \
+    --unset-env="*" \
+    --set-env \
+    --set-env="OPENAI_API_KEY"=$OPENAI_API_KEY \
+    --cd /app \
+    -- bash -c \
+    "uv run examples/single_agent_examples/execution_agnet/integer_sum.py"
+
+# Run script from host system (if wanted, replace ursa with /path/to/ursa.sqfs)
+mkdir -p scripts
+echo "import ursa; print('Hello from ursa')" > scripts/my_script.py
+ch-run -W ursa \
+    --unset-env="*" \
+    --set-env \
+    --set-env="OPENAI_API_KEY"=$OPENAI_API_KEY \
+    --bind ${PWD}/scripts:/mnt/workspace \
+    --cd /app \
+    -- bash -c \
+    "uv run /mnt/workspace/integer_sum.py"
+```
+
+## 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 ruff with `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="https://github.com/lanl/ursa/raw/main/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
+[monthly-downloads]: https://img.shields.io/pypi/dm/ursa-ai?style=flat-square&label=Downloads&color=blue

ursa_ai-0.9.1.dist-info/RECORD

@@ -0,0 +1,51 @@
+ursa/__init__.py,sha256=x0SWdgHPTAtDEx6oSpwW6fCh74mi653U3Wx-i-Xz2ww,95
+ursa/agents/__init__.py,sha256=XApBa4fFidXOt0TKrqNZ3UtybKkdGPtyGp9vOgYwP1I,1524
+ursa/agents/acquisition_agents.py,sha256=lk8XrOQZz-I-komWghJmfjrE7so6prgJ9l6IAXKap3w,28089
+ursa/agents/arxiv_agent.py,sha256=goHdPpUoW1beB-3eBE79KWw4yVQHQNg3rbqUkdmzH54,14507
+ursa/agents/base.py,sha256=GrYdHs8a2D6-D5BewQV8SO_Qn_zKeyBgN4KEHClOkNQ,28663
+ursa/agents/chat_agent.py,sha256=Z40l8tBOIsfR04qQdC5GxE_XqfIOAXZeQ_Xlj3SOzZw,1709
+ursa/agents/code_review_agent.py,sha256=ZvR6CdghHJVvUEeLCBNdSbmDZpMTLDHkB_mwGc4_gTQ,10975
+ursa/agents/execution_agent.py,sha256=lZLd61OEfJenXeROZBMX84yBP1diF0dHNDISt3DBST0,34325
+ursa/agents/hypothesizer_agent.py,sha256=Xe_lmrYCIcnx7ITJ_bcxKQNhx1yPdVfTVz0fiHR1gFk,23166
+ursa/agents/lammps_agent.py,sha256=mCy8q7a-6Np9QzMtMYBzOdifyrFbCr6DtU6P_CeYh6E,16434
+ursa/agents/mp_agent.py,sha256=gE8wEq17Ps536gFy0PkieaT67juN0VQjuyqRnXsgTMk,6838
+ursa/agents/optimization_agent.py,sha256=HhAixoNWjP9bdjU6qHHCOUUi-v057x0r6P-26tns6P0,13994
+ursa/agents/planning_agent.py,sha256=GPnTTl46IY2wY_2Chuko2G4Yi0XCR4F3FuVkwYfmYO4,7299
+ursa/agents/rag_agent.py,sha256=n5XaR7dFhZFnEeIN6SCXFTZUQa4d7amJ2QGYUe5uD1Y,10141
+ursa/agents/recall_agent.py,sha256=BytpPV14IqTSdoMn77BlxXp6ybTgIKjtnNspQDCqvWc,1756
+ursa/agents/websearch_agent.py,sha256=dD_LvCRGlJS2JNceVPvjpSD1rF0CWOpVWzeUrfrcLDY,6681
+ursa/cli/__init__.py,sha256=IdkgDCVB61vBvWJqfQIx_pzLg_NbHbIFsnmQze5TMMw,11594
+ursa/cli/hitl.py,sha256=vs2Qb2KStiD08cNUJ4G3TEejQhlZRtAkZBKlgZTnmYA,17710
+ursa/cli/hitl_api.py,sha256=B3-5pNZdHZ1oerS7B2OsijurY5dn1X1rDk4aGiY5kZg,2974
+ursa/observability/metrics_charts.py,sha256=fqf7jvPINxIDFP3Lm2BbQiW2GaXGzRSGD6F3cMqj4DM,38218
+ursa/observability/metrics_io.py,sha256=Qmbzwe067WjTfILbG0hx3vMCpqAzjPpuLOT1Dx8ZIJ8,279
+ursa/observability/metrics_session.py,sha256=sYCsgrziwqtg1X4InVc-DtLy6cl55OBekbECZkuAGs8,23449
+ursa/observability/pricing.json,sha256=jqWdSyTj3zg7E3-pkMQ2Fzd3PyqJVhQqxeBrblMb1JU,7672
+ursa/observability/pricing.py,sha256=vMF6jGA71TZQkSx0_dIXu5rWrKI9ISV_lnFQXoWaZdY,10598
+ursa/observability/timing.py,sha256=9fhbTWEns4h0Ma48r4C0bqXiFbQS5nhqwV_vQC10L28,49441
+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=PZ2PCWTweBgh_jWgLibAFydfSJVPEUzwPVrR9xr--7U,2756
+ursa/prompt_library/hypothesizer_prompts.py,sha256=ieupOF5tUy_u8actOjPbK-y5Qkrgw6EYxAfw6RXBebs,762
+ursa/prompt_library/literature_prompts.py,sha256=zhBiN3Q-1Z2hp-hkXXp0T8Ipc-6YUM9gw85DjNu1F6I,421
+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=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/logo_generator.py,sha256=lsSd6RbYrXKsB3kI7nvITtfK8xlJ1e_gCxm1jxV-tnY,17849
+ursa/util/memory_logger.py,sha256=GiKYbQBpxlNRLKyqKFJyrbSbVCkXpRB7Yr5so43tUAw,6097
+ursa/util/optimization_schema.py,sha256=b2wO0BjCIgQb15Q3adDu-ZyG3dfncme9OGQw6FmDGDc,2731
+ursa/util/parse.py,sha256=QCQM_gEHs89WZ63CzNoaxWVRDUTADNyOrJZyUZmG36E,13657
+ursa_ai-0.9.1.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
+ursa_ai-0.9.1.dist-info/METADATA,sha256=vB1oxFzOAkjQ7TYLb9z6fbcdm1Ga9SmoX2btlPL-5yg,10908
+ursa_ai-0.9.1.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+ursa_ai-0.9.1.dist-info/entry_points.txt,sha256=B0bbgM-NcAqLujZ1lfmChJZpQMYQDYFJck4moU89Y4E,39
+ursa_ai-0.9.1.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
+ursa_ai-0.9.1.dist-info/RECORD,,

ursa_ai-0.9.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (79.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

ursa_ai-0.9.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+ursa = ursa.cli:main

ursa_ai-0.9.1.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_ai-0.9.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+ursa