from sqlalchemy import create_engine
import pandas as pd
"""
[Link]("data",exist_ok=True)
!wget [Link] -P ./data/
file_url = "./data/[Link]"
"""
df = pd.read_csv("./data/[Link]").fillna(value = 0)
database_file_path = "./db/[Link]"
engine = create_engine(f'sqlite:///{database_file_path}')
df.to_sql('all_states_history', con=engine, if_exists='replace', index=False)
tools_sql = [
{
"type": "function",
"function": {
"name": "get_hospitalized_increase_for_state_on_date",
"description": "Retrieves the daily increase in hospitalizations for a
specific state on a specific date.",
"parameters": {
"type": "object",
"properties": {
"state_abbr": {
"type": "string",
"description": "The abbreviation of the state (e.g., 'NY',
'CA')."
},
"specific_date": {
"type": "string",
"description": "The specific date for the query in 'YYYY-
MM-DD' format."
}
},
"required": ["state_abbr", "specific_date"]
}
}
},
{
"type": "function",
"function": {
"name": "get_positive_cases_for_state_on_date",
"description": "Retrieves the daily increase in positive cases for a
specific state on a specific date.",
"parameters": {
"type": "object",
"properties": {
"state_abbr": {
"type": "string",
"description": "The abbreviation of the state (e.g., 'NY',
'CA')."
},
"specific_date": {
"type": "string",
"description": "The specific date for the query in 'YYYY-
MM-DD' format."
}
},
"required": ["state_abbr", "specific_date"]
}
}
}
]
# define functions
import numpy as np
def get_hospitalized_increase_for_state_on_date(state_abbr, specific_date):
try:
query = f"""
SELECT date, hospitalizedIncrease
FROM all_states_history
WHERE state = '{state_abbr}' AND date = '{specific_date}';
"""
with [Link]() as connection:
result = pd.read_sql_query(query, connection)
if not [Link]:
return result.to_dict('records')[0]
else:
return [Link]
except Exception as e:
print(e)
return [Link]
def get_positive_cases_for_state_on_date(state_abbr, specific_date):
try:
query = f"""
SELECT date, state, positiveIncrease AS positive_cases
FROM all_states_history
WHERE state = '{state_abbr}' AND date = '{specific_date}';
"""
with [Link]() as connection:
result = pd.read_sql_query(query, connection)
if not [Link]:
return result.to_dict('records')[0]
else:
return [Link]
except Exception as e:
print(e)
return [Link]