ursa-ai 0.2.5__py3-none-any.whl → 0.2.7__py3-none-any.whl

ursa/prompt_library/websearch_prompts.py

@@ -7,15 +7,15 @@
 # - Review the results of the search and identify the source or sources that contain the needed information.
 # - Summarize the information from multiple sources to identify well-supported or inconsistent information.
 # - Perform additional searches until you are confident that you have the information that is requested.
-# - Summarize the information and provide the sources back to the user. 
+# - Summarize the information and provide the sources back to the user.
 # - If you cannot find the requested information, be honest with the user that the information was unavailable.
 # """
 
 # websearch_prompt = """
 # You are an experienced researcher tasked with finding accurate, credible, and relevant information online to address the user's request.
-# 
+#
 # Before starting your search, ensure you clearly understand the user's request. Perform the following actions:
-# 
+#
 # 1. Formulate one or more specific search queries designed to retrieve precise and authoritative information.
 # 2. Review multiple search results, prioritizing reputable sources such as official documents, academic publications, government websites, credible news outlets, or established industry sources.
 # 3. Evaluate the quality, reliability, and recency of each source used.
@@ -24,29 +24,29 @@
 # 6. Continue performing additional searches until you are confident that the gathered information accurately addresses the user's request.
 # 7. Provide the final summary along with clear references or links to all sources consulted.
 # 8. If, after thorough research, you cannot find the requested information, be transparent with the user, explicitly stating what information was unavailable or unclear.
-# 
+#
 # You may also be given feedback by a critic. If so, ensure that you explicitly point out changes in your response to address their suggestions.
-# 
+#
 # Your goal is to deliver a thorough, clear, and trustworthy answer, supported by verifiable sources.
 # """
-# 
+#
 # reflection_prompt = """
 # You are a quality control supervisor responsible for evaluating the researcher's summary of information gathered in response to a user's query.
-# 
+#
 # Carefully assess the researcher’s work according to the following stringent criteria:
-# 
+#
 # - **Correctness:** Ensure the results are credible and the researcher documented reliable sources.
 # - **Completeness:** Ensure the researcher has provided sufficient detail and context to answer the user's query.
-# 
+#
 # Provide a structured evaluation:
-# 
+#
 # 1. Identify the level of strictness that is required for answering the user's query.
 # 2. Clearly list any unsupported assumptions or claims lacking proper citation.
 # 3. Identify any missing information or critical details that should have been included.
 # 4. Suggest specific actions or additional searches the researcher should undertake if the provided information is incomplete or insufficient.
-# 
+#
 # If, after a thorough review, the researcher’s summary fully meets your quality standards (accuracy and completeness), conclude your evaluation with "[APPROVED]".
-# 
+#
 # Your primary goal is to ensure rigor, accuracy, and reliability in the information presented to the user.
 # """
 
@@ -121,11 +121,11 @@ If it does not, rovide a structured evaluation:
 Your primary goal is to ensure rigor, accuracy, and reliability in the information presented to the user.
 """
 
-summarize_prompt = '''
+summarize_prompt = """
 Your goal is to summarize a long user/critic conversation as they work through a complex problem requiring multiple steps.
 
 Your responsibilities is to write a condensed summary of the conversation.
     - Repeat the solution to the original query.
     - Identify all important points from the conversation.
     - Highlight any places where those goals were not achieved and why.
-'''
+"""

ursa/util/diff_renderer.py

@@ -9,17 +9,20 @@ from rich.text import Text
 # unified diff hunk header regex
 _HUNK_RE = re.compile(r"^@@ -(\d+)(?:,\d+)? \+(\d+)(?:,\d+)? @@")
 
+
 @dataclass
 class _LineStyle:
     prefix: str
     bg: str
 
+
 _STYLE = {
     "add": _LineStyle("+ ", "on #003000"),
     "del": _LineStyle("- ", "on #300000"),
     "ctx": _LineStyle("  ", "on grey15"),
 }
 
+
 class DiffRenderer:
     """Renderable diff—`console.print(DiffRenderer(...))`"""
 
@@ -64,7 +67,9 @@ class DiffRenderer:
                 indent_ticks = f" {tick_col} {tick_col}"
                 # pad to the indent width
                 full_indent = indent_ticks.ljust(2 * n + 3)
-                yield Text(f"{full_indent}{raw}".ljust(width), style="white on grey30")
+                yield Text(
+                    f"{full_indent}{raw}".ljust(width), style="white on grey30"
+                )
                 continue
 
             # skip header lines
@@ -98,7 +103,9 @@ class DiffRenderer:
             new_str = str(new_num) if new_num is not None else " "
 
             # Syntax-highlight the code part
-            syntax = Syntax(code, self._lexer_name, line_numbers=False, word_wrap=False)
+            syntax = Syntax(
+                code, self._lexer_name, line_numbers=False, word_wrap=False
+            )
             text_code: Text = syntax.highlight(code)
             if text_code.plain.endswith("\n"):
                 text_code = text_code[:-1]
@@ -118,4 +125,4 @@ class DiffRenderer:
             if pad_len > 0:
                 line_text.append(" " * pad_len, style=style.bg)
 
-            yield line_text
+            yield line_text

ursa/util/memory_logger.py

@@ -33,7 +33,9 @@ class AgentMemory:
         collection_name: str = "agent_memory",
     ) -> None:
         self.path = (
-            Path(path) if path else Path(__file__).resolve().parent / "agent_memory_db"
+            Path(path)
+            if path
+            else Path(__file__).resolve().parent / "agent_memory_db"
         )
         self.collection_name = collection_name
         self.path.mkdir(parents=True, exist_ok=True)
@@ -69,7 +71,9 @@ class AgentMemory:
             Optional metadata dict for each chunk, same length as ``chunks``.
         """
         docs = [
-            Document(page_content=text, metadata=metadatas[i] if metadatas else {})
+            Document(
+                page_content=text, metadata=metadatas[i] if metadatas else {}
+            )
             for i, text in enumerate(chunks)
         ]
 
@@ -107,7 +111,8 @@ class AgentMemory:
             if len(text) > 0:  # only add non-empty documents
                 docs.append(
                     Document(
-                        page_content=text, metadata=metadatas[i] if metadatas else {}
+                        page_content=text,
+                        metadata=metadatas[i] if metadatas else {},
                     )
                 )
         self.vectorstore.add_documents(docs)
@@ -161,9 +166,7 @@ def delete_database(path: Optional[str | Path] = None):
         ``agent_memory_db`` is created in the package’s base directory.
     """
 
-    db_path = (
-        Path(path) if path else Path("~/.cache/ursa/rag/db/")
-    )
+    db_path = Path(path) if path else Path("~/.cache/ursa/rag/db/")
     if os.path.exists(db_path):
         shutil.rmtree(db_path)
         print(f"Database: {db_path} has been deleted.")

{ursa_ai-0.2.5.dist-info → ursa_ai-0.2.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.2.5
+Version: 0.2.7
 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
@@ -42,10 +42,10 @@ Dynamic: license-file
 
 # URSA - The Universal Research and Scientific Agent
 
-<img src="./logos/logo.png" alt="URSA Logo" width="200" height="200">
+<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][total-downloads]](https://pepy.tech/projects/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.
@@ -115,7 +115,7 @@ You have a duty for ensuring that you use URSA responsibly.
 
 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">
+<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.
@@ -127,4 +127,4 @@ Redistribution and use in source and binary forms, with or without modification,
 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
+[monthly-downloads]: https://img.shields.io/pypi/dm/ursa-ai?style=flat-square&label=Downloads&color=blue

ursa_ai-0.2.7.dist-info/RECORD

@@ -0,0 +1,26 @@
+ursa/agents/__init__.py,sha256=fhuGhYRfOSJ5o2yxjOvdEqJFy2j-VWLDkhiNi3wYNRw,498
+ursa/agents/arxiv_agent.py,sha256=nLZ1huMREr-INSDv1RMmOM8a-DcLBjNBWLSi3oFI5hw,14274
+ursa/agents/base.py,sha256=kGbiGn8qu1eKQSv2Y9YZWEv8ngYsyUMTBkAxk8iD9R0,1334
+ursa/agents/code_review_agent.py,sha256=yVO7nzYI3o0k2HguFw3OSY1IrCy5W8V0YiriYAeviY4,11562
+ursa/agents/execution_agent.py,sha256=Hw7EZem8qYedXjeLi5RDPrPIhqlW1G-vE66MlZ7g1BY,16607
+ursa/agents/hypothesizer_agent.py,sha256=p3bLHyqsiGRwYS4nycYcwnpye2j1umWdaOYspGAFRU0,23309
+ursa/agents/mp_agent.py,sha256=Kv793S2x6gavdgBD68wxvTPNFKLDLyI0FSs8iXCkcVQ,9732
+ursa/agents/planning_agent.py,sha256=5KSRk_gDsUrv_6zSxd7CqXhhMCYtnlfNlxSI9tSbqzc,5422
+ursa/agents/recall_agent.py,sha256=UcNRZLbx3j3cHaLEZul4__KzWV4SnUhLTjX9GeoYbHM,823
+ursa/agents/websearch_agent.py,sha256=zDS4IF-WJgsvSmV42HEO582rt3zCh_fJjteh7VpSNe4,7715
+ursa/prompt_library/code_review_prompts.py,sha256=-HuhwW9W_p2LDn44bXLntxLADHCOyl-2KIXxRHto66w,2444
+ursa/prompt_library/execution_prompts.py,sha256=JBBmzVV0605uwFXNv0pxH0fXHqtmOgcDzabjpq3wt2A,2153
+ursa/prompt_library/hypothesizer_prompts.py,sha256=is1SpCbsUtsYsyWOFz3H6M4nCnfyOMPj2p0mOcaEucc,763
+ursa/prompt_library/literature_prompts.py,sha256=zhBiN3Q-1Z2hp-hkXXp0T8Ipc-6YUM9gw85DjNu1F6I,421
+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/run_command.py,sha256=sQRuHtRyJYWEyL9dpW_Ukc-xQ5vmKKJK1i_6z3uKEfA,690
+ursa/tools/write_code.py,sha256=DtCsUMZegYm0mk-HMPG5Zo3Ba1gbGfnXHsv1NZTdDs8,1220
+ursa/util/diff_renderer.py,sha256=gHawyUtBLeOq32A25_etDSy-HXAPyZQrnzfYGtHoEIQ,4086
+ursa/util/memory_logger.py,sha256=Qu8JRjqvXvchnVh6s-91te_xnfOAK1fJDyf1DvsRWnI,5737
+ursa/util/parse.py,sha256=M0cjyQWmjatxX4WbVmDRUiirTLyW-t_Aemlrlrsc5nA,2811
+ursa_ai-0.2.7.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
+ursa_ai-0.2.7.dist-info/METADATA,sha256=93ph0QLYscdrN2SpPVlEVjVtUaxDBpIE2BYMyTTJAkY,6848
+ursa_ai-0.2.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ursa_ai-0.2.7.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
+ursa_ai-0.2.7.dist-info/RECORD,,

ursa_ai-0.2.5.dist-info/RECORD

@@ -1,26 +0,0 @@
-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.5.dist-info/licenses/LICENSE,sha256=4Vr6_u2zTHIUvYjoOBg9ztDbfpV3hyCFv3mTCS87gYU,1482
-ursa_ai-0.2.5.dist-info/METADATA,sha256=2q2rVeCTs4907XF2os0yuda-59iEzvVb2Kt7p0X1P88,6768
-ursa_ai-0.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ursa_ai-0.2.5.dist-info/top_level.txt,sha256=OjA1gRYSUAeiXGnpqPC8iOOGfcjFO1IlP848qMnYSdY,5
-ursa_ai-0.2.5.dist-info/RECORD,,