fabricatio 0.2.6.dev2__tar.gz → 0.2.6.dev3__tar.gz

{fabricatio-0.2.6.dev2 → fabricatio-0.2.6.dev3}/Makefile

@@ -12,16 +12,16 @@ tools:
 	rm $(DATA)/scripts/*.dwarf || true
 
 dev: tools
-	uvx  --with-editable . maturin develop --uv -r
+	uvx --project . maturin develop --uv -r
 
 bdist:clean tools
-	uvx  --with-editable . maturin sdist -o $(DIST)
-	uvx  --with-editable . maturin build  -r -o $(DIST)
+	uvx --project . maturin sdist -o $(DIST)
+	uvx --project . maturin build  -r -o $(DIST)
 
 clean:
 	rm -rf $(DIST) $(DATA)
 
 publish:tools
-	uvx  --with-editable . maturin publish --skip-existing
-	uvx  --with-editable . maturin upload --skip-existing $(DIST)/*
+	uvx --project . maturin publish --skip-existing
+	uvx --project . maturin upload --skip-existing $(DIST)/*
 .PHONY: tools

{fabricatio-0.2.6.dev2 → fabricatio-0.2.6.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.6.dev2
+Version: 0.2.6.dev3
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12

fabricatio-0.2.6.dev3/examples/correct/correct_loop.py

@@ -0,0 +1,30 @@
+"""Example of review usage."""
+
+import asyncio
+
+from fabricatio import Role, logger
+from questionary import confirm
+from rich import print as r_print
+
+
+async def main() -> None:
+    """Main function."""
+    role = Role(
+        name="Reviewer",
+        description="A role that reviews the code.",
+    )
+
+    code = await role.aask(
+        "write a cli app using rust with clap which can generate a basic manifest of a standard rust project, output code only,no extra explanation, you should using derive mode of clap."
+    )
+
+    logger.success(f"Code: \n{code}")
+
+    while await confirm("Do you want to review the code?").ask_async():
+        code = await role.correct_string(code, topic="If the cli app is of good design", supervisor_check=False)
+        r_print(code)
+    logger.success(f"Corrected: \n{code}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

fabricatio-0.2.6.dev3/examples/search_bibtex/.gitignore

@@ -0,0 +1 @@
+*.bib

fabricatio-0.2.6.dev3/examples/search_bibtex/search.py

@@ -0,0 +1,12 @@
+from fabricatio import BibManager
+
+b = BibManager("Exported Items.bib")
+print(b.get_cite_key("Fault Diagnosis of Wind Turbines"))
+print(
+    b.get_cite_key_fuzzy(
+        "Fault Diagnosis of Wind Turbines"
+    )
+)
+
+print(b.list_titles())
+print(b.list_titles(True))

{fabricatio-0.2.6.dev2 → fabricatio-0.2.6.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fabricatio"
-version = "0.2.6-dev2"
+version = "0.2.6-dev3"
 description = "A LLM multi-agent framework."
 readme = "README.md"
 license = { file = "LICENSE" }
@@ -85,7 +85,7 @@ full = [
 ]
 
 [tool.ruff]
-include = ["pyproject.toml", "python/fabricatio/*.py","python/fabricatio/*.pyi", "examples/*.py"]
+include = ["pyproject.toml", "python/fabricatio/*.py", "python/fabricatio/*.pyi", "examples/*.py"]
 line-length = 120
 target-version = "py312"
 [tool.ruff.format]

{fabricatio-0.2.6.dev2 → fabricatio-0.2.6.dev3}/python/fabricatio/_rust.pyi

@@ -1,6 +1,7 @@
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
+
 class TemplateManager:
     """Template rendering engine using Handlebars templates.
 
@@ -9,8 +10,9 @@ class TemplateManager:
 
     See: https://crates.io/crates/handlebars
     """
+
     def __init__(
-        self, template_dirs: List[Path], suffix: Optional[str] = None, active_loading: Optional[bool] = None
+            self, template_dirs: List[Path], suffix: Optional[str] = None, active_loading: Optional[bool] = None
     ) -> None:
         """Initialize the template manager.
 
@@ -54,6 +56,7 @@ class TemplateManager:
             RuntimeError: If template rendering fails
         """
 
+
 def blake3_hash(content: bytes) -> str:
     """Calculate the BLAKE3 cryptographic hash of data.
 
@@ -64,6 +67,7 @@ def blake3_hash(content: bytes) -> str:
         Hex-encoded BLAKE3 hash string
     """
 
+
 class BibManager:
     """BibTeX bibliography manager for parsing and querying citation data."""
 
@@ -100,3 +104,13 @@ class BibManager:
             Uses nucleo_matcher for high-quality fuzzy text searching
             See: https://crates.io/crates/nucleo-matcher
         """
+
+    def list_titles(self, is_verbatim: Optional[bool] = False) -> List[str]:
+        """List all titles in the bibliography.
+
+        Args:
+            is_verbatim: Whether to return verbatim titles (without formatting)
+
+        Returns:
+            List of all titles in the bibliography
+        """

fabricatio-0.2.6.dev3/src/bib_tools.rs

@@ -0,0 +1,88 @@
+use biblatex::{Bibliography, ChunksExt};
+use nucleo_matcher::pattern::{AtomKind, CaseMatching, Normalization, Pattern};
+use nucleo_matcher::{Config, Matcher, Utf32Str};
+use pyo3::exceptions::PyRuntimeError;
+use pyo3::prelude::*;
+use rayon::prelude::*;
+
+#[pyclass]
+pub struct BibManager {
+    source: Bibliography,
+}
+
+
+#[pymethods]
+impl BibManager {
+    /// Create a new BibManager instance.
+    #[new]
+    fn new(path: String) -> PyResult<Self> {
+        let bib = std::fs::read_to_string(path)
+            .map_err(|e| PyErr::new::<PyRuntimeError, _>(e.to_string()))?;
+
+        let source = Bibliography::parse(&bib)
+            .map_err(|e| PyErr::new::<PyRuntimeError, _>(e.to_string()))?;
+
+        Ok(BibManager { source })
+    }
+
+    /// find the cite key of an article with given title
+    fn get_cite_key(&self, title: String) -> Option<String> {
+        let title_lower = title.to_lowercase();
+
+        self.source.iter().par_bridge()
+            .find_map_any(|entry| {
+                let entry_title = entry.title()
+                    .map_err(|e| PyErr::new::<PyRuntimeError, _>(format!("{}", e)))
+                    .ok()?
+                    .to_biblatex_string(false)
+                    .replace("{", "")
+                    .replace("}", "")
+                    .to_lowercase();
+
+                (entry_title == title_lower).then(|| entry.key.clone())
+            })
+    }
+
+    /// Find the corresponding cite key of an article with given query string using fuzzy matcher
+    fn get_cite_key_fuzzy(&self, query: String) -> Option<String> {
+        let mut matcher = Matcher::new(Config::DEFAULT);
+        let pattern = Pattern::new(
+            query.as_str(),
+            CaseMatching::Ignore,
+            Normalization::Smart,
+            AtomKind::Substring,
+        );
+
+
+        self.source.iter()
+            .map(|entry| {
+                let mut buf = vec![];
+                let text = entry.to_biblatex_string().replace("{", "")
+                    .replace("}", "");
+                (pattern.score(Utf32Str::new(text.as_str(), &mut buf), &mut matcher), entry)
+            })
+            .par_bridge()
+            // Use filter_map's more concise form with pattern matching
+            .filter_map(|(maybe_score, entry)| maybe_score.map(|score| (score, entry)))
+            .max_by_key(|(score, _)| *score)
+            .map(|(_, entry)| entry.key.clone())
+    }
+    #[pyo3(signature = (is_verbatim=false))]
+    fn list_titles(&self, is_verbatim: bool) -> Vec<String> {
+        self.source.iter().map(|entry| {
+            entry.title()
+                .map_err(|e| PyErr::new::<PyRuntimeError, _>(format!("{}", e)))
+                .ok()
+                .unwrap()
+                .to_biblatex_string(is_verbatim)
+                .replace("{", "")
+                .replace("}", "")
+        }).collect::<Vec<_>>()
+    }
+}
+
+
+pub(crate) fn register(_: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_class::<BibManager>()?;
+    Ok(())
+}