gered/llm-ls
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

feat: add support for tls_skip_verify_insecure (#15)

Browse source
This commit is contained in:
Luc Georges 2023-09-15 15:42:35 +02:00 committed by GitHub
parent 9508485fec
commit c869f96060
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
crates/llm-ls/src/main.rs
View file

@ -107,6 +107,7 @@ struct Backend {
client: Client,
document_map: Arc<RwLock<HashMap<String, Document>>>,
http_client: reqwest::Client,
unsafe_http_client: reqwest::Client,
workspace_folders: Arc<RwLock<Option<Vec<WorkspaceFolder>>>>,
tokenizer_map: Arc<RwLock<HashMap<String, Arc<Tokenizer>>>>,
}
@ -127,6 +128,7 @@ struct CompletionParams {
model_eos: String,
tokenizer_path: Option<String>,
context_window: usize,
tls_skip_verify_insecure: bool,
}
fn internal_error<E: Display>(err: E) -> Error {
@ -362,8 +364,15 @@ impl Backend {
tokenizer,
params.context_window,
)?;
let http_client = if params.tls_skip_verify_insecure {
info!("tls verification is disabled");
&self.unsafe_http_client
} else {
&self.http_client
};
let result = request_completion(
&self.http_client,
http_client,
&params.model,
params.request_params,
params.api_token.as_ref(),
@ -483,12 +492,17 @@ async fn main() {
.init();
let http_client = reqwest::Client::new();
let unsafe_http_client = reqwest::Client::builder()
.danger_accept_invalid_certs(true)
.build()
.expect("failed to build reqwest unsafe client");
let (service, socket) = LspService::build(|client| Backend {
cache_dir,
client,
document_map: Arc::new(RwLock::new(HashMap::new())),
http_client,
unsafe_http_client,
workspace_folders: Arc::new(RwLock::new(None)),
tokenizer_map: Arc::new(RwLock::new(HashMap::new())),
})