themissingCRAM commited on
Commit
f7947fc
·
1 Parent(s): 36a05b5

switch to llama llm 2

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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: any = 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="meta-llama/Meta-Llama-3.1-8B-Instruct",
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