ursa-ai 0.7.0rc1__tar.gz → 0.7.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of ursa-ai might be problematic. Click here for more details.

Files changed (54) hide show
  1. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/PKG-INFO +5 -1
  2. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/README.md +2 -0
  3. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/pyproject.toml +5 -1
  4. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/__init__.py +13 -2
  5. ursa_ai-0.7.1/src/ursa/agents/acquisition_agents.py +812 -0
  6. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/arxiv_agent.py +1 -1
  7. ursa_ai-0.7.1/src/ursa/agents/base.py +669 -0
  8. ursa_ai-0.7.1/src/ursa/agents/chat_agent.py +58 -0
  9. ursa_ai-0.7.1/src/ursa/agents/execution_agent.py +800 -0
  10. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/lammps_agent.py +81 -31
  11. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/planning_agent.py +27 -2
  12. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/websearch_agent.py +2 -2
  13. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/cli/__init__.py +5 -1
  14. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/cli/hitl.py +46 -34
  15. ursa_ai-0.7.1/src/ursa/observability/pricing.json +85 -0
  16. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/observability/pricing.py +20 -18
  17. ursa_ai-0.7.1/src/ursa/util/parse.py +405 -0
  18. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa_ai.egg-info/PKG-INFO +5 -1
  19. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa_ai.egg-info/SOURCES.txt +3 -0
  20. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa_ai.egg-info/requires.txt +2 -0
  21. ursa_ai-0.7.0rc1/src/ursa/agents/base.py +0 -408
  22. ursa_ai-0.7.0rc1/src/ursa/agents/execution_agent.py +0 -554
  23. ursa_ai-0.7.0rc1/src/ursa/util/parse.py +0 -89
  24. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/LICENSE +0 -0
  25. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/setup.cfg +0 -0
  26. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/__init__.py +0 -0
  27. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/code_review_agent.py +0 -0
  28. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/hypothesizer_agent.py +0 -0
  29. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/mp_agent.py +0 -0
  30. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/optimization_agent.py +0 -0
  31. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/rag_agent.py +0 -0
  32. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/agents/recall_agent.py +0 -0
  33. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/observability/timing.py +0 -0
  34. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/__init__.py +0 -0
  35. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/code_review_prompts.py +0 -0
  36. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/execution_prompts.py +0 -0
  37. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/hypothesizer_prompts.py +0 -0
  38. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/literature_prompts.py +0 -0
  39. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/optimization_prompts.py +0 -0
  40. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/planning_prompts.py +0 -0
  41. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/prompt_library/websearch_prompts.py +0 -0
  42. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/tools/__init__.py +0 -0
  43. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/tools/feasibility_checker.py +0 -0
  44. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/tools/feasibility_tools.py +0 -0
  45. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/tools/run_command.py +0 -0
  46. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/tools/write_code.py +0 -0
  47. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/util/__init__.py +0 -0
  48. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/util/diff_renderer.py +0 -0
  49. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/util/helperFunctions.py +0 -0
  50. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/util/memory_logger.py +0 -0
  51. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa/util/optimization_schema.py +0 -0
  52. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa_ai.egg-info/dependency_links.txt +0 -0
  53. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa_ai.egg-info/entry_points.txt +0 -0
  54. {ursa_ai-0.7.0rc1 → ursa_ai-0.7.1}/src/ursa_ai.egg-info/top_level.txt +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: ursa-ai
3
- Version: 0.7.0rc1
3
+ Version: 0.7.1
4
4
  Summary: Agents for science at LANL
5
5
  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>
6
6
  License-Expression: BSD-3-Clause
@@ -39,6 +39,8 @@ Requires-Dist: langgraph-checkpoint-sqlite<3.0,>=2.0.10
39
39
  Requires-Dist: langchain-ollama<0.4,>=0.3.6
40
40
  Requires-Dist: ddgs>=9.5.5
41
41
  Requires-Dist: typer>=0.16.1
42
+ Requires-Dist: trafilatura<1.7,>=1.6.1
43
+ Requires-Dist: selectolax<0.5,>=0.4.0
42
44
  Dynamic: license-file
43
45
 
44
46
  # URSA - The Universal Research and Scientific Agent
@@ -51,6 +53,8 @@ Dynamic: license-file
51
53
  The flexible agentic workflow for accelerating scientific tasks.
52
54
  Composes information flow between agents for planning, code writing and execution, and online research to solve complex problems.
53
55
 
56
+ The original arxiv paper is [here](https://arxiv.org/abs/2506.22653).
57
+
54
58
  ## Installation
55
59
  You can install `ursa` via `pip` or `uv`.
56
60
 
@@ -8,6 +8,8 @@
8
8
  The flexible agentic workflow for accelerating scientific tasks.
9
9
  Composes information flow between agents for planning, code writing and execution, and online research to solve complex problems.
10
10
 
11
+ The original arxiv paper is [here](https://arxiv.org/abs/2506.22653).
12
+
11
13
  ## Installation
12
14
  You can install `ursa` via `pip` or `uv`.
13
15
 
@@ -39,6 +39,8 @@ dependencies = [
39
39
  "langchain-ollama>=0.3.6,<0.4",
40
40
  "ddgs>=9.5.5",
41
41
  "typer>=0.16.1",
42
+ "trafilatura>=1.6.1,<1.7",
43
+ "selectolax>=0.4.0,<0.5",
42
44
  ]
43
45
  classifiers = [
44
46
  "Operating System :: OS Independent",
@@ -78,6 +80,9 @@ pycodestyle.max-doc-length = 80
78
80
  [tool.ruff.lint.extend-per-file-ignores]
79
81
  "tests/**/*.py" = ["D"]
80
82
 
83
+ [tool.setuptools.package-data]
84
+ ursa = ["observability/pricing.json"]
85
+
81
86
  [dependency-groups]
82
87
  dev = [
83
88
  "langgraph-checkpoint-sqlite>=2.0.10",
@@ -94,7 +99,6 @@ docs = [
94
99
  ]
95
100
  lammps = [
96
101
  "atomman>=1.5.2",
97
- "trafilatura>=1.6.1",
98
102
  ]
99
103
  opt = [
100
104
  "ortools>=9.14,<9.15",
@@ -1,8 +1,19 @@
1
- from .arxiv_agent import ArxivAgent as ArxivAgent
1
+ from .acquisition_agents import (
2
+ ArxivAgent as ArxivAgent,
3
+ )
4
+ from .acquisition_agents import (
5
+ OSTIAgent as OSTIAgent,
6
+ )
7
+ from .acquisition_agents import (
8
+ WebSearchAgent as WebSearchAgent,
9
+ )
10
+ from .arxiv_agent import ArxivAgentLegacy as ArxivAgentLegacy
2
11
  from .arxiv_agent import PaperMetadata as PaperMetadata
3
12
  from .arxiv_agent import PaperState as PaperState
4
13
  from .base import BaseAgent as BaseAgent
5
14
  from .base import BaseChatModel as BaseChatModel
15
+ from .chat_agent import ChatAgent as ChatAgent
16
+ from .chat_agent import ChatState as ChatState
6
17
  from .code_review_agent import CodeReviewAgent as CodeReviewAgent
7
18
  from .code_review_agent import CodeReviewState as CodeReviewState
8
19
  from .execution_agent import ExecutionAgent as ExecutionAgent
@@ -17,5 +28,5 @@ from .planning_agent import PlanningState as PlanningState
17
28
  from .rag_agent import RAGAgent as RAGAgent
18
29
  from .rag_agent import RAGState as RAGState
19
30
  from .recall_agent import RecallAgent as RecallAgent
20
- from .websearch_agent import WebSearchAgent as WebSearchAgent
31
+ from .websearch_agent import WebSearchAgentLegacy as WebSearchAgentLegacy
21
32
  from .websearch_agent import WebSearchState as WebSearchState