Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix build with `--features google` (#2566) 

* Fix `cargo build --features google`

* Add `cargo test --features google`
This commit is contained in:
Alvaro Bartolome 2024-09-26 11:41:38 +02:00 committed by GitHub
parent 0b7df77178
commit 0aa66d693a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/tests.yaml vendored
View File

@ -58,3 +58,6 @@ jobs:
- name: Run Rust tests - name: Run Rust tests
run: | run: |
cargo test cargo test
- name: Run Rust tests with google feature
run: |
cargo test --features google

3
router/src/server.rs
View File

@ -2029,7 +2029,8 @@ async fn start(
#[cfg(feature = "google")] #[cfg(feature = "google")]
{ {
use crate::VertexInstance; use crate::vertex::__path_vertex_compatibility;
use crate::vertex::{VertexInstance, VertexRequest, VertexResponse};
#[derive(OpenApi)] #[derive(OpenApi)]
#[openapi( #[openapi(