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

fix: improve import readability

Browse Source
This commit is contained in:
drbh 2024-02-20 13:21:35 -05:00
parent 3d71a44f52
commit 10621a30be
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
router/src/server.rs
View File

@ -2,16 +2,17 @@
use crate::health::Health; use crate::health::Health;
use crate::infer::{InferError, InferResponse, InferStreamResponse}; use crate::infer::{InferError, InferResponse, InferStreamResponse};
use crate::validation::ValidationError; use crate::validation::ValidationError;
// TODO: improve readability of crate imports
use crate::{ use crate::{
BestOfSequence, ChatCompletion, ChatCompletionChoice, ChatCompletionChunk, ChatCompletionDelta, BestOfSequence, Details, ErrorResponse, FinishReason, GenerateParameters, GenerateRequest,
ChatCompletionLogprobs, ChatRequest, CompatGenerateRequest, Completion, CompletionComplete, GenerateResponse, GrammarType, HubModelInfo, HubTokenizerConfig, Infer, Info, Message,
CompletionCompleteChunk, CompletionRequest, Details, ErrorResponse, FinishReason, PrefillToken, SimpleToken, StreamDetails, StreamResponse, Token, TokenizeResponse, Usage,
GenerateParameters, GenerateRequest, GenerateResponse, HubModelInfo, HubTokenizerConfig, Infer, Validation,
Info, Message, PrefillToken, SimpleToken, StreamDetails, StreamResponse, Token, };
TokenizeResponse, Usage, Validation, use crate::{
ChatCompletion, ChatCompletionChoice, ChatCompletionChunk, ChatCompletionDelta,
ChatCompletionLogprobs, ChatRequest, CompatGenerateRequest, Completion, CompletionComplete,
CompletionCompleteChunk, CompletionRequest, VertexRequest, VertexResponse,
}; };
use crate::{GrammarType, VertexRequest, VertexResponse};
use axum::extract::Extension; use axum::extract::Extension;
use axum::http::{HeaderMap, Method, StatusCode}; use axum::http::{HeaderMap, Method, StatusCode};
use axum::response::sse::{Event, KeepAlive, Sse}; use axum::response::sse::{Event, KeepAlive, Sse};