File size: 1,467 Bytes
66b6353 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
import requests
from langchain_community.utilities import SQLDatabase
from langchain_community.tools.sql_database.tool import ListSQLDatabaseTool, InfoSQLDatabaseTool
def get_all_groq_model(api_key:str=None) -> list:
if api_key is None:
raise ValueError("API key is required")
url = "https://api.groq.com/openai/v1/models"
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json"
}
response = requests.get(url, headers=headers)
data = response.json()['data']
model_ids = [model['id'] for model in data]
return model_ids
def validate_api_key(api_key:str) -> bool:
if len(api_key) == 0:
return False
try:
get_all_groq_model(api_key=api_key)
return True
except Exception as e:
return False
def validate_uri(uri:str) -> bool:
try:
SQLDatabase.from_uri(uri)
return True
except Exception as e:
return False
def get_info(uri:str) -> dict[str, str] | None:
db = SQLDatabase.from_uri(uri)
dialect = db.dialect
# List all the tables accessible to the user.
access_tables = ListSQLDatabaseTool(db=db).invoke("")
# List the table schemas of all the accessible tables.
tables_schemas = InfoSQLDatabaseTool(db=db).invoke(access_tables)
return {'sql_dialect': dialect, 'tables': access_tables, 'tables_schema': tables_schemas}
if __name__ == "__main__":
print(get_all_groq_model()) |