move to grok-beta

Signed-off-by: apple muncy <[email protected]>

app.py

@@ -1,10 +1,11 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, OpenAIServerModel,  HfApiModel,load_tool,tool
 import datetime
 import requests
 import pytz
 import yaml
+import os
+import json
 from tools.final_answer import FinalAnswerTool
-from tools.web_search import DuckDuckGoSearchTool
 from tools.visit_webpage import VisitWebpageTool
 
 
@@ -38,12 +39,12 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 
 final_answer = FinalAnswerTool()
-web_search = DuckDuckGoSearchTool()
 visit_webpage = VisitWebpageTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
+"""
 model = HfApiModel(
     max_tokens=8384,
     temperature=0.5,
@@ -52,6 +53,46 @@ model = HfApiModel(
     custom_role_conversions=None,
     
 )
+"""
+# Initialize the model
+model = OpenAIServerModel(
+    model_id="grok-beta",
+    api_base="https://api.x.ai/v1",  # Correct base URL
+    api_key=os.environ.get("XAI_API_KEY", "your-api-key-here"),
+)
+
+
+@tool
+def call_grok_api(query: str) -> str:
+    """Calls Gork's API with a JSON-encoded query string and returns the response.
+
+    Args:
+        query (str): A string representing the query to be JSON-encoded and sent to Gork's API.
+
+    Returns:
+        str: A string containing the response from Gork's API, or an error message if the request fails.
+    """
+    try:
+        # Prepare JSON-encoded payload
+        payload = json.dumps({
+            "model": "grok-beta",  # Use available model, e.g., grok-beta
+            "messages": [{"role": "user", "content": query}],
+            "stream": False
+        })
+        # Make API request
+        api_key = os.environ.get("XAI_API_KEY", "your-api-key-here")
+        response = requests.post(
+            "https://api.x.ai/v1/chat/completions",  # Correct xAI API endpoint
+            headers={
+                "Authorization": f"Bearer {api_key}",
+                "Content-Type": "application/json"
+            },
+            data=payload
+        )
+        response.raise_for_status()
+        return response.json()["choices"][0]["message"]["content"]
+    except requests.exceptions.RequestException as e:
+        return f"Error calling xAI API: {str(e)}"
 
 
 # Import tool from Hub
@@ -62,7 +103,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, web_search, visit_webpage ], ## add your tools here (don't remove final answer)
+    tools=[call_grok_api, final_answer,  visit_webpage ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
@@ -71,7 +112,8 @@ agent = CodeAgent(
     description=None,
     prompt_templates=prompt_templates,
     add_base_tools=True,
+    executor_type="local",
 )
 
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()