| import os |
| from langchain import OpenAI |
| from langchain.chat_models import ChatOpenAI |
|
|
|
|
| def get_sql_index_tool(sql_index, table_context_dict): |
| table_context_str = "\n".join(table_context_dict.values()) |
| def run_sql_index_query(query_text): |
| try: |
| response = sql_index.query(query_text) |
| except Exception as e: |
| return f"Error running SQL {e}.\nNot able to retrieve answer." |
| text = str(response) |
| sql = response.extra_info['sql_query'] |
| return f"Here are the details on the SQL table: {table_context_str}\nSQL Query Used: {sql}\nSQL Result: {text}\n" |
| |
| return run_sql_index_query |
|
|
|
|
|
|
| def get_llm(llm_name, model_temperature, api_key): |
| os.environ['OPENAI_API_KEY'] = api_key |
| if llm_name == "text-davinci-003": |
| return OpenAI(temperature=model_temperature, model_name=llm_name) |
| else: |
| return ChatOpenAI(temperature=model_temperature, model_name=llm_name) |
