diff --git a/litellm/llms/digitalocean/chat/transformation.py b/litellm/llms/digitalocean/chat/transformation.py
index 3073e220a92..9f9b63f4620 100644
--- a/litellm/llms/digitalocean/chat/transformation.py
+++ b/litellm/llms/digitalocean/chat/transformation.py
@@ -1,15 +1,8 @@
-"""
-Translate from OpenAI's `/v1/chat/completions` to DigitalOcean AI Agent's `/v1/chat/completions`
-"""
-
 from typing import List, Optional, Tuple, Union, Dict
 
-from pydantic import BaseModel
-
 from litellm.secret_managers.main import get_secret_str
 from litellm.types.llms.openai import (
     AllMessageValues,
-    ChatCompletionAssistantMessage,
     ChatCompletionToolParam,
     ChatCompletionToolParamFunctionChunk,
 )