huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 12:24:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

use Draft202012

Browse Source
This commit is contained in:
OlivierDehaene 2024-02-19 17:33:23 +01:00
parent 2251f5b16f
commit 0533e67ea6
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
router/Cargo.toml
View File

@ -22,7 +22,7 @@ text-generation-client = { path = "client" }
clap = { version = "4.4.5", features = ["derive", "env"] } clap = { version = "4.4.5", features = ["derive", "env"] }
futures = "0.3.28" futures = "0.3.28"
hf-hub = { version = "0.3.0", features = ["tokio"] } hf-hub = { version = "0.3.0", features = ["tokio"] }
jsonschema = "0.17.1" jsonschema = { version = "0.17.1", features = ["draft202012"] }
metrics = "0.21.1" metrics = "0.21.1"
metrics-exporter-prometheus = { version = "0.12.1", features = [] } metrics-exporter-prometheus = { version = "0.12.1", features = [] }
nohash-hasher = "0.2.0" nohash-hasher = "0.2.0"

3
router/src/lib.rs
View File

@ -46,7 +46,7 @@ impl HubTokenizerConfig {
} }
mod json_object_or_string_to_string { mod json_object_or_string_to_string {
use jsonschema::JSONSchema; use jsonschema::{Draft, JSONSchema};
use serde::{Deserialize, Deserializer}; use serde::{Deserialize, Deserializer};
use serde_json::Value; use serde_json::Value;
@ -59,6 +59,7 @@ mod json_object_or_string_to_string {
let value = Value::deserialize(deserializer)?; let value = Value::deserialize(deserializer)?;
JSONSchema::options() JSONSchema::options()
.with_draft(Draft::Draft202012)
.compile(&value) .compile(&value)
.map_err(|e| serde::de::Error::custom(format!("invalid JSONSchema: {e}")))?; .map_err(|e| serde::de::Error::custom(format!("invalid JSONSchema: {e}")))?;