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 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|