Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,13 +28,13 @@ print(pyodbc.drivers())
|
|
| 28 |
|
| 29 |
st.title("Generate responses from SQL tables")
|
| 30 |
|
| 31 |
-
mapping = {'
|
| 32 |
'HISTORY_AVAVBAIL':'AV availability', \
|
| 33 |
'HISTORY_BUFamilyAvailability': 'Family and Business Unit (BU) availability', \
|
| 34 |
-
'
|
| 35 |
'MasterSkuAvBom_PA': 'Business unit to SKU to AV Mapping', \
|
| 36 |
'SMF_WT_BASE': 'Forecast and Orders', \
|
| 37 |
-
'
|
| 38 |
'HISTORY_Sku_Shortage': 'SKU Shortage'}
|
| 39 |
inv_mapping = {val: key for (key,val) in mapping.items()}
|
| 40 |
|
|
@@ -96,7 +96,7 @@ if __name__ == '__main__':
|
|
| 96 |
query={"odbc_connect": connection_string}
|
| 97 |
)
|
| 98 |
engine = create_engine(connection_url)
|
| 99 |
-
db = SQLDatabase(engine=engine, sample_rows_in_table_info=3)
|
| 100 |
|
| 101 |
prompt = PromptTemplate(template=template, input_variables=["dialect","input","table_info","top_k"])
|
| 102 |
llm = AzureChatOpenAI(
|
|
|
|
| 28 |
|
| 29 |
st.title("Generate responses from SQL tables")
|
| 30 |
|
| 31 |
+
mapping = {'History_All_Skus_Availability': 'SKU Availability', \
|
| 32 |
'HISTORY_AVAVBAIL':'AV availability', \
|
| 33 |
'HISTORY_BUFamilyAvailability': 'Family and Business Unit (BU) availability', \
|
| 34 |
+
'HISTORY_OpenOrderShortage': 'Part Shortage', \
|
| 35 |
'MasterSkuAvBom_PA': 'Business unit to SKU to AV Mapping', \
|
| 36 |
'SMF_WT_BASE': 'Forecast and Orders', \
|
| 37 |
+
'DAILY_INVENTORY': 'Part Inventory', \
|
| 38 |
'HISTORY_Sku_Shortage': 'SKU Shortage'}
|
| 39 |
inv_mapping = {val: key for (key,val) in mapping.items()}
|
| 40 |
|
|
|
|
| 96 |
query={"odbc_connect": connection_string}
|
| 97 |
)
|
| 98 |
engine = create_engine(connection_url)
|
| 99 |
+
db = SQLDatabase(engine=engine, sample_rows_in_table_info=3, view_support=True)
|
| 100 |
|
| 101 |
prompt = PromptTemplate(template=template, input_variables=["dialect","input","table_info","top_k"])
|
| 102 |
llm = AzureChatOpenAI(
|