Create basic_agent

basic_agent

@@ -0,0 +1,25 @@
+from llama_index.core.agent.workflow import AgentWorkflow
+from llama_index.core.workflow import Context
+from llama_index.core.tools import FunctionTool
+from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
+from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
+
+class BasicAgent:
+    def __init__(self):
+      llm = HuggingFaceInferenceAPI(model_name="Qwen/Qwen2.5-Coder-32B-Instruct")
+
+      # Initialize tools
+      tool_spec = DuckDuckGoSearchToolSpec()
+      search_tool = FunctionTool.from_defaults(tool_spec.duckduckgo_full_search)
+      
+      # Create Alfred with all the tools
+      self.agent = AgentWorkflow.from_tools_or_functions(
+          [search_tool],
+          llm=llm
+      )
+
+      # self.ctx = Context(self.agent)
+
+    def __call__(self, question: str) -> str:
+      response = await self.agent.run(user_msg=question) # ctx=self.ctx)
+      return response.response.content