diff --git a/caikit_nlp/toolkit/text_generation/tgis_utils.py b/caikit_nlp/toolkit/text_generation/tgis_utils.py
index a8e55c2b..392455c0 100644
--- a/caikit_nlp/toolkit/text_generation/tgis_utils.py
+++ b/caikit_nlp/toolkit/text_generation/tgis_utils.py
@@ -14,10 +14,9 @@
 """This file is for helper functions related to TGIS."""
 
 # Standard
-from typing import Iterable, Optional
+from typing import Iterable
 
 # Third Party
-import fastapi
 import grpc
 
 # First Party
@@ -34,7 +33,6 @@
     TokenizationResults,
     TokenStreamDetails,
 )
-from caikit.interfaces.runtime.data_model import RuntimeServerContextType
 from caikit_tgis_backend import TGISBackend
 from caikit_tgis_backend.protobufs import generation_pb2
 import alog