From eb7c712aa6c4e20421a7318943be5ccc974d864b Mon Sep 17 00:00:00 2001
From: Sakshi Srivastava
 <140403864+SakshiSrivastava2024@users.noreply.github.com>
Date: Fri, 4 Jul 2025 03:07:20 +0530
Subject: [PATCH] Fix: Add missing OpenAI import in vLLM module (#3091)

---
 mem0/llms/vllm.py | 1 +
 1 file changed, 1 insertion(+)

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):