summaryrefslogtreecommitdiffstats
path: root/src/client/common.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/common.rs')
-rw-r--r--src/client/common.rs30
1 files changed, 4 insertions, 26 deletions
diff --git a/src/client/common.rs b/src/client/common.rs
index 265ab0a..f25560d 100644
--- a/src/client/common.rs
+++ b/src/client/common.rs
@@ -4,10 +4,7 @@ use crate::{
config::{GlobalConfig, Input},
function::{eval_tool_calls, FunctionDeclaration, ToolCall, ToolResult},
render::{render_error, render_stream},
- utils::{
- prompt_input_integer, prompt_input_string, tokenize, watch_abort_signal, AbortSignal,
- PromptKind,
- },
+ utils::*,
};
use anyhow::{bail, Context, Result};
@@ -15,10 +12,10 @@ use async_trait::async_trait;
use fancy_regex::Regex;
use indexmap::IndexMap;
use lazy_static::lazy_static;
-use reqwest::{Client as ReqwestClient, ClientBuilder, Proxy, RequestBuilder};
+use reqwest::{Client as ReqwestClient, RequestBuilder};
use serde::Deserialize;
use serde_json::{json, Value};
-use std::{env, future::Future, time::Duration};
+use std::{future::Future, time::Duration};
use tokio::sync::mpsc::unbounded_channel;
const MODELS_YAML: &str = include_str!("../../models.yaml");
@@ -340,7 +337,7 @@ pub trait Client: Sync + Send {
let extra = self.extra_config();
let timeout = extra.and_then(|v| v.connect_timeout).unwrap_or(10);
let proxy = extra.and_then(|v| v.proxy.clone());
- builder = set_proxy(builder, &proxy)?;
+ builder = set_proxy(builder, proxy.as_ref())?;
let client = builder
.connect_timeout(Duration::from_secs(timeout))
.build()
@@ -771,22 +768,3 @@ fn to_json(kind: &PromptKind, value: &str) -> Value {
},
}
}
-
-fn set_proxy(builder: ClientBuilder, proxy: &Option<String>) -> Result<ClientBuilder> {
- let proxy = if let Some(proxy) = proxy {
- if proxy.is_empty() || proxy == "-" {
- return Ok(builder);
- }
- proxy.clone()
- } else if let Some(proxy) = ["HTTPS_PROXY", "https_proxy", "ALL_PROXY", "all_proxy"]
- .into_iter()
- .find_map(|v| env::var(v).ok())
- {
- proxy
- } else {
- return Ok(builder);
- };
- let builder =
- builder.proxy(Proxy::all(&proxy).with_context(|| format!("Invalid proxy `{proxy}`"))?);
- Ok(builder)
-}