fix: added tools

agent_course_tools.py

@@ -0,0 +1,16 @@
+from llama_index.tools.wikipedia import WikipediaToolSpec
+from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+from llama_index.core.tools.tool_spec.load_and_search import (
+    LoadAndSearchToolSpec,
+)
+
+wiki_spec = WikipediaToolSpec()
+wiki_tool = LoadAndSearchToolSpec.from_defaults(wiki_spec.to_tool_list()[1]).to_tool_list()
+
+ddgo_spec = DuckDuckGoSearchToolSpec()
+ddgo_tool = LoadAndSearchToolSpec.from_defaults(ddgo_spec.to_tool_list()[1]).to_tool_list()
+
+# Create the Agent with load/search tools
+#agent = OpenAIAgent.from_tools(
+#    LoadAndSearchToolSpec.from_defaults(wiki_tool).to_tool_list(), verbose=True
+#)

app.py

@@ -3,6 +3,10 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+from llama_index.core.agent.workflow import AgentWorkflow
+from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
+#from llama_index.core.tools.tool_spec.load_and_search import LoadAndSearchToolSpec
+import agent_course_tools as tools
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -13,6 +17,15 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
+        llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
+        agent = AgentWorkflow.from_tools_or_functions([
+            tools.wiki_tool,
+            tools.ddgo_tool
+            ],
+            llm=llm,
+            system_prompt="You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.",
+        )
+
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
         fixed_answer = "This is a default answer."

requirements.txt

@@ -1,3 +1,7 @@
 gradio
 requests
-llama-index
+llama-index
+llama-index-tools-wikipedia
+llama-index-llms-litellm
+llama-index-readers-file
+llama-index-tools-duckduckgo