themissingCRAM
commited on
Commit
·
4674139
1
Parent(s):
5618e69
system prompt
Browse files
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 |
-
|
97 |
-
|
98 |
-
|
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?
|