Chris4K commited on
Commit
62fc389
·
verified ·
1 Parent(s): 1bdc7fe

Update app_agent_config.py

Browse files
Files changed (1) hide show
  1. 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
- global tool_checkboxes, url_endpoint, log_enabled
8
- tool_checkboxes = []
9
- url_endpoint = ""
10
- log_enabled = False
 
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