victor7246 commited on
Commit
2008452
·
verified ·
1 Parent(s): ad1ef8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -28,13 +28,13 @@ print(pyodbc.drivers())
28
 
29
  st.title("Generate responses from SQL tables")
30
 
31
- mapping = {'TBL_History_All_Skus_Availability': 'SKU Availability', \
32
  'HISTORY_AVAVBAIL':'AV availability', \
33
  'HISTORY_BUFamilyAvailability': 'Family and Business Unit (BU) availability', \
34
- 'TBL_HISTORY_OpenOrderShortage': 'Part Shortage', \
35
  'MasterSkuAvBom_PA': 'Business unit to SKU to AV Mapping', \
36
  'SMF_WT_BASE': 'Forecast and Orders', \
37
- 'TBL_DAILY_INVENTORY': 'Part Inventory', \
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(