Update app.py
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
|
|
3 |
import requests
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
-
from smolagents import CodeAgent, LiteLLMModel
|
7 |
|
8 |
# (Keep Constants as is)
|
9 |
# --- Constants ---
|
@@ -13,7 +13,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
13 |
# import custom tools to use
|
14 |
from tools import SerperSearchTool
|
15 |
|
16 |
-
|
17 |
|
18 |
# Initialize the watsonx model
|
19 |
llm = LiteLLMModel(
|
@@ -26,7 +26,7 @@ llm = LiteLLMModel(
|
|
26 |
|
27 |
# Create Alfred with all the tools
|
28 |
agent = CodeAgent(
|
29 |
-
tools=[
|
30 |
model=llm,
|
31 |
add_base_tools=True, # Add any additional base tools
|
32 |
planning_interval=3 # Enable planning every 3 steps
|
|
|
3 |
import requests
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
+
from smolagents import CodeAgent, LiteLLMModel
|
7 |
|
8 |
# (Keep Constants as is)
|
9 |
# --- Constants ---
|
|
|
13 |
# import custom tools to use
|
14 |
from tools import SerperSearchTool
|
15 |
|
16 |
+
web_search_tool = SerperSearchTool()
|
17 |
|
18 |
# Initialize the watsonx model
|
19 |
llm = LiteLLMModel(
|
|
|
26 |
|
27 |
# Create Alfred with all the tools
|
28 |
agent = CodeAgent(
|
29 |
+
tools=[web_search_tool],
|
30 |
model=llm,
|
31 |
add_base_tools=True, # Add any additional base tools
|
32 |
planning_interval=3 # Enable planning every 3 steps
|