Refactor stock news fetching to use GROQ API and enhance web agent capabilities; update requirements and improve health check endpoint with uptime tracking.

Author: yashksaini-coder

controllers/stockNews.py

@@ -1,31 +1,59 @@
-import finnhub
-import time
-import requests
-import dotenv 
+import groq
 import os
+from dotenv import load_dotenv
+load_dotenv()
 
-dotenv.load_dotenv()
+# AI assistant imports
+from fastapi.responses import HTMLResponse, JSONResponse
+from agno.agent import Agent
+from agno.models.groq import Groq
+from agno.tools.yfinance import YFinanceTools
+from agno.tools.duckduckgo import DuckDuckGoTools
+from agno.agent import Agent, RunResponse
+from agno.tools.wikipedia import WikipediaTools
+from agno.tools.calculator import CalculatorTools
 
-NEWS_API_KEY = os.getenv("NEWS_API_KEY")
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+groq_client = groq.Client(api_key=GROQ_API_KEY)
 
-if not NEWS_API_KEY:
-    raise ValueError("Please provide a NEWS API key")
+if not GROQ_API_KEY:
+    raise ValueError("Please provide a GROQ API key")
+
+# Enhanced web search agent with more capabilities
+web_agent = Agent(
+    name="web_agent",
+    role="comprehensive web research and information gathering specialist",
+    model=Groq(id="llama-3.1-8b-instant", api_key=GROQ_API_KEY),
+    tools=[
+        DuckDuckGoTools(search=True, news=True),
+        WikipediaTools(),
+    ],
+    instructions=[
+        "You are an advanced web research specialist capable of handling complex queries",
+        "Your primary objectives are to:",
+        "1. Break down complex queries into manageable sub-tasks",
+        "2. Gather information from multiple sources for comprehensive answers",
+        "3. Verify information across different sources",
+        "4. Provide well-structured, detailed responses with proper citations",
+        "5. Handle ambiguous queries by asking clarifying questions when needed",
+        "6. Maintain context throughout multi-step queries",
+        "7. Format responses in clear, organized markdown with proper sections",
+        "When dealing with complex queries:",
+        "- Start by analyzing the query components",
+        "- Identify required information sources",
+        "- Gather data systematically",
+        "- Synthesize information coherently",
+        "- Provide clear reasoning for your conclusions"
+    ]
+)
 
-session = requests.Session()
-session.headers.update({
-    "User-Agent": "Chrome/122.0.0.0"
-})
 def fetch_news():
+    """Fetch latest news articles related to stocks and financial markets"""
     try:
-
-        finnhub_client = finnhub.Client(api_key=NEWS_API_KEY)
-
-        news_list =finnhub_client.general_news('general', min_id=4)
-        news_stack=[]
-        for news in news_list[:10]:
-            news_stack.append([news['headline'],news['url']])
-        print("✅ Data fetching done successfully!")   
-        return news_stack
+        response: RunResponse = web_agent.run("Latest news articles related to stocks and financial markets")
+        return {
+            "question": "Latest news articles related to stocks and financial markets",
+            "answer": response.content
+        }
     except Exception as e:
-        print(f"❌ Error fetching news: {e}")
-    time.sleep(5)
+        raise RuntimeError(f"Error fetching news: {str(e)}")

requirements.txt

@@ -10,17 +10,17 @@ click==8.1.8
 colorama==0.4.6
 distro==1.9.0
 dnspython==2.7.0
-docstring_parser==0.16
-duckduckgo_search==7.3.2
-email_validator==2.2.0
+docstring-parser==0.16
+duckduckgo-search==7.3.2
+email-validator==2.2.0
 fastapi==0.115.8
 fastapi-cache2==0.2.2
 fastapi-cli==0.0.7
 finnhub-python==2.4.22
-Flask==3.1.0
+flask==3.1.0
 frozendict==2.4.6
 gitdb==4.0.12
-GitPython==3.1.44
+gitpython==3.1.44
 google-auth==2.38.0
 google-genai==1.5.0
 groq==0.18.0
@@ -31,29 +31,30 @@ httptools==0.6.4
 httpx==0.28.1
 idna==3.10
 itsdangerous==2.2.0
-Jinja2==3.1.5
+jinja2==3.1.5
 lxml==5.3.1
 markdown-it-py==3.0.0
-MarkupSafe==3.0.2
+markupsafe==3.0.2
 mdurl==0.1.2
 multitasking==0.0.11
 numpy==2.2.3
 pandas==2.2.3
 peewee==3.17.9
 pendulum==3.0.0
+pip==24.2
 platformdirs==4.3.6
 primp==0.12.1
 pyasn1==0.6.1
-pyasn1_modules==0.4.1
+pyasn1-modules==0.4.1
 pydantic==2.10.6
+pydantic-core==2.27.2
 pydantic-settings==2.7.1
-pydantic_core==2.27.2
-Pygments==2.19.1
+pygments==2.19.1
 python-dateutil==2.9.0.post0
 python-dotenv==1.0.1
 python-multipart==0.0.20
 pytz==2025.1
-PyYAML==6.0.2
+pyyaml==6.0.2
 redis==5.2.1
 requests==2.32.3
 rich==13.9.4
@@ -68,12 +69,13 @@ starlette==0.45.3
 time-machine==2.16.0
 tomli==2.2.1
 typer==0.15.1
-typing_extensions==4.12.2
+typing-extensions==4.12.2
 tzdata==2025.1
 urllib3==2.3.0
 uvicorn==0.34.0
 watchfiles==1.0.4
 webencodings==0.5.1
 websockets==14.2
-Werkzeug==3.1.3
+werkzeug==3.1.3
+wikipedia==1.4.0
 yfinance==0.2.54

routes/agentRoutes.py

@@ -21,14 +21,18 @@
 if not GROQ_API_KEY:
     raise ValueError("Please provide a GROQ API key")
 
+start_time = datetime.datetime.now(datetime.timezone.utc)
+
 @router.get("/health", response_class=HTMLResponse)
 async def health_check(request: Request):
     """Health check endpoint to verify the API server status and connections."""
+    uptime = (datetime.datetime.now(datetime.timezone.utc) - start_time).total_seconds()
     try:
         response_data = {
             "status": "healthy",
             "timestamp": datetime.datetime.now().isoformat(),
             "uptime": "OK",
+            "uptime_seconds": uptime,
             "api": {
                 "groq_api": "connected" if GROQ_API_KEY else "not configured",
                 "gemini_api":"connected" if GEMINI_API_KEY else "not configured",

routes/stockRoutes.py

@@ -56,7 +56,7 @@ async def stock_news(request: Request):
                 "full_path": f"{request.url.scheme}://{request.url.netloc}/stock-news",
                 "description": "Returns latest news articles related to stocks and financial markets",
                 "parameters": [],
-                "example_response": json.dumps(result[:2], indent=2),
+                "example_response": json.dumps(result, indent=2),
                 "current_year": datetime.datetime.now().year
             })
         return result