From 86043ce3af6b25a0c7bc56e152d2ac536f113b3d Mon Sep 17 00:00:00 2001
From: Luc Georges <luc@huggingface.co>
Date: Mon, 12 Feb 2024 15:58:56 +0100
Subject: [PATCH] fix: change visiblity of internal functions

---
 crates/llm-ls/src/backend.rs | 10 +++++++---
 crates/llm-ls/src/error.rs   |  4 ++--
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/crates/llm-ls/src/backend.rs b/crates/llm-ls/src/backend.rs
index 357a3d5..da15312 100644
--- a/crates/llm-ls/src/backend.rs
+++ b/crates/llm-ls/src/backend.rs
@@ -151,7 +151,7 @@ fn parse_openai_text(text: &str) -> Result<Vec<Generation>> {
     }
 }
 
-pub fn build_body(
+pub(crate) fn build_body(
     backend: &Backend,
     model: String,
     prompt: String,
@@ -170,7 +170,11 @@ pub fn build_body(
     request_body
 }
 
-pub fn build_headers(backend: &Backend, api_token: Option<&String>, ide: Ide) -> Result<HeaderMap> {
+pub(crate) fn build_headers(
+    backend: &Backend,
+    api_token: Option<&String>,
+    ide: Ide,
+) -> Result<HeaderMap> {
     match backend {
         Backend::HuggingFace { .. } => build_api_headers(api_token, ide),
         Backend::Ollama { .. } => Ok(build_ollama_headers()),
@@ -179,7 +183,7 @@ pub fn build_headers(backend: &Backend, api_token: Option<&String>, ide: Ide) ->
     }
 }
 
-pub fn parse_generations(backend: &Backend, text: &str) -> Result<Vec<Generation>> {
+pub(crate) fn parse_generations(backend: &Backend, text: &str) -> Result<Vec<Generation>> {
     match backend {
         Backend::HuggingFace { .. } => parse_api_text(text),
         Backend::Ollama { .. } => parse_ollama_text(text),
diff --git a/crates/llm-ls/src/error.rs b/crates/llm-ls/src/error.rs
index aa51588..d6f305e 100644
--- a/crates/llm-ls/src/error.rs
+++ b/crates/llm-ls/src/error.rs
@@ -3,7 +3,7 @@ use std::fmt::Display;
 use tower_lsp::jsonrpc::Error as LspError;
 use tracing::error;
 
-pub fn internal_error<E: Display>(err: E) -> LspError {
+pub(crate) fn internal_error<E: Display>(err: E) -> LspError {
     let err_msg = err.to_string();
     error!(err_msg);
     LspError {
@@ -55,7 +55,7 @@ pub enum Error {
     UnknownBackend(String),
 }
 
-pub type Result<T> = std::result::Result<T, Error>;
+pub(crate) type Result<T> = std::result::Result<T, Error>;
 
 impl From<Error> for LspError {
     fn from(err: Error) -> Self {