use heed3::RoTxn; use get_routes::handler; use helix_db::{field_remapping, identifier_remapping, traversal_remapping, exclude_field}; use helix_db::helix_engine::vector_core::vector::HVector; use helix_db::{ helix_engine::graph_core::ops::{ g::G, in_::{in_::InAdapter, in_e::InEdgesAdapter, to_n::ToNAdapter}, out::{from_n::FromNAdapter, out::OutAdapter, out_e::OutEdgesAdapter}, source::{ add_e::{AddEAdapter, EdgeType}, add_n::AddNAdapter, e_from_id::EFromIdAdapter, e_from_type::EFromTypeAdapter, n_from_id::NFromIdAdapter, n_from_type::NFromTypeAdapter, n_from_index::NFromIndexAdapter, }, tr_val::{Traversable, TraversalVal}, util::{ dedup::DedupAdapter, filter_mut::FilterMut, filter_ref::FilterRefAdapter, range::RangeAdapter, update::UpdateAdapter, map::MapAdapter, paths::ShortestPathAdapter, props::PropsAdapter, drop::Drop, }, vectors::{insert::InsertVAdapter, search::SearchVAdapter}, bm25::search_bm25::SearchBM25Adapter, }, helix_engine::types::GraphError, helix_gateway::router::router::HandlerInput, node_matches, props, protocol::count::Count, protocol::remapping::ResponseRemapping, protocol::response::Response, protocol::traversal_value::TraversalValue, protocol::{ filterable::Filterable, remapping::Remapping, return_values::ReturnValue, value::Value, id::ID, }, }; use sonic_rs::{Deserialize, Serialize}; use std::collections::{HashMap, HashSet}; use std::sync::Arc; use std::time::Instant; use std::cell::RefCell; use chrono::{DateTime, Utc}; pub struct File14 { pub name: String, pub age: i32, } #[handler] pub fn file14 (input: &HandlerInput, response: &mut Response) -> Result<(), GraphError> { let mut remapping_vals: RefCell> = RefCell::new(HashMap::new()); let db = Arc::clone(&input.graph.storage); let txn = db.graph_env.read_txn().unwrap(); let res = G::new(Arc::clone(&db), &txn) .search_bm25("File14", "John", 10).collect_to::>(); let mut return_vals: HashMap = HashMap::new(); return_vals.insert("res".to_string(), ReturnValue::from_traversal_value_array_with_mixin(res.clone(), remapping_vals.borrow_mut())); txn.commit().unwrap(); response.body = sonic_rs::to_vec(&return_vals).unwrap(); Ok(()) }