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 File7 { pub name: String, pub age: i32, } pub struct EdgeFile7 { pub from: File7, pub to: File7, } pub struct File7Vec { pub content: String, } #[derive(Serialize, Deserialize)] pub struct file7Input { pub vec: Vec } #[handler] pub fn file7 (input: &HandlerInput, response: &mut Response) -> Result<(), GraphError> { let data: file7Input = 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 txn = db.graph_env.read_txn().unwrap(); let vecs = G::new(Arc::clone(&db), &txn) .search_v:: bool>(&data.vec, 10, None).collect_to::>(); let mut return_vals: HashMap = HashMap::new(); return_vals.insert("hello".to_string(), ReturnValue::from(Value::from("hello"))); txn.commit().unwrap(); response.body = sonic_rs::to_vec(&return_vals).unwrap(); Ok(()) }