Merge pull request #1 from DashyDashOrg/v0.0.4

V0.0.4

Author: alessiodashydash

README.md

@@ -23,9 +23,6 @@ import os
 import pandas as pd
 from pandas_llm import PandasLLM
 
-# your key
-OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
-
 # Data
 # Please note that these names, ages, and donations are randomly generated 
 # and do not correspond to real individuals or their donations.
@@ -41,10 +38,12 @@ data = [('John Doe', 25, 50),
         ('Olivia Jackson', 29, 55)]
 df = pd.DataFrame(data, columns=['name', 'age', 'donation'])
 
-conv_df = PandasLLM(data=df, config={ "openai_api_key":OPENAI_API_KEY})
+conv_df = PandasLLM(data=df, openai_api_key = os.environ.get("OPENAI_API_KEY"))
 result = conv_df.prompt("What is the average donation of people older than 30 who donated more than $50?")
 
 print(f"Result ({type(result)}):\n {result}")
+# Result (<class 'numpy.float64'>):
+#  75.0
 ```
 
 ## Contributing

pandas_llm/example-chatbot.py

@@ -8,6 +8,7 @@
 from pandas_llm import PandasLLM
 
 # Data
+#  Please note that these names, ages, and donations are randomly generated and do not correspond to real individuals or their donations.
 data = [('John Doe', 25, 50), 
         ('Jane Smith', 38, 70),
         ('Alex Johnson', 45, 80),
@@ -54,19 +55,15 @@ def main():
         print("No OpenAI API key provided. Exiting.")
         return
 
-    config = {
-        "openai_api_key":openai_key
-    }
-
-    conv_df = PandasLLM(data=df, config=config)
+    conv_df = PandasLLM(data=df, llm_api_key = openai_key)
     print()
     banner = """
     Welcome to the Donation Data CLI.
     The donation dataset has three columns (name, age, donation)
     Please note that these names, ages, and donations are randomly generated and do not correspond to real individuals or their donations.
     
     You can ask questions like:
-    - show me the name of donators
+    - show me the name of donors
     - What is the average age of people who donated?
     - What is the average donation amount?
     - What is the average donation of people older than 30?

pandas_llm/example.py

@@ -2,9 +2,6 @@
 import pandas as pd
 from pandas_llm import PandasLLM
 
-# your key
-OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
-
 # Data
 # Please note that these names, ages, and donations are randomly generated 
 # and do not correspond to real individuals or their donations.
@@ -20,8 +17,9 @@
         ('Olivia Jackson', 29, 55)]
 df = pd.DataFrame(data, columns=['name', 'age', 'donation'])
 
-conv_df = PandasLLM(data=df, config={ "openai_api_key":OPENAI_API_KEY})
+conv_df = PandasLLM(data=df, llm_api_key = os.environ.get("OPENAI_API_KEY"))
 result = conv_df.prompt("What is the average donation of people older than 30 who donated more than $50?")
 
 print(f"Result ({type(result)}):\n {result}")
-
+# Result (<class 'numpy.float64'>):
+#  75.0