From 2c496e6376d8d13f63a7e4451aecbf3a9d63ff94 Mon Sep 17 00:00:00 2001
From: Mingxiangyu <49153313+Mingxiangyu@users.noreply.github.com>
Date: Fri, 4 Jul 2025 05:41:10 +0800
Subject: [PATCH] Fix the error that occurs when VLLM is called (#3076)

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

diff --git a/mem0/llms/vllm.py b/mem0/llms/vllm.py
index 8ff51fce..bf66d4aa 100644
--- a/mem0/llms/vllm.py
+++ b/mem0/llms/vllm.py
@@ -1,6 +1,8 @@
 import json
 import os
 from typing import Dict, List, Optional
+from openai import OpenAI
+
 
 from mem0.configs.llms.base import BaseLlmConfig
 from mem0.llms.base import LLMBase