Make text embedding title optional
- Make text embedding title optional, as sending it causes erros on tasks that don't support title, like CLASSIFICATION. - Update dependencies
This commit is contained in:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -5,3 +5,4 @@ Cargo.lock
|
||||
*.pdb
|
||||
/target
|
||||
.cargo/config.toml
|
||||
.DS_Store
|
||||
|
||||
16
Cargo.toml
16
Cargo.toml
@@ -8,17 +8,17 @@ edition = "2021"
|
||||
[dependencies]
|
||||
deadqueue = "0.2.4"
|
||||
futures-util = "0.3.30"
|
||||
gcp_auth = "0.10.0"
|
||||
reqwest = { version = "0.11.9", features = ["json", "gzip"] }
|
||||
reqwest-eventsource = "0.5.0"
|
||||
serde = { version = "*", features = ["derive"] }
|
||||
serde_json = { version = "*"}
|
||||
gcp_auth = "0.11.1"
|
||||
reqwest = { version = "0.12.3", features = ["json", "gzip"] }
|
||||
reqwest-eventsource = "0.6.0"
|
||||
serde = { version = "1.0.197", features = ["derive"] }
|
||||
serde_json = { version = "1.0.115"}
|
||||
tracing = "0.1.40"
|
||||
tokio = { version = "1.36.0" }
|
||||
tokio = { version = "1.37.0" }
|
||||
|
||||
[dev-dependencies]
|
||||
console = "0.15.8"
|
||||
dialoguer = "0.11.0"
|
||||
indicatif = "0.17.7"
|
||||
tokio = { version = "1.35.1", features = ["full"] }
|
||||
indicatif = "0.17.8"
|
||||
tokio = { version = "1.37.0", features = ["full"] }
|
||||
tracing-subscriber = "0.3.18"
|
||||
|
||||
@@ -21,12 +21,12 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
|
||||
let embedding_request = TextEmbeddingRequest {
|
||||
instances: vec![
|
||||
TextEmbeddingRequestInstance {
|
||||
title: String::from("Embed testing"),
|
||||
title: Some(String::from("Embed testing")),
|
||||
content: String::from("Embed testing"),
|
||||
task_type: String::from("RETRIEVAL_DOCUMENT"),
|
||||
},
|
||||
TextEmbeddingRequestInstance {
|
||||
title: String::from("Embed testing 2"),
|
||||
title: Some(String::from("Embed testing 2")),
|
||||
content: String::from("Embed testing 2"),
|
||||
task_type: String::from("RETRIEVAL_DOCUMENT"),
|
||||
},
|
||||
|
||||
@@ -11,7 +11,7 @@ pub struct TextEmbeddingRequest {
|
||||
pub struct TextEmbeddingRequestInstance {
|
||||
pub content: String,
|
||||
pub task_type: String,
|
||||
pub title: String,
|
||||
pub title: Option<String>,
|
||||
}
|
||||
|
||||
#[derive(Clone, Debug, Serialize, Deserialize)]
|
||||
|
||||
Reference in New Issue
Block a user