diff --git a/DocSum/docsum.py b/DocSum/docsum.py
index 9685adff2..52e328b78 100644
--- a/DocSum/docsum.py
+++ b/DocSum/docsum.py
@@ -15,9 +15,9 @@
     ChatCompletionResponse,
     ChatCompletionResponseChoice,
     ChatMessage,
+    DocSumChatCompletionRequest,
     UsageInfo,
 )
-from comps.cores.proto.api_protocol import DocSumChatCompletionRequest
 from fastapi import File, Request, UploadFile
 from fastapi.responses import StreamingResponse