binaychandra commited on
Commit
47d3d15
Β·
1 Parent(s): adf2969

Modified openai keys and files

Browse files
Files changed (3) hide show
  1. app.py +1 -1
  2. fakedatagenerator.ipynb +2 -2
  3. langchain_helper.py +12 -6
app.py CHANGED
@@ -32,7 +32,7 @@ st.markdown("""
32
  """, unsafe_allow_html=True)
33
 
34
  # Title section
35
- st.markdown("<h1 class='title'>Manufacturing Process Analysis</h1>", unsafe_allow_html=True)
36
 
37
  with st.chat_message("assistant"):
38
  st.write("Hello πŸ‘‹ How can I help you today?")
 
32
  """, unsafe_allow_html=True)
33
 
34
  # Title section
35
+ st.markdown("<h1 class='title'>E-Commerce Analysis</h1>", unsafe_allow_html=True)
36
 
37
  with st.chat_message("assistant"):
38
  st.write("Hello πŸ‘‹ How can I help you today?")
fakedatagenerator.ipynb CHANGED
@@ -658,7 +658,7 @@
658
  ],
659
  "metadata": {
660
  "kernelspec": {
661
- "display_name": ".venv",
662
  "language": "python",
663
  "name": "python3"
664
  },
@@ -672,7 +672,7 @@
672
  "name": "python",
673
  "nbconvert_exporter": "python",
674
  "pygments_lexer": "ipython3",
675
- "version": "3.9.13"
676
  }
677
  },
678
  "nbformat": 4,
 
658
  ],
659
  "metadata": {
660
  "kernelspec": {
661
+ "display_name": "langenv",
662
  "language": "python",
663
  "name": "python3"
664
  },
 
672
  "name": "python",
673
  "nbconvert_exporter": "python",
674
  "pygments_lexer": "ipython3",
675
+ "version": "3.9.17"
676
  }
677
  },
678
  "nbformat": 4,
langchain_helper.py CHANGED
@@ -1,12 +1,12 @@
1
  import os
2
- from langchain_openai import AzureOpenAI
3
  from langchain_core.prompts import ChatPromptTemplate
4
  from langchain.agents.agent_types import AgentType
5
  from langchain_experimental.agents import create_pandas_dataframe_agent
6
  from langchain_community.utilities import SQLDatabase
7
  from langchain_experimental.sql import SQLDatabaseChain
8
  from langchain.prompts import SemanticSimilarityExampleSelector
9
- from langchain_openai import AzureOpenAIEmbeddings
10
  from langchain_community.vectorstores import Chroma
11
  from langchain.prompts import FewShotPromptTemplate
12
  from langchain.prompts.prompt import PromptTemplate
@@ -19,16 +19,22 @@ import plotly
19
  import plotly.express as px
20
  from plotly.express import bar, line, scatter, area, pie
21
 
22
- from dotenv import load_dotenv
23
- load_dotenv()
 
 
 
 
 
24
 
25
  def get_few_shot_db_chain(user_message):
26
- llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct", temperature=0.2)
 
27
 
28
  engine = create_engine("sqlite:///ecomm.db")
29
  db = SQLDatabase(engine=engine, sample_rows_in_table_info=3)
30
 
31
- embeddings = AzureOpenAIEmbeddings(model="text-embedding-3-small")
32
 
33
  to_vectorize = [" ".join(example.values()) for example in few_shots]
34
 
 
1
  import os
2
+ from langchain_openai import AzureOpenAI, OpenAI
3
  from langchain_core.prompts import ChatPromptTemplate
4
  from langchain.agents.agent_types import AgentType
5
  from langchain_experimental.agents import create_pandas_dataframe_agent
6
  from langchain_community.utilities import SQLDatabase
7
  from langchain_experimental.sql import SQLDatabaseChain
8
  from langchain.prompts import SemanticSimilarityExampleSelector
9
+ from langchain_openai import AzureOpenAIEmbeddings, OpenAIEmbeddings
10
  from langchain_community.vectorstores import Chroma
11
  from langchain.prompts import FewShotPromptTemplate
12
  from langchain.prompts.prompt import PromptTemplate
 
19
  import plotly.express as px
20
  from plotly.express import bar, line, scatter, area, pie
21
 
22
+ # from dotenv import load_dotenv
23
+ # load_dotenv()
24
+
25
+ from dotenv import load_dotenv, find_dotenv
26
+ _ = load_dotenv(find_dotenv()) # read local .env file
27
+
28
+ current_model_id = os.getenv('model_id')
29
 
30
  def get_few_shot_db_chain(user_message):
31
+ #llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct", temperature=0.2)
32
+ llm = OpenAI(model = current_model_id)
33
 
34
  engine = create_engine("sqlite:///ecomm.db")
35
  db = SQLDatabase(engine=engine, sample_rows_in_table_info=3)
36
 
37
+ embeddings = OpenAIEmbeddings(model="text-embedding-3-small")
38
 
39
  to_vectorize = [" ".join(example.values()) for example in few_shots]
40