ursa-ai 0.9.1__py3-none-any.whl

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.
Files changed (51) hide show
  1. ursa/__init__.py +3 -0
  2. ursa/agents/__init__.py +32 -0
  3. ursa/agents/acquisition_agents.py +812 -0
  4. ursa/agents/arxiv_agent.py +429 -0
  5. ursa/agents/base.py +728 -0
  6. ursa/agents/chat_agent.py +60 -0
  7. ursa/agents/code_review_agent.py +341 -0
  8. ursa/agents/execution_agent.py +915 -0
  9. ursa/agents/hypothesizer_agent.py +614 -0
  10. ursa/agents/lammps_agent.py +465 -0
  11. ursa/agents/mp_agent.py +204 -0
  12. ursa/agents/optimization_agent.py +410 -0
  13. ursa/agents/planning_agent.py +219 -0
  14. ursa/agents/rag_agent.py +304 -0
  15. ursa/agents/recall_agent.py +54 -0
  16. ursa/agents/websearch_agent.py +196 -0
  17. ursa/cli/__init__.py +363 -0
  18. ursa/cli/hitl.py +516 -0
  19. ursa/cli/hitl_api.py +75 -0
  20. ursa/observability/metrics_charts.py +1279 -0
  21. ursa/observability/metrics_io.py +11 -0
  22. ursa/observability/metrics_session.py +750 -0
  23. ursa/observability/pricing.json +97 -0
  24. ursa/observability/pricing.py +321 -0
  25. ursa/observability/timing.py +1466 -0
  26. ursa/prompt_library/__init__.py +0 -0
  27. ursa/prompt_library/code_review_prompts.py +51 -0
  28. ursa/prompt_library/execution_prompts.py +50 -0
  29. ursa/prompt_library/hypothesizer_prompts.py +17 -0
  30. ursa/prompt_library/literature_prompts.py +11 -0
  31. ursa/prompt_library/optimization_prompts.py +131 -0
  32. ursa/prompt_library/planning_prompts.py +79 -0
  33. ursa/prompt_library/websearch_prompts.py +131 -0
  34. ursa/tools/__init__.py +0 -0
  35. ursa/tools/feasibility_checker.py +114 -0
  36. ursa/tools/feasibility_tools.py +1075 -0
  37. ursa/tools/run_command.py +27 -0
  38. ursa/tools/write_code.py +42 -0
  39. ursa/util/__init__.py +0 -0
  40. ursa/util/diff_renderer.py +128 -0
  41. ursa/util/helperFunctions.py +142 -0
  42. ursa/util/logo_generator.py +625 -0
  43. ursa/util/memory_logger.py +183 -0
  44. ursa/util/optimization_schema.py +78 -0
  45. ursa/util/parse.py +405 -0
  46. ursa_ai-0.9.1.dist-info/METADATA +304 -0
  47. ursa_ai-0.9.1.dist-info/RECORD +51 -0
  48. ursa_ai-0.9.1.dist-info/WHEEL +5 -0
  49. ursa_ai-0.9.1.dist-info/entry_points.txt +2 -0
  50. ursa_ai-0.9.1.dist-info/licenses/LICENSE +8 -0
  51. ursa_ai-0.9.1.dist-info/top_level.txt +1 -0
ursa/__init__.py ADDED
@@ -0,0 +1,3 @@
1
+ from importlib.metadata import version as get_version
2
+
3
+ __version__ = get_version("ursa-ai")
@@ -0,0 +1,32 @@
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
11
+ from .arxiv_agent import PaperMetadata as PaperMetadata
12
+ from .arxiv_agent import PaperState as PaperState
13
+ from .base import BaseAgent as BaseAgent
14
+ from .base import BaseChatModel as BaseChatModel
15
+ from .chat_agent import ChatAgent as ChatAgent
16
+ from .chat_agent import ChatState as ChatState
17
+ from .code_review_agent import CodeReviewAgent as CodeReviewAgent
18
+ from .code_review_agent import CodeReviewState as CodeReviewState
19
+ from .execution_agent import ExecutionAgent as ExecutionAgent
20
+ from .execution_agent import ExecutionState as ExecutionState
21
+ from .hypothesizer_agent import HypothesizerAgent as HypothesizerAgent
22
+ from .hypothesizer_agent import HypothesizerState as HypothesizerState
23
+ from .lammps_agent import LammpsAgent as LammpsAgent
24
+ from .lammps_agent import LammpsState as LammpsState
25
+ from .mp_agent import MaterialsProjectAgent as MaterialsProjectAgent
26
+ from .planning_agent import PlanningAgent as PlanningAgent
27
+ from .planning_agent import PlanningState as PlanningState
28
+ from .rag_agent import RAGAgent as RAGAgent
29
+ from .rag_agent import RAGState as RAGState
30
+ from .recall_agent import RecallAgent as RecallAgent
31
+ from .websearch_agent import WebSearchAgentLegacy as WebSearchAgentLegacy
32
+ from .websearch_agent import WebSearchState as WebSearchState