diff --git a/mem0/llms/vllm.py b/mem0/llms/vllm.py
index 1f0fc822..8ff51fce 100644
--- a/mem0/llms/vllm.py
+++ b/mem0/llms/vllm.py
@@ -5,6 +5,7 @@ from typing import Dict, List, Optional
 from mem0.configs.llms.base import BaseLlmConfig
 from mem0.llms.base import LLMBase
 from mem0.memory.utils import extract_json
+from openai import OpenAI
 
 
 class VllmLLM(LLMBase):