Spaces:
Sleeping
Sleeping
updated the openai credentials
Browse files
main.py
CHANGED
|
@@ -5,6 +5,9 @@ from langchain_community.document_loaders.csv_loader import CSVLoader
|
|
| 5 |
from langchain_openai import AzureChatOpenAI
|
| 6 |
from langchain_community.utilities import SQLDatabase
|
| 7 |
from langchain_community.agent_toolkits import create_sql_agent
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
# Streamlit UI
|
| 10 |
def main():
|
|
@@ -47,8 +50,8 @@ def main():
|
|
| 47 |
data = loader.load()
|
| 48 |
|
| 49 |
# Initialize LLM
|
| 50 |
-
api_key = "
|
| 51 |
-
azure_endpoint =
|
| 52 |
api_version = "2023-03-15-preview"
|
| 53 |
|
| 54 |
llm = AzureChatOpenAI(
|
|
|
|
| 5 |
from langchain_openai import AzureChatOpenAI
|
| 6 |
from langchain_community.utilities import SQLDatabase
|
| 7 |
from langchain_community.agent_toolkits import create_sql_agent
|
| 8 |
+
import os
|
| 9 |
+
from dotenv import load_dotenv
|
| 10 |
+
load_dotenv()
|
| 11 |
|
| 12 |
# Streamlit UI
|
| 13 |
def main():
|
|
|
|
| 50 |
data = loader.load()
|
| 51 |
|
| 52 |
# Initialize LLM
|
| 53 |
+
api_key = os.environ["api_key"]
|
| 54 |
+
azure_endpoint = os.environ["azure_endpoint"]
|
| 55 |
api_version = "2023-03-15-preview"
|
| 56 |
|
| 57 |
llm = AzureChatOpenAI(
|