secure the api calling for the phonemizer service

This commit is contained in:
matthieu 2025-01-30 19:23:03 +04:00
parent 3ba26023d0
commit f14f90c141
3 changed files with 12 additions and 12 deletions

View file

@ -4,7 +4,6 @@ use autofeur::save::Save;
use kdam::tqdm;
#[tokio::main]
/// Generates the DB file foe easy usage.
async fn main() {
let mut save = Save::default();

View file

@ -1,4 +1,7 @@
use std::{env, ops::Add};
use std::{
env::{self, VarError},
ops::Add,
};
use anyhow::anyhow;
use hypher::hyphenate;
@ -7,14 +10,13 @@ use itertools::Itertools;
use crate::save::Save;
async fn call_inference_service(word: &str) -> anyhow::Result<String> {
let server: Result<String, anyhow::Error> =
env::var("PHONEMIZER").or_else(|_| Ok("http://localhost:8000/".to_string()));
Ok(
reqwest::get(format!("{}?grapheme={}", server.unwrap(), word))
.await?
.text()
.await?,
)
let server: String = env::var("PHONEMIZER")
.or_else(|_| Ok::<String, VarError>("http://localhost:8000/".to_string()))
.unwrap();
let url = reqwest::Url::parse_with_params(&server, &[("grapheme", word)])?;
Ok(reqwest::get(url).await?.text().await?)
}
impl Save<'_> {

View file

@ -13,7 +13,6 @@ pub struct TrieNode<'a> {
}
impl<'a> TrieNode<'a> {
// Create new node
pub fn new<'b>(is_final: bool) -> TrieNode<'b> {
TrieNode {
is_final,