File size: 467 Bytes
cdb9e0a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import pandas as pd
from pandasai import SmartDataframe
from langchain_groq.chat_models import ChatGroq
from langchain_community.llms import Ollama
import sqlite3
import os
llm = ChatGroq(model_name="llama3-70b-8192", api_key = os.environ["GROQ_API_KEY"])
df = pd.read_excel('data.xlsx')
df = SmartDataframe(df, config={"llm": llm})
print( df.chat('Which are the 5 happiest countries?'))
print( df.chat('What is the sum of the GDPs of the 2 happiest countries?')) |