Skip to content

Commit

Permalink
patch url parser
Browse files Browse the repository at this point in the history
  • Loading branch information
ChuckHend committed Jan 25, 2024
1 parent 51d95ea commit ff54e3b
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion Cargo.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[package]
name = "vectorize"
version = "0.8.0"
version = "0.8.1"
edition = "2021"
publish = false

Expand Down
4 changes: 2 additions & 2 deletions src/transformers/http_handler.rs
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ use crate::transformers::types::{
};
use pgrx::prelude::*;

use super::generic::get_generic_svc_url;
use super::types::TransformerMetadata;

pub async fn handle_response<T: for<'de> serde::Deserialize<'de>>(
Expand Down Expand Up @@ -85,8 +86,7 @@ pub fn sync_get_model_info(model_name: &str) -> Result<TransformerMetadata> {
}

pub async fn get_model_info(model_name: &str) -> Result<TransformerMetadata> {
let svc_url = guc::get_guc(guc::VectorizeGuc::EmbeddingServiceUrl)
.expect("vectorize.embedding_service_url must be set to a valid service");
let svc_url = get_generic_svc_url()?;
let info_url = svc_url.replace("/embeddings", "/info");
let client = reqwest::Client::new();
let req = client.get(info_url).query(&[("model_name", model_name)]);
Expand Down

0 comments on commit ff54e3b

Please sign in to comment.