fabricatio 0.2.5.dev0__tar.gz → 0.2.5.dev1__tar.gz

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/Cargo.lock

@@ -112,6 +112,19 @@ version = "0.22.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "72b3254f16251a8381aa12e40e3c4d2f0199f8c6508fbecb9d91f575e0fbb8c6"
 
+[[package]]
+name = "biblatex"
+version = "0.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a35a7317fcbdbef94b60d0dd0a658711a936accfce4a631fea4bf8e527eff3c2"
+dependencies = [
+ "numerals",
+ "paste",
+ "strum",
+ "unicode-normalization",
+ "unscanny",
+]
+
 [[package]]
 name = "bitflags"
 version = "2.8.0"
@@ -254,6 +267,31 @@ dependencies = [
  "cfg-if",
 ]
 
+[[package]]
+name = "crossbeam-deque"
+version = "0.8.6"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9dd111b7b7f7d55b72c0a6ae361660ee5853c9af73f70c3c2ef6858b950e2e51"
+dependencies = [
+ "crossbeam-epoch",
+ "crossbeam-utils",
+]
+
+[[package]]
+name = "crossbeam-epoch"
+version = "0.9.18"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5b82ac4a3c2ca9c3460964f020e1402edd5753411d7737aa39c3714ad1b5420e"
+dependencies = [
+ "crossbeam-utils",
+]
+
+[[package]]
+name = "crossbeam-utils"
+version = "0.8.21"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d0a5c400df2834b80a4c3327b3aad3a4c4cd4de0629063962b03235697506a28"
+
 [[package]]
 name = "crypto-common"
 version = "0.1.6"
@@ -372,6 +410,12 @@ dependencies = [
  "syn",
 ]
 
+[[package]]
+name = "either"
+version = "1.15.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "48c757948c5ede0e46177b7add2e67155f70e33c07fea8284df6576da70b3719"
+
 [[package]]
 name = "encoding_rs"
 version = "0.8.35"
@@ -401,10 +445,12 @@ dependencies = [
 name = "fabricatio"
 version = "0.0.0"
 dependencies = [
+ "biblatex",
  "blake3",
  "handlebars",
  "pyo3",
  "pythonize",
+ "rayon",
  "serde_json",
  "walkdir",
 ]
@@ -1021,6 +1067,12 @@ dependencies = [
  "num-modular",
 ]
 
+[[package]]
+name = "numerals"
+version = "0.1.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e25be21376a772d15f97ae789845340a9651d3c4246ff5ebb6a2b35f9c37bd31"
+
 [[package]]
 name = "object"
 version = "0.36.7"
@@ -1086,6 +1138,12 @@ version = "0.2.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "04744f49eae99ab78e0d5c0b603ab218f515ea8cfe5a456d7629ad883a3b6e7d"
 
+[[package]]
+name = "paste"
+version = "1.0.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
+
 [[package]]
 name = "percent-encoding"
 version = "2.3.1"
@@ -1252,6 +1310,26 @@ dependencies = [
  "proc-macro2",
 ]
 
+[[package]]
+name = "rayon"
+version = "1.10.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b418a60154510ca1a002a752ca9714984e21e4241e804d32555251faf8b78ffa"
+dependencies = [
+ "either",
+ "rayon-core",
+]
+
+[[package]]
+name = "rayon-core"
+version = "1.12.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1465873a3dfdaa8ae7cb14b4383657caab0b3e8a0aa9ae8e04b044854c8dfce2"
+dependencies = [
+ "crossbeam-deque",
+ "crossbeam-utils",
+]
+
 [[package]]
 name = "redox_syscall"
 version = "0.5.8"
@@ -1540,6 +1618,28 @@ version = "0.11.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "7da8b5736845d9f2fcb837ea5d9e2628564b3b043a70948a3f0b778838c5fb4f"
 
+[[package]]
+name = "strum"
+version = "0.26.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8fec0f0aef304996cf250b31b5a10dee7980c85da9d759361292b8bca5a18f06"
+dependencies = [
+ "strum_macros",
+]
+
+[[package]]
+name = "strum_macros"
+version = "0.26.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4c6bee85a5a24955dc440386795aa378cd9cf82acd5f764469152d2270e581be"
+dependencies = [
+ "heck",
+ "proc-macro2",
+ "quote",
+ "rustversion",
+ "syn",
+]
+
 [[package]]
 name = "subtle"
 version = "2.6.1"
@@ -1659,6 +1759,21 @@ dependencies = [
  "zerovec",
 ]
 
+[[package]]
+name = "tinyvec"
+version = "1.9.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "09b3661f17e86524eccd4371ab0429194e0d7c008abb45f7a7495b1719463c71"
+dependencies = [
+ "tinyvec_macros",
+]
+
+[[package]]
+name = "tinyvec_macros"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1f3ccbac311fea05f86f61904b462b55fb3df8837a366dfc601a0161d0532f20"
+
 [[package]]
 name = "tokio"
 version = "1.43.0"
@@ -1789,12 +1904,27 @@ version = "1.0.16"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "a210d160f08b701c8721ba1c726c11662f877ea6b7094007e1ca9a1041945034"
 
+[[package]]
+name = "unicode-normalization"
+version = "0.1.24"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5033c97c4262335cded6d6fc3e5c18ab755e1a3dc96376350f3d8e9f009ad956"
+dependencies = [
+ "tinyvec",
+]
+
 [[package]]
 name = "unindent"
 version = "0.2.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "c7de7d73e1754487cb58364ee906a499937a0dfabd86bcb980fa99ec8c8fa2ce"
 
+[[package]]
+name = "unscanny"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e9df2af067a7953e9c3831320f35c1cc0600c30d44d9f7a12b01db1cd88d6b47"
+
 [[package]]
 name = "untrusted"
 version = "0.9.0"

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/Cargo.toml

@@ -6,10 +6,12 @@ edition = "2024"
 crate-type = ["cdylib"]
 
 [dependencies]
+biblatex = "0.10.0"
 blake3 = "1.6.0"
 handlebars = "6.3.1"
 pyo3 = { version = "0.23.4", features = ["extension-module"] }
 pythonize = "0.23.0"
+rayon = "1.10.0"
 serde_json = "1.0.138"
 walkdir = "2.5.0"
 

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fabricatio
-Version: 0.2.5.dev0
+Version: 0.2.5.dev1
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Rust
 Classifier: Programming Language :: Python :: 3.12

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fabricatio"
-version = "0.2.5-dev0"
+version = "0.2.5-dev1"
 description = "A LLM multi-agent framework."
 readme = "README.md"
 license = { file = "LICENSE" }

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/python/fabricatio/__init__.py

@@ -2,6 +2,7 @@
 
 from importlib.util import find_spec
 
+from fabricatio._rust import BibManager
 from fabricatio._rust_instances import template_manager
 from fabricatio.actions.article import ExtractArticleEssence, GenerateArticleProposal, GenerateOutline
 from fabricatio.actions.output import DumpFinalizedOutput
@@ -22,6 +23,7 @@ from fabricatio.workflows.articles import WriteOutlineWorkFlow
 __all__ = [
     "Action",
     "ArticleEssence",
+    "BibManager",
     "Capture",
     "CodeBlockCapture",
     "DumpFinalizedOutput",

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/python/fabricatio/_rust.pyi

@@ -51,3 +51,11 @@ def blake3_hash(content: bytes) -> str:
     Returns:
         str: The BLAKE3 hash of the data.
     """
+
+class BibManager:
+    """BibManager class for managing BibTeX files."""
+    def __init__(self, path: str) -> None:
+        """Initialize the BibManager with the given path."""
+
+    def get_cite_key(self, title: str) -> str:
+        """Get the cite key for the given title."""

fabricatio-0.2.5.dev1/src/bib_tools.rs

@@ -0,0 +1,45 @@
+use biblatex::{Bibliography, ChunksExt};
+use pyo3::exceptions::PyRuntimeError;
+use pyo3::prelude::*;
+use rayon::prelude::*;
+use std::fs::File;
+use std::io::Read;
+
+#[pyclass]
+pub struct BibManager {
+    source:Bibliography,
+}
+
+
+#[pymethods]
+impl BibManager {
+    /// Create a new BibManager instance.
+    #[new]
+    fn new(path:String) -> PyResult<Self> {
+        let mut bib=String::new();
+
+        File::open(path)?.read_to_string(&mut bib).map_err(|e| PyErr::new::<PyRuntimeError,_>(format!("{}", e)))?;
+        Ok(BibManager {
+            source: Bibliography::parse(bib.as_str()).map_err(|e| PyErr::new::<PyRuntimeError,_>(format!("{}", e)))?,
+        })
+    }
+
+
+
+    /// find the cite key of an article with given title
+    fn get_cite_key(&self,title:String)->Option<String>{
+        self.source.iter().par_bridge()
+            .find_map_any(|entry|{
+                if entry.title().map_err(|e| PyErr::new::<PyRuntimeError,_>(format!("{}", e))).ok()?.to_biblatex_string(false).to_lowercase()==title.to_lowercase(){
+                    return Some(entry.key.clone())
+                }
+                None
+            })
+    }
+}
+
+
+pub(crate) fn register(_: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
+    m.add_class::<BibManager>()?;
+    Ok(())
+}

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/src/lib.rs

@@ -1,6 +1,7 @@
 mod templates;
 mod hash;
 mod hbs_helpers;
+mod bib_tools;
 
 use pyo3::prelude::*;
 
@@ -14,7 +15,7 @@ fn _rust(python: Python, m: &Bound<'_, PyModule>) -> PyResult<()> {
     // Add the TemplateManager class to the Python module
     templates::register(python, m)?;
     hash::register(python, m)?;
-
+    bib_tools::register(python, m)?;
     Ok(())
 }
 

{fabricatio-0.2.5.dev0 → fabricatio-0.2.5.dev1}/uv.lock

@@ -222,7 +222,7 @@ wheels = [
 
 [[package]]
 name = "fabricatio"
-version = "0.2.5.dev0"
+version = "0.2.5.dev1"
 source = { editable = "." }
 dependencies = [
     { name = "appdirs" },