Chris4K commited on
Commit
032b798
·
1 Parent(s): 7330cbd
Files changed (2) hide show
  1. app_agent_config.py +5 -0
  2. view/app_sidebar.py +1 -5
app_agent_config.py CHANGED
@@ -11,4 +11,9 @@ class AgentConfig:
11
  self.log_enabled = False
12
  self.context = ""
13
  self.tool_loader = ToolLoader(tool_names)
 
 
 
 
 
14
 
 
11
  self.log_enabled = False
12
  self.context = ""
13
  self.tool_loader = ToolLoader(tool_names)
14
+ self.agent_urls = [
15
+ "https://api-inference.huggingface.co/models/bigcode/starcoder",
16
+ "https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
17
+ "https://api-inference.huggingface.co/models/gpt2"
18
+ ]
19
 
view/app_sidebar.py CHANGED
@@ -16,11 +16,7 @@ def app_sidebar(controller):
16
  def configure(agent_config):
17
  st.markdown("Change the agent's configuration here.")
18
 
19
- agent_config.url_endpoint = st.selectbox("Select Inference URL", [
20
- "https://api-inference.huggingface.co/models/bigcode/starcoder",
21
- "https://api-inference.huggingface.co/models/OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
22
- "https://api-inference.huggingface.co/models/gpt2"
23
- ])
24
 
25
  agent_config.log_enabled = st.checkbox("Enable Logging")
26
 
 
16
  def configure(agent_config):
17
  st.markdown("Change the agent's configuration here.")
18
 
19
+ agent_config.url_endpoint = st.selectbox("Select Inference URL", agent_config.agent_urls)
 
 
 
 
20
 
21
  agent_config.log_enabled = st.checkbox("Enable Logging")
22