Cod Calling 2

app.py

@@ -1,31 +1,8 @@
-from smolagents import ToolCallingAgent, CodeAgent, DuckDuckGoSearchTool, LiteLLMModel
+from smolagents import ToolCallingAgent, CodeAgent, DuckDuckGoSearch, DuckDuckGoSearchTool, LiteLLMModel
 import os
 #import litellm
 #litellm._turn_on_debug()  # Enable debugging for litellm
 
-'''
-import os
-import litellm
-litellm._turn_on_debug()  # Enable debugging for litellm
-
-# Set the API key for Sambanova
-os.environ['SAMBANOVA_API_KEY'] = os.getenv('sambanova_token')
-
-# Test the completion directly with litellm
-try:
-    response = litellm.completion(
-        model="sambanova/Qwen2.5-Coder-32B-Instruct",
-        messages=[{"role": "user", "content": "Tell me about yourself."}],
-        max_tokens=100,
-        api_base="https://api.sambanova.ai/v1",  # Base URL for Sambanova
-    )
-    print("\n++++\nResponse from litellm:\n")
-    print(response)
-except Exception as e:
-    print(f"\n++++\nError occurred:\n")
-    print(f"{e}")
-'''
-
 # Set the API key for Sambanova
 os.environ['SAMBANOVA_API_KEY'] = os.getenv('sambanova_token')
 
@@ -39,8 +16,12 @@ model = LiteLLMModel(
 
 # Create the agent
 #agent = ToolCallingAgent(
+#    tools=[DuckDuckGoSearchTool()],
+#    model=model
+#)
+
 agent = CodeAgent(
-    tools=[DuckDuckGoSearchTool()],
+    tools=[DuckDuckGoSearch()],
     model=model
 )