From 3d71a44f52fb10bf364bfa8d9b321a606a48cbc5 Mon Sep 17 00:00:00 2001 From: drbh Date: Tue, 20 Feb 2024 13:15:20 -0500 Subject: [PATCH] fix: add back imports after rebase --- router/src/server.rs | 2 ++ 1 file changed, 2 insertions(+) diff --git a/router/src/server.rs b/router/src/server.rs index df0bc1e7..db11f87f 100644 --- a/router/src/server.rs +++ b/router/src/server.rs @@ -2,6 +2,7 @@ use crate::health::Health; use crate::infer::{InferError, InferResponse, InferStreamResponse}; use crate::validation::ValidationError; +// TODO: improve readability of crate imports use crate::{ BestOfSequence, ChatCompletion, ChatCompletionChoice, ChatCompletionChunk, ChatCompletionDelta, ChatCompletionLogprobs, ChatRequest, CompatGenerateRequest, Completion, CompletionComplete, @@ -10,6 +11,7 @@ use crate::{ Info, Message, PrefillToken, SimpleToken, StreamDetails, StreamResponse, Token, TokenizeResponse, Usage, Validation, }; +use crate::{GrammarType, VertexRequest, VertexResponse}; use axum::extract::Extension; use axum::http::{HeaderMap, Method, StatusCode}; use axum::response::sse::{Event, KeepAlive, Sse};