viz updates (#138)

pnnl · Jun 18, 2024 · 791006a · 791006a
1 parent 4cd1212
commit 791006a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/viz/backend/requirements.txt b/viz/backend/requirements.txt
diff --git a/viz/backend/sqlchain.py b/viz/backend/sqlchain.py
@@ -1,4 +1,5 @@
-from langchain import OpenAI, SQLDatabase, SQLDatabaseChain
+from langchain import SQLDatabase, SQLDatabaseChain
+from langchain.chat_models import ChatOpenAI
 import config
 from langchain.output_parsers import CommaSeparatedListOutputParser
 from langchain.prompts import PromptTemplate, ChatPromptTemplate, HumanMessagePromptTemplate
@@ -7,7 +8,7 @@
 
 
 def sqlchain(input_text):
-    llm = OpenAI(openai_api_key=config.openai_key,
+    llm = ChatOpenAI(openai_api_key=config.openai_key,
                  model_name="gpt-3.5-turbo", temperature=0, verbose=True)
     db = SQLDatabase.from_uri(
         f"postgresql+psycopg2://postgres:{config.sql_key}@localhost:5432/{config.database_name}")

diff --git a/viz/data/jsontocsv.py b/viz/data/jsontocsv.py
@@ -115,7 +115,6 @@ def getBus(data):
                 "kilovolt levels": set(p["KVlevels"]),
                 "number of buses": p["nbus"],
                 "vm": p["Vm"],
-                "start": p["start"],
             })
 
     keys = points[0].keys()