Upload 2 files

agent.py

@@ -2,11 +2,11 @@ from langgraph.graph import StateGraph, END, START
 from langchain_core.rate_limiters import InMemoryRateLimiter
 from langgraph.prebuilt import ToolNode
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
-from tools import get_rendered_html, download_file, post_request, run_code, add_dependencies
+from tools import get_rendered_html, download_file, post_request, run_code, add_dependencies, transcribe_audio
 from typing import TypedDict, Annotated, List, Any
 from langgraph.graph.message import add_messages
-# 👇 सीधे क्लास इम्पोर्ट करें (init_chat_model की जगह)
-from langchain_google_genai import ChatGoogleGenerativeAI 
+# 👇 Switch to Groq
+from langchain_groq import ChatGroq
 import os
 from dotenv import load_dotenv
 
@@ -22,29 +22,32 @@ RECURSION_LIMIT = 5000
 class AgentState(TypedDict):
     messages: Annotated[List, add_messages]
 
-TOOLS = [run_code, get_rendered_html, download_file, post_request, add_dependencies]
+# Define your tools list
+TOOLS = [run_code, get_rendered_html, download_file, post_request, add_dependencies, transcribe_audio]
 
 # -------------------------------------------------
-# GEMINI LLM SETUP (FIXED)
+# GROQ LLM SETUP
 # -------------------------------------------------
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
-# Debugging check to ensure key exists
-if not GOOGLE_API_KEY:
-    print("❌ CRITICAL ERROR: GOOGLE_API_KEY not found in environment variables!")
+if not GROQ_API_KEY:
+    print("❌ CRITICAL ERROR: GROQ_API_KEY not found in environment variables!")
 else:
-    print(f"✅ GOOGLE_API_KEY found (Length: {len(GOOGLE_API_KEY)})")
+    print(f"✅ GROQ_API_KEY found")
 
+# Groq Free Tier often allows ~30 requests per minute.
+# We set a limiter to be safe (e.g., 1 request every 2 seconds).
 rate_limiter = InMemoryRateLimiter(
-    requests_per_second=9/60,  
-    check_every_n_seconds=1,  
-    max_bucket_size=9  
+    requests_per_second=30/60,  # 0.5 requests per second
+    check_every_n_seconds=0.1,
+    max_bucket_size=1
 )
 
-# 👇 Direct Instantiation to fix Auth Error
-llm = ChatGoogleGenerativeAI(
-    model="gemini-2.0-flash", # Stable model version
-    google_api_key=GOOGLE_API_KEY, # Explicitly passing key param
+# 👇 Using Llama 3.3 70B (High intelligence, currently free on Groq)
+# If you hit Token Limits (TPM), switch model to "llama-3.1-8b-instant"
+llm = ChatGroq(
+    model="llama-3.3-70b-versatile", 
+    api_key=GROQ_API_KEY,
     rate_limiter=rate_limiter,
     temperature=0
 ).bind_tools(TOOLS)
@@ -64,6 +67,12 @@ Your job is to:
 5. Read the server response and:
    - If it contains a new quiz URL → fetch it immediately and continue.
    - If no new URL is present → return "END".
+   
+AUDIO TASKS:
+- If you encounter an audio file (mp3, wav), you MUST:
+  1. Use 'download_file' to save it.
+  2. Use 'transcribe_audio' on the saved filename to get the text.
+  3. Use the transcribed text as the answer (or part of the answer).
 
 STRICT RULES — FOLLOW EXACTLY:
 
@@ -74,6 +83,7 @@ GENERAL RULES:
 - NEVER re-submit unless the server explicitly allows or it's within the 3-minute limit.
 - ALWAYS inspect the server response before deciding what to do next.
 - ALWAYS use the tools provided to fetch, scrape, download, render HTML, or send requests.
+- **IMPORTANT**: If the HTML content is too large, focus only on the relevant forms and instructions.
 
 TIME LIMIT RULES:
 - Each task has a hard 3-minute limit.
@@ -109,6 +119,7 @@ llm_with_prompt = prompt | llm
 # AGENT NODE
 # -------------------------------------------------
 def agent_node(state: AgentState):
+    # Invoke the LLM
     result = llm_with_prompt.invoke({"messages": state["messages"]})
     return {"messages": state["messages"] + [result]}
 
@@ -136,6 +147,7 @@ def route(state):
     elif isinstance(last, dict):
         content = last.get("content")
 
+    # Check for END signal
     if isinstance(content, str) and content.strip() == "END":
         return END
     if isinstance(content, list) and len(content) > 0 and isinstance(content[0], dict):
@@ -163,9 +175,12 @@ app = graph.compile()
 # TEST FUNCTION
 # -------------------------------------------------
 def run_agent(url: str) -> str:
-    print(f"🚀 Starting Agent for URL: {url}")
-    app.invoke({
-        "messages": [{"role": "user", "content": url}]},
+    print(f"🚀 Starting Groq Agent for URL: {url}")
+    # Initialize with user message
+    initial_message = {"role": "user", "content": url}
+    
+    app.invoke(
+        {"messages": [initial_message]},
         config={"recursion_limit": RECURSION_LIMIT},
     )
     print("✅ Tasks completed successfully")

requirements.txt

@@ -14,4 +14,6 @@ matplotlib
 pandas
 pypdf2
 numpy
-google-genai
+google-genai
+langchain-groq
+langchain-ollama