Update config env.toml

h2oai · Feb 1, 2024 · 26904e1 · 26904e1
1 parent 5357df8
commit 26904e1
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/sidekick/configs/env.toml b/sidekick/configs/env.toml
@@ -2,10 +2,13 @@
 OPENAI_API_KEY = "" # Needed only for openAI models
 MODEL_NAME = "h2ogpt-sql-sqlcoder-34b-alpha" # Others: e.g. gpt-4, gpt-4-32k, text-davinci-003
 QUANT_TYPE = '4bit'
-H2O_BASE_MODEL_URL = 'http://38.128.233.247'
-H2O_BASE_MODEL_API_KEY = ""
-RECOMMENDATION_MODEL_REMOTE_URL = ""
-RECOMMENDATION_MODEL_API_KEY = ""
+
+H2OGPT_URL = 'http://38.128.233.247'
+H2OGPT_API_TOKEN = ""
+H2OGPTE_URL = ""
+H2OGPTE_API_TOKEN = ""
+
+RECOMMENDATION_MODEL = "h2oai/h2ogpt-4096-llama2-70b-chat"
 VULNERABILITY_SCANNER = "h2oai/h2ogpt-4096-llama2-70b-chat" # other options openai models depending on availability (e.g. 'gpt-3.5-turbo')
 SELF_CORRECTION_MODEL = "h2oai/h2ogpt-4096-llama2-70b-chat" # other options openai models depending on availability (e.g. 'gpt-3.5-turbo')