Add Claude class for Anthropic API integration and update requirements

app.py

@@ -12,9 +12,10 @@ from pathlib import Path
 import openai
 from openai import OpenAI
 import pdfplumber
+import anthropic
 
 
-## utilty functions
+## utilties and class definition
 def is_image_extension(filename: str) -> bool:
     IMAGE_EXTS = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp', '.svg'}
     ext = os.path.splitext(filename)[1].lower() # os.path.splitext(path) returns (root, ext)
@@ -61,8 +62,19 @@ def check_format(answer: str | list, *args, **kwargs) -> list:
         return [answer]
     elif isinstance(answer, dict):
         raise TypeError(f"Final answer must be a list, not a dict. Please check the answer format.")
-        
-        
+    
+class Claude:
+    def __init__(self):
+        self.client = anthropic.Anthropic(api_key=os.environ.get("ANTHROPIC_API_KEY"))
+
+    def generate(self, prompt: str):
+        message = self.client.messages.create(
+            model="claude-sonnet-4-20250514",
+            max_tokens=20000,
+            temperature=1,
+            messages=[{"role": "user", "content": prompt}]
+            )
+        return message.content
 ## tools definition
 @tool
 def download_images(image_urls: str) -> list:
@@ -153,11 +165,15 @@ def generate_audio(prompt: str) -> object:
 
 
 
-
 ## agent definition
 class Agent:
     def __init__(self, ):
-        client = HfApiModel("meta-llama/Llama-3.3-70B-Instruct", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
+        #client = HfApiModel("Qwen/Qwen3-32B", provider="nebius", api_key=os.getenv("NEBIUS_API_KEY"))
+        client = OpenAIServerModel(
+            model_id="claude-sonnet-4-20250514",
+            api_base="https://api.anthropic.com/v1/",
+            api_key=os.environ["ANTHROPIC_API_KEY"],
+)
         self.agent = CodeAgent(
             model=client,
             tools=[DuckDuckGoSearchTool(max_results=5), VisitWebpageTool(max_output_length=20000), generate_image, download_images, transcribe_audio],

requirements.txt

@@ -3,4 +3,5 @@ smolagents
 openai
 gradio
 duckduckgo-search
-pdfplumber
+pdfplumber
+anthropic