jcnok commited on
Commit
f9c0f15
·
verified ·
1 Parent(s): 40d476c
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -4,12 +4,12 @@ import pandas as pd
4
  #from langchain.llms import OpenAI
5
  #New import from langchain, which replaces the above
6
  from langchain_openai import OpenAI
7
-
8
  def query_agent(data, query):
9
 
10
  # Parse the CSV file and create a Pandas DataFrame from its contents.
11
  df = pd.read_csv(data)
12
-
13
  llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
14
 
15
  # Create a Pandas DataFrame agent.
 
4
  #from langchain.llms import OpenAI
5
  #New import from langchain, which replaces the above
6
  from langchain_openai import OpenAI
7
+ import os
8
  def query_agent(data, query):
9
 
10
  # Parse the CSV file and create a Pandas DataFrame from its contents.
11
  df = pd.read_csv(data)
12
+ os.environ["OPENAI_API_KEY"]
13
  llm = ChatOpenAI(model="gpt-3.5-turbo", temperature=0)
14
 
15
  # Create a Pandas DataFrame agent.