themissingCRAM
commited on
Commit
·
f7947fc
1
Parent(s):
36a05b5
switch to llama llm 2
Browse files
app.py
CHANGED
@@ -31,7 +31,7 @@ def dummy():
|
|
31 |
|
32 |
|
33 |
@tool
|
34 |
-
def sql_engine_tool(query: str, engine=engine) -> str:
|
35 |
"""
|
36 |
Allows you to perform SQL queries on the table. Returns a string representation of the result.
|
37 |
The table is named 'receipts'. Its description is as follows:
|
@@ -56,7 +56,7 @@ def init_db():
|
|
56 |
|
57 |
metadata_obj = MetaData()
|
58 |
|
59 |
-
def insert_rows_into_table(rows, table, engine
|
60 |
for row in rows:
|
61 |
stmt = insert(table).values(**row)
|
62 |
with engine.begin() as connection:
|
@@ -97,7 +97,7 @@ def init_db():
|
|
97 |
if __name__ == "__main__":
|
98 |
init_db()
|
99 |
model = HfApiModel(
|
100 |
-
model_id="
|
101 |
token=os.getenv("my_first_agents_hf_tokens"),
|
102 |
)
|
103 |
|
|
|
31 |
|
32 |
|
33 |
@tool
|
34 |
+
def sql_engine_tool(query: str, engine: any = engine) -> str:
|
35 |
"""
|
36 |
Allows you to perform SQL queries on the table. Returns a string representation of the result.
|
37 |
The table is named 'receipts'. Its description is as follows:
|
|
|
56 |
|
57 |
metadata_obj = MetaData()
|
58 |
|
59 |
+
def insert_rows_into_table(rows, table, engine=engine):
|
60 |
for row in rows:
|
61 |
stmt = insert(table).values(**row)
|
62 |
with engine.begin() as connection:
|
|
|
97 |
if __name__ == "__main__":
|
98 |
init_db()
|
99 |
model = HfApiModel(
|
100 |
+
model_id="",
|
101 |
token=os.getenv("my_first_agents_hf_tokens"),
|
102 |
)
|
103 |
|