refactor: update infer initialization in image MIME type detection

Author: lemorage

src/llm/anthropic.rs

@@ -1,6 +1,6 @@
 use crate::llm::{
     LlmGenerateRequest, LlmGenerateResponse, LlmGenerationClient, OutputFormat,
-    ToJsonSchemaOptions, detect_mime_type,
+    ToJsonSchemaOptions, detect_image_mime_type,
 };
 use anyhow::{Context, Result, bail};
 use async_trait::async_trait;
@@ -44,7 +44,7 @@ impl LlmGenerationClient for Client {
         if let Some(image_bytes) = &request.image {
             let base64_image =
                 base64::engine::general_purpose::STANDARD.encode(image_bytes.as_ref());
-            let mime_type = detect_mime_type(image_bytes.as_ref())?;
+            let mime_type = detect_image_mime_type(image_bytes.as_ref())?;
             user_content_parts.push(serde_json::json!({
                 "type": "image",
                 "source": {

src/llm/gemini.rs

@@ -2,7 +2,7 @@ use crate::prelude::*;
 
 use crate::llm::{
     LlmEmbeddingClient, LlmGenerateRequest, LlmGenerateResponse, LlmGenerationClient, OutputFormat,
-    ToJsonSchemaOptions, detect_mime_type,
+    ToJsonSchemaOptions, detect_image_mime_type,
 };
 use base64::Engine;
 use phf::phf_map;
@@ -80,7 +80,7 @@ impl LlmGenerationClient for Client {
         if let Some(image_bytes) = &request.image {
             let base64_image =
                 base64::engine::general_purpose::STANDARD.encode(image_bytes.as_ref());
-            let mime_type = detect_mime_type(image_bytes.as_ref())?;
+            let mime_type = detect_image_mime_type(image_bytes.as_ref())?;
             user_parts.push(serde_json::json!({
                 "inlineData": {
                     "mimeType": mime_type,

src/llm/mod.rs

@@ -5,7 +5,7 @@ use infer::Infer;
 use schemars::schema::SchemaObject;
 use std::borrow::Cow;
 
-static INFER: OnceLock<Infer> = OnceLock::new();
+static INFER: LazyLock<Infer> = LazyLock::new(Infer::new);
 
 #[derive(Debug, Clone, Copy, Serialize, Deserialize)]
 pub enum LlmApiType {
@@ -140,8 +140,8 @@ pub fn new_llm_embedding_client(
     Ok(client)
 }
 
-pub fn detect_mime_type(bytes: &[u8]) -> Result<&'static str> {
-    let infer = INFER.get_or_init(Infer::new);
+pub fn detect_image_mime_type(bytes: &[u8]) -> Result<&'static str> {
+    let infer = &*INFER;
     match infer.get(bytes) {
         Some(info) if info.mime_type().starts_with("image/") => Ok(info.mime_type()),
         _ => bail!("Unknown or unsupported image format"),

src/llm/openai.rs

@@ -1,6 +1,6 @@
 use crate::api_bail;
 
-use super::{LlmEmbeddingClient, LlmGenerationClient, detect_mime_type};
+use super::{LlmEmbeddingClient, LlmGenerationClient, detect_image_mime_type};
 use anyhow::Result;
 use async_openai::{
     Client as OpenAIClient,
@@ -70,7 +70,7 @@ impl LlmGenerationClient for Client {
             Some(img_bytes) => {
                 use base64::{Engine as _, engine::general_purpose::STANDARD};
                 let base64_image = STANDARD.encode(img_bytes.as_ref());
-                let mime_type = detect_mime_type(img_bytes.as_ref())?;
+                let mime_type = detect_image_mime_type(img_bytes.as_ref())?;
                 let image_url = format!("data:{};base64,{}", mime_type, base64_image);
                 ChatCompletionRequestUserMessageContent::Array(vec![
                     ChatCompletionRequestUserMessageContentPart::Text(