From 79c7d6c42a4b4e5db7b2c1a8645aef41d6841f42 Mon Sep 17 00:00:00 2001
From: Kok-Meng Tan <kesamet@gmail.com>
Date: Sun, 13 Oct 2024 13:27:16 +0800
Subject: [PATCH] Hotfix - update base.py

---
 gpt_researcher/config/variables/base.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/gpt_researcher/config/variables/base.py b/gpt_researcher/config/variables/base.py
index 26bda1e4b..8e62a2da6 100644
--- a/gpt_researcher/config/variables/base.py
+++ b/gpt_researcher/config/variables/base.py
@@ -6,11 +6,9 @@ class BaseConfig(TypedDict):
     RETRIEVER: str
     EMBEDDING_PROVIDER: str
     SIMILARITY_THRESHOLD: float
-    LLM_PROVIDER: str
     OLLAMA_BASE_URL: Union[str, None]
-    DEFAULT_LLM_MODEL: str
-    FAST_LLM_MODEL: str
-    SMART_LLM_MODEL: str
+    FAST_LLM: str
+    SMART_LLM: str
     FAST_TOKEN_LIMIT: int
     SMART_TOKEN_LIMIT: int
     BROWSE_CHUNK_MAX_LENGTH: int