themissingCRAM
commited on
Commit
·
b1ab131
1
Parent(s):
f1d9f2e
sql
Browse files- Constants.py +169 -0
- __pycache__/Constants.cpython-311.pyc +0 -0
- app.py +28 -166
- sqlite3.sqlite +0 -0
Constants.py
ADDED
@@ -0,0 +1,169 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from datetime import datetime
|
2 |
+
'''
|
3 |
+
ai generated data
|
4 |
+
'''
|
5 |
+
|
6 |
+
bakery_orders_data = [
|
7 |
+
{
|
8 |
+
"order_id": 1,
|
9 |
+
"ingredient_name": "Flour",
|
10 |
+
"supplier_name": "Supplier A",
|
11 |
+
"quantity_ordered": 50.00,
|
12 |
+
"unit": "kg",
|
13 |
+
"price_per_unit": 1.20,
|
14 |
+
"total_cost": 60.00, # This can be calculated as quantity_ordered * price_per_unit
|
15 |
+
"order_date": datetime.strptime("2023-10-01 10:00:00","%Y-%m-%d %H:%M:%S" )
|
16 |
+
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"order_id": 2,
|
20 |
+
"ingredient_name": "Sugar",
|
21 |
+
"supplier_name": "Supplier B",
|
22 |
+
"quantity_ordered": 20.00,
|
23 |
+
"unit": "kg",
|
24 |
+
"price_per_unit": 0.80,
|
25 |
+
"total_cost": 16.00,
|
26 |
+
"order_date": datetime.strptime( "2023-10-02 11:30:00","%Y-%m-%d %H:%M:%S" )
|
27 |
+
},
|
28 |
+
{
|
29 |
+
"order_id": 3,
|
30 |
+
"ingredient_name": "Butter",
|
31 |
+
"supplier_name": "Supplier C",
|
32 |
+
"quantity_ordered": 10.00,
|
33 |
+
"unit": "kg",
|
34 |
+
"price_per_unit": 3.00,
|
35 |
+
"total_cost": 30.00,
|
36 |
+
"order_date": datetime.strptime( "2023-10-03 09:15:00","%Y-%m-%d %H:%M:%S" )
|
37 |
+
},
|
38 |
+
{
|
39 |
+
"order_id": 4,
|
40 |
+
"ingredient_name": "Eggs",
|
41 |
+
"supplier_name": "Supplier D",
|
42 |
+
"quantity_ordered": 5.00,
|
43 |
+
"unit": "dozen",
|
44 |
+
"price_per_unit": 2.50,
|
45 |
+
"total_cost": 12.50,
|
46 |
+
"order_date": datetime.strptime( "2023-10-04 14:45:00","%Y-%m-%d %H:%M:%S" )
|
47 |
+
},
|
48 |
+
{
|
49 |
+
"order_id": 5,
|
50 |
+
"ingredient_name": "Baking Powder",
|
51 |
+
"supplier_name": "Supplier E",
|
52 |
+
"quantity_ordered": 2.00,
|
53 |
+
"unit": "kg",
|
54 |
+
"price_per_unit": 4.00,
|
55 |
+
"total_cost": 8.00,
|
56 |
+
"order_date": datetime.strptime( "2023-10-05 08:30:00","%Y-%m-%d %H:%M:%S" )
|
57 |
+
}
|
58 |
+
]
|
59 |
+
|
60 |
+
baking_recipes = [
|
61 |
+
{
|
62 |
+
"name": "Lemon Blueberry Muffins",
|
63 |
+
"ingredients": [
|
64 |
+
"1 1/2 cups all-purpose flour",
|
65 |
+
"1/2 cup sugar",
|
66 |
+
"2 tsp baking powder",
|
67 |
+
"1/2 tsp salt",
|
68 |
+
"1/2 cup milk",
|
69 |
+
"1/3 cup vegetable oil",
|
70 |
+
"1 large egg",
|
71 |
+
"1 tsp vanilla extract",
|
72 |
+
"1 cup fresh blueberries",
|
73 |
+
"Zest of 1 lemon",
|
74 |
+
"Juice of 1 lemon"
|
75 |
+
],
|
76 |
+
"instructions": [
|
77 |
+
"Preheat the oven to 375°F (190°C) and line a muffin tin with paper liners.",
|
78 |
+
"In a bowl, mix flour, sugar, baking powder, and salt.",
|
79 |
+
"In another bowl, whisk together milk, oil, egg, vanilla, lemon zest, and lemon juice.",
|
80 |
+
"Combine wet and dry ingredients until just mixed, then fold in blueberries.",
|
81 |
+
"Spoon the batter into the muffin tin and bake for 18-20 minutes."
|
82 |
+
]
|
83 |
+
},
|
84 |
+
{
|
85 |
+
"name": "Chocolate Chip Cookies",
|
86 |
+
"ingredients": [
|
87 |
+
"1 cup unsalted butter, softened",
|
88 |
+
"3/4 cup sugar",
|
89 |
+
"3/4 cup brown sugar",
|
90 |
+
"1 tsp vanilla extract",
|
91 |
+
"2 large eggs",
|
92 |
+
"2 1/4 cups all-purpose flour",
|
93 |
+
"1 tsp baking soda",
|
94 |
+
"1/2 tsp salt",
|
95 |
+
"2 cups chocolate chips"
|
96 |
+
],
|
97 |
+
"instructions": [
|
98 |
+
"Preheat the oven to 350°F (175°C).",
|
99 |
+
"In a bowl, cream together butter, sugar, and brown sugar until smooth.",
|
100 |
+
"Beat in vanilla and eggs one at a time.",
|
101 |
+
"In another bowl, combine flour, baking soda, and salt.",
|
102 |
+
"Gradually blend dry ingredients into the wet mixture.",
|
103 |
+
"Stir in chocolate chips.",
|
104 |
+
"Drop by rounded tablespoons onto ungreased baking sheets and bake for 9-11 minutes."
|
105 |
+
]
|
106 |
+
},
|
107 |
+
{
|
108 |
+
"name": "Banana Bread",
|
109 |
+
"ingredients": [
|
110 |
+
"3 ripe bananas, mashed",
|
111 |
+
"1/3 cup melted butter",
|
112 |
+
"1 tsp baking soda",
|
113 |
+
"Pinch of salt",
|
114 |
+
"3/4 cup sugar",
|
115 |
+
"1 large egg, beaten",
|
116 |
+
"1 tsp vanilla extract",
|
117 |
+
"1 1/2 cups all-purpose flour"
|
118 |
+
],
|
119 |
+
"instructions": [
|
120 |
+
"Preheat the oven to 350°F (175°C) and grease a 4x8 inch loaf pan.",
|
121 |
+
"In a mixing bowl, mix the mashed bananas with the melted butter.",
|
122 |
+
"Stir in baking soda and salt.",
|
123 |
+
"Add sugar, egg, and vanilla, and mix well.",
|
124 |
+
"Finally, mix in the flour until just incorporated.",
|
125 |
+
"Pour the batter into the prepared loaf pan and bake for 60-65 minutes."
|
126 |
+
]
|
127 |
+
},
|
128 |
+
{
|
129 |
+
"name": "Cinnamon Rolls",
|
130 |
+
"ingredients": [
|
131 |
+
"4 cups all-purpose flour",
|
132 |
+
"1/4 cup sugar",
|
133 |
+
"1 packet (2 1/4 tsp) active dry yeast",
|
134 |
+
"1 tsp salt",
|
135 |
+
"1 cup milk, warmed",
|
136 |
+
"1/4 cup unsalted butter, melted",
|
137 |
+
"2 large eggs",
|
138 |
+
"1/2 cup brown sugar",
|
139 |
+
"2 tbsp cinnamon",
|
140 |
+
"1/4 cup unsalted butter, softened (for filling)",
|
141 |
+
"1 cup powdered sugar (for icing)",
|
142 |
+
"2-3 tbsp milk (for icing)"
|
143 |
+
],
|
144 |
+
"instructions": [
|
145 |
+
"In a bowl, combine flour, sugar, yeast, and salt.",
|
146 |
+
"In another bowl, mix warm milk, melted butter, and eggs.",
|
147 |
+
"Combine wet and dry ingredients and knead until smooth.",
|
148 |
+
"Let the dough rise for 1 hour.",
|
149 |
+
"Roll out the dough, spread softened butter, and sprinkle with brown sugar and cinnamon.",
|
150 |
+
"Roll up the dough and cut into slices.",
|
151 |
+
"Place in a greased baking dish and let rise for another 30 minutes.",
|
152 |
+
"Bake at 350°F (175°C) for 25-30 minutes.",
|
153 |
+
"Mix powdered sugar and milk to make icing and drizzle over warm rolls."
|
154 |
+
]
|
155 |
+
},
|
156 |
+
{
|
157 |
+
"name": "Pumpkin Pie",
|
158 |
+
"ingredients": [
|
159 |
+
"1 (9-inch) pie crust",
|
160 |
+
"1 can (15 oz) pumpkin puree",
|
161 |
+
"3/4 cup sugar",
|
162 |
+
"1/2 tsp salt",
|
163 |
+
"1 tsp ground cinnamon",
|
164 |
+
"1/2 tsp ground ginger",
|
165 |
+
"1/4 tsp ground nutmeg",
|
166 |
+
"1/4 tsp ground cloves",
|
167 |
+
"3 large eggs",
|
168 |
+
"1 can (12 oz) evaporated milk"
|
169 |
+
]},]
|
__pycache__/Constants.cpython-311.pyc
ADDED
Binary file (4.37 kB). View file
|
|
app.py
CHANGED
@@ -13,9 +13,9 @@ from sqlalchemy import (
|
|
13 |
Integer,
|
14 |
Float,
|
15 |
insert,
|
16 |
-
text,
|
17 |
)
|
18 |
-
|
19 |
import gradio as gr
|
20 |
import os
|
21 |
from smolagents import Tool, CodeAgent, HfApiModel, stream_to_gradio
|
@@ -24,123 +24,12 @@ from dotenv import load_dotenv
|
|
24 |
from langchain.docstore.document import Document
|
25 |
import chromadb
|
26 |
from chromadb.utils import embedding_functions
|
27 |
-
|
28 |
load_dotenv()
|
29 |
#sample questions
|
30 |
# What is the average each customer paid?
|
31 |
# Create a sql statement and invoke your sql_engine tool
|
32 |
|
33 |
-
baking_recipes = [
|
34 |
-
{
|
35 |
-
"name": "Lemon Blueberry Muffins",
|
36 |
-
"ingredients": [
|
37 |
-
"1 1/2 cups all-purpose flour",
|
38 |
-
"1/2 cup sugar",
|
39 |
-
"2 tsp baking powder",
|
40 |
-
"1/2 tsp salt",
|
41 |
-
"1/2 cup milk",
|
42 |
-
"1/3 cup vegetable oil",
|
43 |
-
"1 large egg",
|
44 |
-
"1 tsp vanilla extract",
|
45 |
-
"1 cup fresh blueberries",
|
46 |
-
"Zest of 1 lemon",
|
47 |
-
"Juice of 1 lemon"
|
48 |
-
],
|
49 |
-
"instructions": [
|
50 |
-
"Preheat the oven to 375°F (190°C) and line a muffin tin with paper liners.",
|
51 |
-
"In a bowl, mix flour, sugar, baking powder, and salt.",
|
52 |
-
"In another bowl, whisk together milk, oil, egg, vanilla, lemon zest, and lemon juice.",
|
53 |
-
"Combine wet and dry ingredients until just mixed, then fold in blueberries.",
|
54 |
-
"Spoon the batter into the muffin tin and bake for 18-20 minutes."
|
55 |
-
]
|
56 |
-
},
|
57 |
-
{
|
58 |
-
"name": "Chocolate Chip Cookies",
|
59 |
-
"ingredients": [
|
60 |
-
"1 cup unsalted butter, softened",
|
61 |
-
"3/4 cup sugar",
|
62 |
-
"3/4 cup brown sugar",
|
63 |
-
"1 tsp vanilla extract",
|
64 |
-
"2 large eggs",
|
65 |
-
"2 1/4 cups all-purpose flour",
|
66 |
-
"1 tsp baking soda",
|
67 |
-
"1/2 tsp salt",
|
68 |
-
"2 cups chocolate chips"
|
69 |
-
],
|
70 |
-
"instructions": [
|
71 |
-
"Preheat the oven to 350°F (175°C).",
|
72 |
-
"In a bowl, cream together butter, sugar, and brown sugar until smooth.",
|
73 |
-
"Beat in vanilla and eggs one at a time.",
|
74 |
-
"In another bowl, combine flour, baking soda, and salt.",
|
75 |
-
"Gradually blend dry ingredients into the wet mixture.",
|
76 |
-
"Stir in chocolate chips.",
|
77 |
-
"Drop by rounded tablespoons onto ungreased baking sheets and bake for 9-11 minutes."
|
78 |
-
]
|
79 |
-
},
|
80 |
-
{
|
81 |
-
"name": "Banana Bread",
|
82 |
-
"ingredients": [
|
83 |
-
"3 ripe bananas, mashed",
|
84 |
-
"1/3 cup melted butter",
|
85 |
-
"1 tsp baking soda",
|
86 |
-
"Pinch of salt",
|
87 |
-
"3/4 cup sugar",
|
88 |
-
"1 large egg, beaten",
|
89 |
-
"1 tsp vanilla extract",
|
90 |
-
"1 1/2 cups all-purpose flour"
|
91 |
-
],
|
92 |
-
"instructions": [
|
93 |
-
"Preheat the oven to 350°F (175°C) and grease a 4x8 inch loaf pan.",
|
94 |
-
"In a mixing bowl, mix the mashed bananas with the melted butter.",
|
95 |
-
"Stir in baking soda and salt.",
|
96 |
-
"Add sugar, egg, and vanilla, and mix well.",
|
97 |
-
"Finally, mix in the flour until just incorporated.",
|
98 |
-
"Pour the batter into the prepared loaf pan and bake for 60-65 minutes."
|
99 |
-
]
|
100 |
-
},
|
101 |
-
{
|
102 |
-
"name": "Cinnamon Rolls",
|
103 |
-
"ingredients": [
|
104 |
-
"4 cups all-purpose flour",
|
105 |
-
"1/4 cup sugar",
|
106 |
-
"1 packet (2 1/4 tsp) active dry yeast",
|
107 |
-
"1 tsp salt",
|
108 |
-
"1 cup milk, warmed",
|
109 |
-
"1/4 cup unsalted butter, melted",
|
110 |
-
"2 large eggs",
|
111 |
-
"1/2 cup brown sugar",
|
112 |
-
"2 tbsp cinnamon",
|
113 |
-
"1/4 cup unsalted butter, softened (for filling)",
|
114 |
-
"1 cup powdered sugar (for icing)",
|
115 |
-
"2-3 tbsp milk (for icing)"
|
116 |
-
],
|
117 |
-
"instructions": [
|
118 |
-
"In a bowl, combine flour, sugar, yeast, and salt.",
|
119 |
-
"In another bowl, mix warm milk, melted butter, and eggs.",
|
120 |
-
"Combine wet and dry ingredients and knead until smooth.",
|
121 |
-
"Let the dough rise for 1 hour.",
|
122 |
-
"Roll out the dough, spread softened butter, and sprinkle with brown sugar and cinnamon.",
|
123 |
-
"Roll up the dough and cut into slices.",
|
124 |
-
"Place in a greased baking dish and let rise for another 30 minutes.",
|
125 |
-
"Bake at 350°F (175°C) for 25-30 minutes.",
|
126 |
-
"Mix powdered sugar and milk to make icing and drizzle over warm rolls."
|
127 |
-
]
|
128 |
-
},
|
129 |
-
{
|
130 |
-
"name": "Pumpkin Pie",
|
131 |
-
"ingredients": [
|
132 |
-
"1 (9-inch) pie crust",
|
133 |
-
"1 can (15 oz) pumpkin puree",
|
134 |
-
"3/4 cup sugar",
|
135 |
-
"1/2 tsp salt",
|
136 |
-
"1 tsp ground cinnamon",
|
137 |
-
"1/2 tsp ground ginger",
|
138 |
-
"1/4 tsp ground nutmeg",
|
139 |
-
"1/4 tsp ground cloves",
|
140 |
-
"3 large eggs",
|
141 |
-
"1 can (12 oz) evaporated milk"
|
142 |
-
]},]
|
143 |
-
|
144 |
|
145 |
@spaces.GPU
|
146 |
def dummy():
|
@@ -243,57 +132,40 @@ def init_db(_engine):
|
|
243 |
with _engine.begin() as connection:
|
244 |
connection.execute(stmt)
|
245 |
|
246 |
-
|
247 |
-
|
248 |
-
table_name,
|
249 |
-
metadata_obj,
|
250 |
-
Column("receipt_id", Integer, primary_key=True),
|
251 |
-
Column("customer_name", String(16), primary_key=True),
|
252 |
-
Column("price", Float),
|
253 |
-
Column("tip", Float),
|
254 |
-
)
|
255 |
-
metadata_obj.create_all(_engine)
|
256 |
-
|
257 |
-
rows = [
|
258 |
-
{"receipt_id": 1, "customer_name": "Alan Payne", "price": 12.06, "tip": 1.20},
|
259 |
-
{"receipt_id": 2, "customer_name": "Alex Mason", "price": 23.86, "tip": 0.24},
|
260 |
-
{
|
261 |
-
"receipt_id": 3,
|
262 |
-
"customer_name": "Woodrow Wilson",
|
263 |
-
"price": 53.43,
|
264 |
-
"tip": 5.43,
|
265 |
-
},
|
266 |
-
{
|
267 |
-
"receipt_id": 4,
|
268 |
-
"customer_name": "Margaret James",
|
269 |
-
"price": 21.11,
|
270 |
-
"tip": 1.00,
|
271 |
-
},
|
272 |
-
]
|
273 |
-
insert_rows_into_table(rows, receipts)
|
274 |
-
|
275 |
-
table_name = "waiters"
|
276 |
-
waiters = Table(
|
277 |
-
table_name,
|
278 |
metadata_obj,
|
279 |
-
Column("
|
280 |
-
Column("
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
281 |
)
|
282 |
metadata_obj.create_all(_engine)
|
|
|
283 |
|
284 |
-
rows = [
|
285 |
-
{"receipt_id": 1, "waiter_name": "Corey Johnson"},
|
286 |
-
{"receipt_id": 2, "waiter_name": "Michael Watts"},
|
287 |
-
{"receipt_id": 3, "waiter_name": "Michael Watts"},
|
288 |
-
{"receipt_id": 4, "waiter_name": "Margaret James"},
|
289 |
-
]
|
290 |
-
insert_rows_into_table(rows, waiters)
|
291 |
return _engine
|
292 |
|
293 |
|
294 |
if __name__ == "__main__":
|
295 |
-
engine = create_engine("sqlite
|
296 |
engine = init_db(engine)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
297 |
model = HfApiModel(
|
298 |
model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
|
299 |
# model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
|
@@ -306,16 +178,6 @@ if __name__ == "__main__":
|
|
306 |
max_steps=10,
|
307 |
verbosity_level=1,
|
308 |
)
|
309 |
-
source_docs = HuggingFaceDatasetLoader("MuskumPillerum/General-Knowledge", "Answer").load()[:100]
|
310 |
-
text_splitter = RecursiveCharacterTextSplitter(
|
311 |
-
chunk_size=500,
|
312 |
-
chunk_overlap=50,
|
313 |
-
add_start_index=True,
|
314 |
-
strip_whitespace=True,
|
315 |
-
separators=["\n\n", "\n", ".", " ", ""],
|
316 |
-
)
|
317 |
-
docs_processed = text_splitter.split_documents(source_docs)
|
318 |
-
retriever_tool = RetrieverTool(docs_processed)
|
319 |
|
320 |
|
321 |
retriever_agent = CodeAgent(
|
|
|
13 |
Integer,
|
14 |
Float,
|
15 |
insert,
|
16 |
+
text, Numeric, DateTime,func
|
17 |
)
|
18 |
+
import datetime
|
19 |
import gradio as gr
|
20 |
import os
|
21 |
from smolagents import Tool, CodeAgent, HfApiModel, stream_to_gradio
|
|
|
24 |
from langchain.docstore.document import Document
|
25 |
import chromadb
|
26 |
from chromadb.utils import embedding_functions
|
27 |
+
from Constants import bakery_orders_data,baking_recipes
|
28 |
load_dotenv()
|
29 |
#sample questions
|
30 |
# What is the average each customer paid?
|
31 |
# Create a sql statement and invoke your sql_engine tool
|
32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
33 |
|
34 |
@spaces.GPU
|
35 |
def dummy():
|
|
|
132 |
with _engine.begin() as connection:
|
133 |
connection.execute(stmt)
|
134 |
|
135 |
+
baking_ingredients_order = Table(
|
136 |
+
'baking_ingredients_order', # Table name
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
137 |
metadata_obj,
|
138 |
+
Column("order_id", Integer, primary_key=True, autoincrement=True),
|
139 |
+
Column("ingredient_name", String(100), nullable=False),
|
140 |
+
Column("supplier_name", String(100), nullable=False),
|
141 |
+
Column("quantity_ordered", Numeric(10, 2), nullable=False),
|
142 |
+
Column("unit", String(50), nullable=False),
|
143 |
+
Column("price_per_unit", Numeric(10, 2), nullable=False),
|
144 |
+
Column("total_cost", Numeric(10, 2),
|
145 |
+
default=func.expr('quantity_ordered * price_per_unit'),
|
146 |
+
nullable=False),
|
147 |
+
Column("order_date", DateTime, default=datetime.datetime.now())
|
148 |
)
|
149 |
metadata_obj.create_all(_engine)
|
150 |
+
insert_rows_into_table(bakery_orders_data, baking_ingredients_order)
|
151 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
152 |
return _engine
|
153 |
|
154 |
|
155 |
if __name__ == "__main__":
|
156 |
+
engine = create_engine("sqlite:///sqlite3.sqlite")
|
157 |
engine = init_db(engine)
|
158 |
+
source_docs = HuggingFaceDatasetLoader("MuskumPillerum/General-Knowledge", "Answer").load()[:100]
|
159 |
+
text_splitter = RecursiveCharacterTextSplitter(
|
160 |
+
chunk_size=500,
|
161 |
+
chunk_overlap=50,
|
162 |
+
add_start_index=True,
|
163 |
+
strip_whitespace=True,
|
164 |
+
separators=["\n\n", "\n", ".", " ", ""],
|
165 |
+
)
|
166 |
+
docs_processed = text_splitter.split_documents(source_docs)
|
167 |
+
retriever_tool = RetrieverTool(docs_processed)
|
168 |
+
|
169 |
model = HfApiModel(
|
170 |
model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
|
171 |
# model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
|
|
|
178 |
max_steps=10,
|
179 |
verbosity_level=1,
|
180 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
181 |
|
182 |
|
183 |
retriever_agent = CodeAgent(
|
sqlite3.sqlite
ADDED
Binary file (8.19 kB). View file
|
|