def generate_sql(question, schema): # Format input as: "System: [message]\n\nUser: [question + schema]" system_message = "You are a SQL expert. Convert natural language questions into SQL queries." user_message = f"### Question: \n{question}\n\n### Schema: \n{schema}" full_input = f"System: {system_message}\n\nUser: {user_message}" inputs = tokenizer(full_input, return_tensors="pt").to(model.device) outputs = model.generate( input_ids=inputs.input_ids, max_new_tokens=100, temperature=0.2, do_sample=True, use_cache=True ) return tokenizer.decode(outputs[0], skip_special_tokens=True) # Example usagequestion = "Find the total revenue from orders placed in the last 30 days."schema = "CREATE TABLE orders (order_id INT, order_date DATE, order_total DECIMAL, customer_id INT);"sql_query = generate_sql(question, schema)print(sql_query)