themissingCRAM commited on
Commit
1b0cc5f
·
1 Parent(s): 97413fe

switch to llama llm 4 && git push

Browse files



@tool

def sql_engine_tool(
query: str,
) -> str:
"""
Allows you to perform SQL queries on the table. Returns a string representation of the result.
The table is named receipts. Its description is as follows:
Columns:
- receipt_id: INTEGER
- customer_name: VARCHAR(16)
- price: FLOAT
- tip: FLOAT

Args:
query: The query to perform. This should be correct SQL.

"""
output = ""
with engine.connect() as con:
rows = con.execute(text(query))
for row in rows:
output += "\n" + str(row)
return output

Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -47,6 +47,7 @@ def sql_engine_tool(
47
 
48
  """
49
  output = ""
 
50
  with engine.connect() as con:
51
  rows = con.execute(text(query))
52
  for row in rows:
@@ -94,6 +95,7 @@ def init_db(engine):
94
  insert_rows_into_table(rows, receipts)
95
  with engine.begin() as conn:
96
  print("SELECT test", conn.execute(text("SELECT * FROM receipts")).fetchall())
 
97
  return engine
98
 
99
 
@@ -101,7 +103,7 @@ if __name__ == "__main__":
101
  engine = create_engine("sqlite:///:memory:")
102
  engine = init_db(engine)
103
  model = HfApiModel(
104
- model_id="",
105
  token=os.getenv("my_first_agents_hf_tokens"),
106
  )
107
 
 
47
 
48
  """
49
  output = ""
50
+ print(engine)
51
  with engine.connect() as con:
52
  rows = con.execute(text(query))
53
  for row in rows:
 
95
  insert_rows_into_table(rows, receipts)
96
  with engine.begin() as conn:
97
  print("SELECT test", conn.execute(text("SELECT * FROM receipts")).fetchall())
98
+ print(engine)
99
  return engine
100
 
101
 
 
103
  engine = create_engine("sqlite:///:memory:")
104
  engine = init_db(engine)
105
  model = HfApiModel(
106
+ model_id="meta-llama/Meta-Llama-3.1-8B-Instruct",
107
  token=os.getenv("my_first_agents_hf_tokens"),
108
  )
109