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 |
|