Update app_agent_config.py
Browse files- app_agent_config.py +15 -28
app_agent_config.py
CHANGED
@@ -4,33 +4,20 @@ from tool_loader import ToolLoader
|
|
4 |
from tool_config import tool_names
|
5 |
from logger import log_enabled
|
6 |
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
|
|
11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
|
13 |
-
def app_agent_config():
|
14 |
-
# Code for URL and Tools checkboxes
|
15 |
-
# Examples for the user perspective
|
16 |
-
st.markdown("Change the agents configuration here.")
|
17 |
-
global tool_checkboxes, url_endpoint, log_enabled
|
18 |
-
|
19 |
-
# Add a dropdown for selecting the inference URL
|
20 |
-
|
21 |
-
url_endpoint = st.selectbox("Select Inference URL", [
|
22 |
-
"https://api-inference.huggingface.co/models/bigcode/starcoder",
|
23 |
-
"https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
|
24 |
-
"https://api-inference.huggingface.co/models/gpt2"
|
25 |
-
])
|
26 |
-
|
27 |
-
|
28 |
-
# Create tool loader instance
|
29 |
-
tool_loader = ToolLoader(tool_names)
|
30 |
-
|
31 |
-
# Add a checkbox for enabling logging
|
32 |
-
|
33 |
-
log_enabled = st.checkbox("Enable Logging")
|
34 |
-
|
35 |
-
tool_checkboxes = [st.checkbox(f"{tool.name} --- {tool.description} ") for tool in tool_loader.tools]
|
36 |
-
|
|
|
4 |
from tool_config import tool_names
|
5 |
from logger import log_enabled
|
6 |
|
7 |
+
class AgentConfig:
|
8 |
+
def __init__(self):
|
9 |
+
self.tool_checkboxes = []
|
10 |
+
self.url_endpoint = ""
|
11 |
+
self.log_enabled = False
|
12 |
|
13 |
+
def configure(self):
|
14 |
+
st.markdown("Change the agent's configuration here.")
|
15 |
+
self.url_endpoint = st.selectbox("Select Inference URL", [
|
16 |
+
"https://api-inference.huggingface.co/models/bigcode/starcoder",
|
17 |
+
"https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
|
18 |
+
"https://api-inference.huggingface.co/models/gpt2"
|
19 |
+
])
|
20 |
+
tool_loader = ToolLoader(tool_names)
|
21 |
+
self.log_enabled = st.checkbox("Enable Logging")
|
22 |
+
self.tool_checkboxes = [st.checkbox(f"{tool.name} --- {tool.description} ") for tool in tool_loader.tools]
|
23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|