agents

agent.py

@@ -0,0 +1,54 @@
+import os
+from smolagents import HfApiModel, CodeAgent, ToolCallingAgent, DuckDuckGoSearchTool
+from huggingface_hub import login
+from dotenv import load_dotenv
+
+load_dotenv()
+login(os.environ["HF_API_KEY"])
+
+# Tools
+tools = [
+    DuckDuckGoSearchTool(),
+]
+
+# LLM Model 
+model = HfApiModel(
+    "Qwen/Qwen2.5-72B-Instruct",
+    provider="together",
+    max_tokens=4096,
+    temperature=0.1
+    # token=get_huggingface_token(),
+)
+
+# Code Agent 
+codeAgent = CodeAgent(
+    model=model,
+    tools=tools,
+    max_steps=10,
+    additional_authorized_imports=['pandas', 'numpy'],
+    verbosity_level=2
+)
+
+codeAgent.logger.console.width=66
+
+
+# Tool Calling Agent
+llm = HfApiModel(
+    "Qwen/Qwen2.5-72B-Instruct",
+    temperature=0.6
+)
+
+toolCallingAgent = ToolCallingAgent(
+    model=model,
+    tools=tools,
+    max_steps=20,
+)
+
+toolCallingAgent.logger.console.width=66
+
+question = """
+ I'm making a grocery list for my mom, but she's a professor of botany and she's a real stickler when it comes to categorizing things. I need to add different foods to different categories on the grocery list, but if I make a mistake, she won't buy anything inserted in the wrong category. Here's the list I have so far:\n\nmilk, eggs, flour, whole bean coffee, Oreos, sweet potatoes, fresh basil, plums, green beans, rice, corn, bell pepper, whole allspice, acorns, broccoli, celery, zucchini, lettuce, peanuts\n\nI need to make headings for the fruits and vegetables. Could you please create a list of just the vegetables from my list? If you could do that, then I can figure out how to categorize the rest of the list into the appropriate categories. But remember that my mom is a real stickler, so make sure that no botanical fruits end up on the vegetable list, or she won't get them when she's at the store. Please alphabetize the list of vegetables, and place each item in a comma separated list.
+"""
+
+res = toolCallingAgent.run(question)
+print(res)