tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

added x-client header in the nanogpt api call (#8368)

Browse Source
This commit is contained in:
Lifei Zhou
2026-04-07 22:51:03 +10:00
committed by GitHub
parent 7449a96664
commit c4aa7f56df
2 changed files with 30 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/providers/nanogpt.rs
+10 -5
View File
@@ -29,11 +29,16 @@ pub struct NanoGptProvider {
}
impl NanoGptProvider {
async fn check_subscription(api_key: &str) -> bool {
let client = match ApiClient::new(
NANOGPT_SUBSCRIPTION_HOST.to_string(),
fn build_client(host: &str, api_key: &str) -> Result<ApiClient> {
ApiClient::new(
host.to_string(),
AuthMethod::BearerToken(api_key.to_string()),
) {
)?
.with_header("x-client", "goose")
}
async fn check_subscription(api_key: &str) -> bool {
let client = match Self::build_client(NANOGPT_SUBSCRIPTION_HOST, api_key) {
Ok(c) => c,
Err(_) => return false,
};
@@ -62,7 +67,7 @@ impl NanoGptProvider {
NANOGPT_API_HOST.to_string()
};
let api_client = ApiClient::new(host, AuthMethod::BearerToken(api_key))?;
let api_client = Self::build_client(&host, &api_key)?;
Ok(Self {
api_client,