diff --git a/openhands/sdk/llm/message.py b/openhands/sdk/llm/message.py index 0c5b3bf2b6..a29f5b7518 100644 --- a/openhands/sdk/llm/message.py +++ b/openhands/sdk/llm/message.py @@ -1,7 +1,6 @@ from collections.abc import Sequence from typing import Any, Literal, cast -import mcp.types from litellm import ChatCompletionMessageToolCall from litellm.types.utils import Message as LiteLLMMessage from pydantic import BaseModel, ConfigDict, Field, field_validator @@ -23,7 +22,7 @@ def to_llm_dict( raise NotImplementedError("Subclasses should implement this method.") -class TextContent(mcp.types.TextContent, BaseContent): +class TextContent(BaseContent): type: Literal["text"] = "text" text: str # We use populate_by_name since mcp.types.TextContent