Skip to content

Commit a6ef463

Browse files
authored
Merge pull request #27 from MichaelClifford/streamlit_client
small updates to streamlit_app that were missed in #24
2 parents ca1b238 + 162cdac commit a6ef463

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

app/src/streamlit_app.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
import streamlit as st
2-
from llama_stack_client.lib.agents.agent import Agent
3-
from llama_stack_client import LlamaStackClient, APIConnectionError
2+
from llama_stack_client import LlamaStackClient, Agent, APIConnectionError
43
from dotenv import load_dotenv
54
import os
65
import uuid
@@ -46,7 +45,7 @@ def reset_agent():
4645

4746
active_tool_list = []
4847
for toolgroup_id in toolgroup_selection:
49-
active_tool_list.extend([t.identifier for t in client.tools.list(toolgroup_id=toolgroup_id)])
48+
active_tool_list.extend([f"{toolgroup_id[5:]}:{t.identifier}" for t in client.tools.list(toolgroup_id=toolgroup_id)])
5049

5150
st.markdown(f"Active Tools: 🛠 {len(active_tool_list)}")
5251
st.json(active_tool_list)

0 commit comments

Comments
 (0)