themissingCRAM commited on
Commit
4674139
·
1 Parent(s): 5618e69

system prompt

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -91,13 +91,19 @@ def init_db():
91
 
92
  if __name__ == "__main__":
93
  init_db()
 
 
 
 
 
94
  agent = CodeAgent(
95
  tools=[sql_engine],
96
- model=HfApiModel(
97
- model_id="HuggingFaceH4/zephyr-7b-beta",
98
- token=os.getenv("my_first_agents_hf_tokens"),
99
- ),
100
  max_steps=1,
101
  verbosity_level=1,
102
  )
103
  GradioUI(agent).launch()
 
 
91
 
92
  if __name__ == "__main__":
93
  init_db()
94
+ model = HfApiModel(
95
+ model_id="HuggingFaceH4/zephyr-7b-beta",
96
+ token=os.getenv("my_first_agents_hf_tokens"),
97
+ )
98
+
99
  agent = CodeAgent(
100
  tools=[sql_engine],
101
+ system_prompt="""
102
+
103
+ You are a text to sql converter
104
+ """,
105
  max_steps=1,
106
  verbosity_level=1,
107
  )
108
  GradioUI(agent).launch()
109
+ # What is the average each customer paid?