File size: 2,593 Bytes
23f2703 b970e0d 23f2703 2bcb787 af2f617 b970e0d 2bcb787 af2f617 3b91e23 2bcb787 af2f617 c9f0eeb 23f2703 7bf11dd 23f2703 3b91e23 c9f0eeb 3b91e23 c9f0eeb 0dd6b64 3b91e23 c9f0eeb 7dbebbe 2bcb787 7dbebbe c9f0eeb 2bcb787 c9f0eeb 7dbebbe 2bcb787 c9f0eeb 3b91e23 c9f0eeb 23f2703 c9f0eeb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
import streamlit as st
from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, ToolCallingAgent, ManagedAgent, VisitWebpageTool
web_agent = ToolCallingAgent(
tools=[DuckDuckGoSearchTool(max_results=10), VisitWebpageTool()],
model=HfApiModel(),
)
managed_web_agent = ManagedAgent(
agent=web_agent,
name="search",
description="Runs tech web searches. Give it your query as an argument. Also, this agent should link to the sources you are using. If any specific sites are provided, this agent will only those sources.",
)
manager_agent = CodeAgent(
tools=[],
model=HfApiModel(model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"),
managed_agents=[managed_web_agent],
additional_authorized_imports=['pyparsing', 'matplotlib', 'datetime', 'statistics', 'bs4', 'request', 'unicodedata', 'queue', 'time', 'collections', 're', 'math', 'stat', 'random',
'itertools'],
)
# Function to log agent actions
def log_agent_action(prompt, result, agent_name):
st.write(f"### Agent Activity ({agent_name}):")
st.write("**Prompt Sent to Agent:**")
st.code(prompt, language="text")
st.write("**Agent Output:**")
st.code(result, language="text")
# Streamlit app title
st.title("AI Tech Assistant Agent researching your query and summarizing it")
# App description
st.write("Generate reports enriched with real-time insights using the AI Tech Report Writing Agent powered by SmolAgents and DuckDuckGo.")
# Input blog topic or prompt
prompt = st.text_area("Enter the your search query:", placeholder="E.g., What is the optimal specs for my computer? I am running normal desktop programs. I want my computer to be fast and efficient. Storage is not so important. Please, don't let it be too expensive. Also, let me know where I can buy the computer in Sweden.")
# Button to generate blog content
if st.button("Generate Summary"):
if prompt:
with st.spinner("Generating content..."):
try:
# Run the blog agent with the given prompt
result = manager_agent.run(prompt)
# Display the generated blog content
st.subheader("Generated Summary:")
st.write(result)
# Log backend activity
log_agent_action(prompt, result, "Tech Research Agent with DuckDuckGo")
except Exception as e:
st.error(f"An error occurred: {e}")
else:
st.warning("Please enter a blog topic or prompt to proceed.")
# Footer
st.markdown("---")
st.caption("Powered by SmolAgents, DuckDuckGo, and Streamlit")
|