admin-valentin/awesome-llm-apps
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Refactor investment agent implementation to use 'Agent' class instead of 'Assistant'

Browse source
This commit is contained in:
ksp2000 2025-02-28 21:45:54 +05:30
parent 1fb633ce37
commit 9e9a64f664
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
ai_agent_tutorials/ai_investment_agent/investment_agent.py
View file

@ -1,6 +1,6 @@
# Import the required libraries # Import the required libraries
import streamlit as st import streamlit as st
from agno.agent import Assistant from agno.agent import Agent
from agno.models.openai import OpenAIChat from agno.models.openai import OpenAIChat
from agno.tools.yfinance import YFinanceTools from agno.tools.yfinance import YFinanceTools
@ -12,11 +12,12 @@ st.caption("This app allows you to compare the performance of two stocks and gen
openai_api_key = st.text_input("OpenAI API Key", type="password") openai_api_key = st.text_input("OpenAI API Key", type="password")
if openai_api_key: if openai_api_key:
# Create an instance of the Assistant # Create an instance of the Agent
assistant = Assistant( agent = Agent(
llm=OpenAIChat(model="gpt-4o", api_key=openai_api_key), llm=OpenAIChat(model="gpt-4o", api_key=openai_api_key),
tools=[YFinanceTools(stock_price=True, analyst_recommendations=True, company_info=True, company_news=True)], tools=[YFinanceTools(stock_price=True, analyst_recommendations=True, company_info=True, company_news=True)],
show_tool_calls=True, show_tool_calls=True,
markdown=True
) )
# Input fields for the stocks to compare # Input fields for the stocks to compare
@ -24,7 +25,7 @@ if openai_api_key:
stock2 = st.text_input("Enter the second stock symbol") stock2 = st.text_input("Enter the second stock symbol")
if stock1 and stock2: if stock1 and stock2:
# Get the response from the assistant # Get the response from the Agent
query = f"Compare {stock1} to {stock2}. Use every tool you have." query = f"Compare {stock1} to {stock2}. Use every tool you have."
response = assistant.run(query, stream=False) response = agent.run(query, stream=False)
st.write(response.content) st.write(response.content)