file handling

agent_utils.py

@@ -3,6 +3,11 @@ import arxiv
 import os
 from tavily import TavilyClient
 
+import requests
+from PIL import Image
+from io import BytesIO
+
+
 def wiki_ret(question: str) -> str:
     """ Retrieve docs from wikipedia """
 
@@ -59,3 +64,31 @@ def tavily_ret(query: str, max_results: int = 3) -> str:
         summaries.append(f"Title: {item['title']}\nURL: {item['url']}\nSnippet: {item['content']}\n")
 
     return "\n\n".join(summaries)
+
+
+
+def handle_file_tool(input: dict) -> str:
+    """Processes file attached to a question"""
+    question = input.get("question", "")
+    file_url = input.get("file", "")
+
+    if not file_url:
+        return "No file provided."
+
+    response = requests.get(file_url)
+    if not response.ok:
+        return f"Failed to download file: {file_url}"
+
+    # Handle image files
+    if file_url.endswith((".png", ".jpg", ".jpeg")):
+        image = Image.open(BytesIO(response.content))
+        return f"Received image for question: '{question}'. (Add vision model here.)"
+
+    # Handle PDFs
+    elif file_url.endswith(".pdf"):
+        with open("temp.pdf", "wb") as f:
+            f.write(response.content)
+        return f"Received PDF for question: '{question}'. (Add PDF parser here.)"
+
+    else:
+        return f"Unsupported file type: {file_url}"

app.py

@@ -17,7 +17,7 @@ from langchain_core.messages import HumanMessage
 from langgraph.graph import START, END, StateGraph
 from langgraph.prebuilt import ToolNode, tools_condition
 
-from agent_utils import wiki_ret, arxiv_ret, tavily_ret
+from agent_utils import wiki_ret, arxiv_ret, tavily_ret, handle_file_tool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -25,7 +25,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 groq_api_key = os.getenv("GROQ_API_KEY")
 
-llm = ChatGroq(api_key=groq_api_key, model="mistral-saba-24b")
+llm = ChatGroq(api_key=groq_api_key, model="llama-3.3-70b-versatile")
 
 class GraphState(TypedDict):
     messages: str #Annotated[list, operator.add]
@@ -82,7 +82,7 @@ class BasicAgent:
     def create_graph(self):
         builder = StateGraph(GraphState)
         builder.add_node("agent", self.agent)
-        builder.add_node("tools", ToolNode(tools = [wiki_ret, arxiv_ret, tavily_ret]))
+        builder.add_node("tools", ToolNode(tools = [wiki_ret, arxiv_ret, tavily_ret, handle_file_tool]))
 
         builder.add_edge(START, "agent")
         builder.add_conditional_edges("agent", tools_condition)

requirements.txt

@@ -6,4 +6,5 @@ langchain-core
 wikipedia
 langgraph
 arxiv
-tavily-python
+tavily-python
+PIL