Spaces:
Runtime error
Runtime error
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 |
|