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 File8 { pub name: String, pub age: i32, } pub struct EdgeFile8 { pub from: File8, pub to: File8, } pub struct File8Vec { pub content: String, } #[derive(Serialize, Deserialize)] pub struct file8Input { pub vec: Vec } #[handler] pub fn file8 (input: &HandlerInput, response: &mut Response) -> Result<(), GraphError> { let data: file8Input = match sonic_rs::from_slice(&input.request.body) { Ok(data) => data, Err(err) => return Err(GraphError::from(err)), }; let mut remapping_vals: RefCell> = RefCell::new(HashMap::new()); let db = Arc::clone(&input.graph.storage); let mut txn = db.graph_env.write_txn().unwrap(); let new_vec = G::new_mut(Arc::clone(&db), &mut txn) .insert_v:: bool>(&data.vec, "File8Vec", None).collect_to::>(); G::new_mut(Arc::clone(&db), &mut txn) .insert_v:: bool>(&data.vec, "File8Vec", None).collect_to::>(); let mut return_vals: HashMap = HashMap::new(); return_vals.insert("new_vec".to_string(), ReturnValue::from_traversal_value_array_with_mixin(new_vec.clone(), remapping_vals.borrow_mut())); txn.commit().unwrap(); response.body = sonic_rs::to_vec(&return_vals).unwrap(); Ok(()) }