ursa-ai 0.2.9__tar.gz → 0.2.11__tar.gz

{ursa_ai-0.2.9/src/ursa_ai.egg-info → ursa_ai-0.2.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.2.9
+Version: 0.2.11
 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

{ursa_ai-0.2.9 → ursa_ai-0.2.11}/pyproject.toml

@@ -63,15 +63,12 @@ enabled = true
 
 [tool.ruff]
 line-length = 80
+preview = true
 
 [tool.ruff.lint]
 extend-select = ["I"]  # W505 (doc-line-too-long); D (pydocstyle)
 pydocstyle.convention = "numpy"
 pycodestyle.max-doc-length = 80
-# extend-unfixable = ["F401"]  # unused imports
-
-[tool.ruff.lint.per-file-ignores]
-"__init__.py" = ["F401"]
 
 # Ignore test file documentation linting.
 [tool.ruff.lint.extend-per-file-ignores]

ursa_ai-0.2.11/src/ursa/agents/__init__.py

@@ -0,0 +1,17 @@
+from .arxiv_agent import ArxivAgent as ArxivAgent
+from .arxiv_agent import PaperMetadata as PaperMetadata
+from .arxiv_agent import PaperState as PaperState
+from .base import BaseAgent as BaseAgent
+from .base import BaseChatModel as BaseChatModel
+from .code_review_agent import CodeReviewAgent as CodeReviewAgent
+from .code_review_agent import CodeReviewState as CodeReviewState
+from .execution_agent import ExecutionAgent as ExecutionAgent
+from .execution_agent import ExecutionState as ExecutionState
+from .hypothesizer_agent import HypothesizerAgent as HypothesizerAgent
+from .hypothesizer_agent import HypothesizerState as HypothesizerState
+from .mp_agent import MaterialsProjectAgent as MaterialsProjectAgent
+from .planning_agent import PlanningAgent as PlanningAgent
+from .planning_agent import PlanningState as PlanningState
+from .recall_agent import RecallAgent as RecallAgent
+from .websearch_agent import WebSearchAgent as WebSearchAgent
+from .websearch_agent import WebSearchState as WebSearchState

{ursa_ai-0.2.9 → ursa_ai-0.2.11}/src/ursa/agents/arxiv_agent.py

@@ -211,12 +211,10 @@ class ArxivAgent(BaseAgent):
                 except Exception as e:
                     full_text = f"Error loading paper: {e}"
 
-            papers.append(
-                {
-                    "arxiv_id": arxiv_id,
-                    "full_text": full_text,
-                }
-            )
+            papers.append({
+                "arxiv_id": arxiv_id,
+                "full_text": full_text,
+            })
 
         return papers
 
@@ -279,28 +277,23 @@ class ArxivAgent(BaseAgent):
                     )
 
                     if relevant_docs_with_scores:
-                        score = sum(
-                            [s for _, s in relevant_docs_with_scores]
-                        ) / len(relevant_docs_with_scores)
+                        score = sum([
+                            s for _, s in relevant_docs_with_scores
+                        ]) / len(relevant_docs_with_scores)
                         relevancy_scores[i] = abs(1.0 - score)
                     else:
                         relevancy_scores[i] = 0.0
 
-                    retrieved_content = "\n\n".join(
-                        [
-                            doc.page_content
-                            for doc, _ in relevant_docs_with_scores
-                        ]
-                    )
+                    retrieved_content = "\n\n".join([
+                        doc.page_content for doc, _ in relevant_docs_with_scores
+                    ])
                 else:
                     retrieved_content = cleaned_text
 
-                summary = chain.invoke(
-                    {
-                        "retrieved_content": retrieved_content,
-                        "context": state["context"],
-                    }
-                )
+                summary = chain.invoke({
+                    "retrieved_content": retrieved_content,
+                    "context": state["context"],
+                })
 
             except Exception as e:
                 summary = f"Error summarizing paper: {e}"
@@ -376,9 +369,10 @@ class ArxivAgent(BaseAgent):
 
         chain = prompt | self.llm | StrOutputParser()
 
-        final_summary = chain.invoke(
-            {"Summaries": combined, "context": state["context"]}
-        )
+        final_summary = chain.invoke({
+            "Summaries": combined,
+            "context": state["context"],
+        })
 
         with open(self.summaries_path + "/final_summary.txt", "w") as f:
             f.write(final_summary)
@@ -406,9 +400,10 @@ class ArxivAgent(BaseAgent):
         return graph
 
     def run(self, arxiv_search_query: str, context: str) -> str:
-        result = self.graph.invoke(
-            {"query": arxiv_search_query, "context": context}
-        )
+        result = self.graph.invoke({
+            "query": arxiv_search_query,
+            "context": context,
+        })
 
         if self.summarize:
             return result.get("final_summary", "No summary generated.")

{ursa_ai-0.2.9 → ursa_ai-0.2.11}/src/ursa/agents/mp_agent.py

@@ -141,9 +141,10 @@ You are a materials-science assistant. Given the following metadata about a mate
         {context}
                 """)
         chain = prompt | self.llm | StrOutputParser()
-        final = chain.invoke(
-            {"summaries": combined, "context": state["context"]}
-        )
+        final = chain.invoke({
+            "summaries": combined,
+            "context": state["context"],
+        })
         return {**state, "final_summary": final}
 
     def _build_graph(self):

{ursa_ai-0.2.9 → ursa_ai-0.2.11}/src/ursa/util/memory_logger.py

@@ -26,20 +26,30 @@ class AgentMemory:
     * Requires `langchain-chroma`, and `chromadb`.
     """
 
+    @classmethod
+    def get_db_path(cls, path: Optional[str | Path]) -> Path:
+        match path:
+            case None:
+                return Path.home() / ".cache" / "ursa" / "rag" / "db"
+            case str():
+                return Path(str)
+            case Path():
+                return path
+            case _:
+                raise TypeError(
+                    f"Type of path is `{type(path)}` "
+                    "but `Optional[str | Path]` was expected."
+                )
+
     def __init__(
         self,
         embedding_model,
         path: Optional[str | Path] = None,
         collection_name: str = "agent_memory",
     ) -> None:
-        self.path = (
-            Path(path)
-            if path
-            else Path(__file__).resolve().parent / "agent_memory_db"
-        )
+        self.path = self.get_db_path(path)
         self.collection_name = collection_name
         self.path.mkdir(parents=True, exist_ok=True)
-
         self.embeddings = embedding_model
 
         # If a DB already exists, load it; otherwise defer creation until `build_index`.
@@ -165,8 +175,7 @@ def delete_database(path: Optional[str | Path] = None):
         Where the on-disk Chroma DB is for deleting.  If *None*, a folder called
         ``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 = AgentMemory.get_db_path(path)
     if os.path.exists(db_path):
         shutil.rmtree(db_path)
         print(f"Database: {db_path} has been deleted.")

{ursa_ai-0.2.9 → ursa_ai-0.2.11/src/ursa_ai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ursa-ai
-Version: 0.2.9
+Version: 0.2.11
 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

ursa_ai-0.2.9/src/ursa/agents/__init__.py

@@ -1,9 +0,0 @@
-from .arxiv_agent import ArxivAgent, PaperMetadata, PaperState
-from .base import BaseAgent, BaseChatModel
-from .code_review_agent import CodeReviewAgent, CodeReviewState
-from .execution_agent import ExecutionAgent, ExecutionState
-from .hypothesizer_agent import HypothesizerAgent, HypothesizerState
-from .mp_agent import MaterialsProjectAgent
-from .planning_agent import PlanningAgent, PlanningState
-from .recall_agent import RecallAgent
-from .websearch_agent import WebSearchAgent, WebSearchState