From e67dcdeea8f8f8af59f7c9218985d9dd644ea5df Mon Sep 17 00:00:00 2001
From: mohammadrezapourreza <m1378.prz@gmail.com>
Date: Mon, 6 May 2024 15:47:04 -0400
Subject: [PATCH] DH-5775/removing the LLM Model from env vars

---
 .env.example        | 2 --
 .test.env           | 1 -
 README.md           | 4 ----
 docs/envars.rst     | 2 --
 docs/quickstart.rst | 1 -
 5 files changed, 10 deletions(-)

diff --git a/.env.example b/.env.example
index 237f0f06..c6b6ebad 100644
--- a/.env.example
+++ b/.env.example
@@ -1,8 +1,6 @@
 # Openai info. All these fields are required for the engine to work.
 OPENAI_API_KEY = #This field is required for the engine to work.
 ORG_ID =
-# what is the LLM model to be used
-LLM_MODEL = "gpt-4-turbo-preview"
 
 # All of our SQL generation agents are using different tools to generate SQL queries, in order to limit the number of times that agents can
 # use different tools you can set the "AGENT_MAX_ITERATIONS" env variable. By default it is set to 20 iterations.
diff --git a/.test.env b/.test.env
index b9bb479b..93411646 100644
--- a/.test.env
+++ b/.test.env
@@ -2,7 +2,6 @@ SQL_GENERATOR='dataherald.tests.sql_generator.test_generator.TestGenerator'
 EVALUATOR='dataherald.tests.evaluator.test_eval.TestEvaluator'
 DB='dataherald.tests.db.test_db.TestDB'
 OPENAI_API_KEY='foo'
-LLM_MODEL='gpt_test'
 PINECONE_API_KEY='foo2'
 PINECONE_ENVIRONMENT='bar'
 GOLDEN_RECORD_COLLECTION='bar2'
diff --git a/README.md b/README.md
index c2a4b768..4287d802 100644
--- a/README.md
+++ b/README.md
@@ -68,14 +68,10 @@ cp .env.example .env
 
 Specifically the following fields must be manually set before the engine is started.
 
-LLM_MODEL is employed by the engine to generate SQL from natural language. You can use the default model (gpt-4-turbo-preview) or use your own deployed model.
-
 ```
 #OpenAI credentials and model 
 # mainly used for embedding models and finetunung 
 OPENAI_API_KEY = 
-# Used for the reasoning LLM or the main LLM which chooses the tools to generate SQL
-LLM_MODEL = 
 ORG_ID =
 
 #Encryption key for storing DB connection data in Mongo
diff --git a/docs/envars.rst b/docs/envars.rst
index d15252b2..6152d159 100644
--- a/docs/envars.rst
+++ b/docs/envars.rst
@@ -8,7 +8,6 @@ provided in the .env.example file with the default values.
 
     OPENAI_API_KEY = 
     ORG_ID =
-    LLM_MODEL = 'gpt-4-turbo-preview'
 
 
     GOLDEN_RECORD_COLLECTION = 'my-golden-records'
@@ -51,7 +50,6 @@ provided in the .env.example file with the default values.
 
    "OPENAI_API_KEY", "The OpenAI key used by the Dataherald Engine", "None", "Yes"
    "ORG_ID", "The OpenAI Organization ID used by the Dataherald Engine", "None", "Yes"
-   "LLM_MODEL", "The Language Model used by the Dataherald Engine. Supported values include gpt-4-32k, gpt-4, gpt-3.5-turbo, gpt-3.5-turbo-16k", "``gpt-4-32k``", "No"
    "GOLDEN_RECORD_COLLECTION", "The name of the collection in Mongo where golden records will be stored", "``my-golden-records``", "No"
    "PINECONE_API_KEY", "The Pinecone API key used", "None", "Yes if using the Pinecone vector store"
    "PINECONE_ENVIRONMENT", "The Pinecone environment", "None", "Yes if using the Pinecone vector store"
diff --git a/docs/quickstart.rst b/docs/quickstart.rst
index 5d7ec363..886a3f6b 100644
--- a/docs/quickstart.rst
+++ b/docs/quickstart.rst
@@ -17,7 +17,6 @@ The following environment variables must be set manually before the engine is st
 
     #OpenAI credentials and model 
     OPENAI_API_KEY = 
-    LLM_MODEL =  
     ORG_ID =
 
     #Encryption key for storing DB connection data in Mongo