update
Browse files- app_agent_config.py +5 -0
- 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 |
|