tracing

.pre-commit-config.yaml

@@ -0,0 +1,54 @@
+# This is a .pre-commit-config.yaml file. The pre-commit tool uses this configuration
+# to manage and enforce coding standards and checks before code commits.
+# It helps automate checks for common issues, ensuring that changes committed to the repository
+# comply with predefined standards and are free of certain types of errors.
+
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v5.0.0 # The version of the general pre-commit hooks we are using.
+    hooks:
+      - id: check-ast
+        # Checks if Python Abstract Syntax Trees (ASTs) are well-formed, catching syntax errors.
+
+      - id: check-yaml
+        # Verifies the syntax of YAML files, ensuring they are correctly structured.
+
+      - id: end-of-file-fixer
+        # Ensures that all files end with a newline, which is a common POSIX standard.
+
+      - id: trailing-whitespace
+        # Removes any trailing whitespace at the end of lines in text files.
+
+      - id: check-executables-have-shebangs
+        # Ensures that executable files have a shebang line, which indicates how the script should be executed.
+
+      - id: debug-statements
+        # Checks for the presence of debug statements (like Python's breakpoint) that shouldn't be in committed code.
+
+      - id: check-added-large-files
+        args: ["--maxkb=5000"]
+        # Prevents accidentally committing large files to the repository by setting a size limit of 5000KB.
+
+      - id: mixed-line-ending
+        # Ensures consistent line ending style across files, avoiding mixtures that can lead to problems on different platforms.
+        # Checks if Jupyter notebooks are clean and free of output cells.
+  - repo: https://github.com/srstevenson/nb-clean
+    rev: 4.0.1
+    hooks:
+      - id: nb-clean
+        args:
+          - --remove-empty-cells
+          - --preserve-cell-outputs
+  # - repo: https://github.com/PyCQA/isort
+  #   rev: 6.0.1 # The version of isort we are using to sort imports in Python files.
+  #   hooks:
+  #     - id: isort
+  #       args: ["--profile", "black", "--filter-files"]
+  #       # Configures isort to use the Black formatting profile and only filter files that were selected by pre-commit.
+  #       language_version: python3 # Specifies that this should run with Python 3.
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: "v0.9.10" # Specific version of ruff, a fast Python linter.
+    hooks:
+      - id: ruff
+        # Runs ruff to perform linting on the code, checking for stylistic and logical errors not covered by other tools.
+      - id: ruff-format

app.py

@@ -1,26 +1,78 @@
 import os
 import gradio as gr
 import requests
-import inspect
 import pandas as pd
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel
+from smolagents import (
+    ToolCallingAgent,
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    HfApiModel,
+    VisitWebpageTool,
+)
+from dotenv import load_dotenv
+import base64
+from opentelemetry.sdk.trace import TracerProvider
+from openinference.instrumentation.smolagents import SmolagentsInstrumentor
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.trace.export import SimpleSpanProcessor
+from opentelemetry import trace
+
+load_dotenv()
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
+# Get your own keys from https://cloud.langfuse.com
+
+
+os.environ["LANGFUSE_HOST"] = "https://cloud.langfuse.com"  # 🇪🇺 EU region example
+# os.environ["LANGFUSE_HOST"] = "https://us.cloud.langfuse.com"  # 🇺🇸 US region example
+
+LANGFUSE_AUTH = base64.b64encode(
+    f"{os.environ.get('LANGFUSE_PUBLIC_KEY')}:{os.environ.get('LANGFUSE_SECRET_KEY')}".encode()
+).decode()
+
+os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = (
+    os.environ.get("LANGFUSE_HOST") + "/api/public/otel"
+)
+os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
+
+# Create a TracerProvider for OpenTelemetry
+trace_provider = TracerProvider()
+
+# Add a SimpleSpanProcessor with the OTLPSpanExporter to send traces
+trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter()))
+
+# Set the global default tracer provider
+
+trace.set_tracer_provider(trace_provider)
+tracer = trace.get_tracer(__name__)
+
+# Instrument smolagents with the configured provider
+SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
+
+
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 def get_agent():
     llm = HfApiModel(
         model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=os.getenv("HF_TOKEN")
     )
-    agent = CodeAgent(
-        tools=[DuckDuckGoSearchTool()],
+    web_agent = ToolCallingAgent(
+        tools=[DuckDuckGoSearchTool(), VisitWebpageTool()],
+        model=llm,
+        max_steps=10,
+        description="A web agent that can search the web and visit webpages.",
+    )
+
+    manager_agent = CodeAgent(
+        tools=[web_agent],
         model=llm,
+        max_steps=10,
     )
-    return agent
+    return manager_agent
 
 
 class BasicAgent:
@@ -97,7 +149,7 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     results_log = []
     answers_payload = []
     print(f"Running agent on {len(questions_data)} questions...")
-    for item in questions_data:
+    for item in questions_data[:1]:
         task_id = item.get("task_id")
         question_text = item.get("question")
         if not task_id or question_text is None:

requirements.in

@@ -1,3 +1,4 @@
 gradio[oauth]
 requests
-smolagents[gradio,litellm,openai,telemetry,toolkit,torch,transformers,vision]
+smolagents[gradio,litellm,openai,telemetry,toolkit,torch,transformers,vision]
+duckduckgo_search>=7.0.0,<8.0.0

requirements.txt

@@ -79,8 +79,10 @@ distro==1.9.0
     # via openai
 dnspython==2.7.0
     # via email-validator
-duckduckgo-search==8.0.1
-    # via smolagents
+duckduckgo-search==7.5.5
+    # via
+    #   -r requirements.in
+    #   smolagents
 email-validator==2.2.0
     # via arize-phoenix
 fastapi==0.115.12