diff --git a/cli/utils.py b/cli/utils.py
index 5256b4e6..b873da21 100644
--- a/cli/utils.py
+++ b/cli/utils.py
@@ -153,10 +153,7 @@ def select_shallow_thinking_agent(provider) -> str:
             ("llama3.1 local", "llama3.1"),
             ("llama3.2 local", "llama3.2"),
         ],
-        "vllm": [
-            ("llama3.1 local", "llama3.1"),
-            ("qwen3", "qwen3"),
-        ]
+        "vllm": [],
     }
 
     if provider == "vllm":
@@ -234,10 +231,7 @@ def select_deep_thinking_agent(provider) -> str:
             ("llama3.1 local", "llama3.1"),
             ("qwen3", "qwen3"),
         ],
-        "vllm": [
-            ("llama3.1 local", "llama3.1"),
-            ("qwen3", "qwen3"),
-        ]
+        "vllm": [],
     }
 
     if provider == "vllm":
diff --git a/tradingagents/agents/utils/memory.py b/tradingagents/agents/utils/memory.py
index d34bcd44..b3eec3a3 100644
--- a/tradingagents/agents/utils/memory.py
+++ b/tradingagents/agents/utils/memory.py
@@ -1,7 +1,6 @@
 import chromadb
 from chromadb.config import Settings
 from openai import OpenAI
-import questionary
 
 
 class FinancialSituationMemory:
diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py
index 26c8d760..6f1f8cf1 100644
--- a/tradingagents/graph/trading_graph.py
+++ b/tradingagents/graph/trading_graph.py
@@ -88,7 +88,7 @@ class TradingAgentsGraph:
         
         # Initialize memories
         if self.config["llm_provider"] == "vllm":
-            questionary.text(
+            self.config["embeddings"] = questionary.text(
                 "Please input the vllm embedding model name (default: None):",
                 default="None",
                 validate=lambda x: len(x.strip()) > 0 or "Please enter a valid embedding model name.",