Spaces:
Sleeping
Sleeping
muzammil-eds
commited on
Commit
•
38c54c8
1
Parent(s):
06e5177
Update app.py
Browse files
app.py
CHANGED
@@ -2,25 +2,21 @@ import os
|
|
2 |
import streamlit as st
|
3 |
import yfinance as yf
|
4 |
import pandas as pd
|
5 |
-
from langchain.agents import
|
|
|
6 |
from langchain.chat_models import ChatOpenAI
|
7 |
from htmlTemplates import css, user_template, bot_template
|
8 |
-
from langchain.llms.base import LLM
|
9 |
-
from typing import Optional, List
|
10 |
-
import g4f
|
11 |
|
|
|
12 |
os.environ['OPENAI_API_KEY'] = os.getenv('OPENAI_API_KEY')
|
13 |
|
14 |
-
|
15 |
-
|
16 |
llm = ChatOpenAI(
|
17 |
model='gpt-3.5-turbo',
|
18 |
max_tokens=500,
|
19 |
temperature=0.7,
|
20 |
)
|
21 |
|
22 |
-
|
23 |
-
|
24 |
def init_ses_states():
|
25 |
st.session_state.setdefault('chat_history', [])
|
26 |
|
@@ -82,9 +78,9 @@ def main():
|
|
82 |
try:
|
83 |
DF = pd.DataFrame(df)
|
84 |
DF.to_csv('data.csv')
|
85 |
-
agent =
|
86 |
llm,
|
87 |
-
|
88 |
verbose=True,
|
89 |
agent_type=AgentType.OPENAI_FUNCTIONS,
|
90 |
)
|
|
|
2 |
import streamlit as st
|
3 |
import yfinance as yf
|
4 |
import pandas as pd
|
5 |
+
from langchain.agents import AgentType
|
6 |
+
from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent
|
7 |
from langchain.chat_models import ChatOpenAI
|
8 |
from htmlTemplates import css, user_template, bot_template
|
|
|
|
|
|
|
9 |
|
10 |
+
# Set OpenAI API Key
|
11 |
os.environ['OPENAI_API_KEY'] = os.getenv('OPENAI_API_KEY')
|
12 |
|
13 |
+
# Initialize LangChain ChatOpenAI agent
|
|
|
14 |
llm = ChatOpenAI(
|
15 |
model='gpt-3.5-turbo',
|
16 |
max_tokens=500,
|
17 |
temperature=0.7,
|
18 |
)
|
19 |
|
|
|
|
|
20 |
def init_ses_states():
|
21 |
st.session_state.setdefault('chat_history', [])
|
22 |
|
|
|
78 |
try:
|
79 |
DF = pd.DataFrame(df)
|
80 |
DF.to_csv('data.csv')
|
81 |
+
agent = create_pandas_dataframe_agent(
|
82 |
llm,
|
83 |
+
DF,
|
84 |
verbose=True,
|
85 |
agent_type=AgentType.OPENAI_FUNCTIONS,
|
86 |
)
|