removed wordpress_auth

app.py

@@ -6,7 +6,6 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from tools.wordpress_auth import WordPressAuthTool
 from tools.wordpress_post import WordPressPostTool
 from tools.blog_generator import BlogGeneratorTool
 from tools.visit_webpage import VisitWebpageTool
@@ -148,14 +147,9 @@ with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
 
 # Initialize WordPress tools
-wordpress_auth = WordPressAuthTool()
-wordpress_post = WordPressPostTool(auth_tool=wordpress_auth)
+wordpress_post = WordPressPostTool(wordpress_credentials)
 blog_generator = BlogGeneratorTool(model=model)
 
-# Set WordPress credentials
-wordpress_auth.forward(action="set_credentials",
-                       credentials=wordpress_credentials)
-
 agent = CodeAgent(
     model=model,
     tools=[
@@ -164,7 +158,6 @@ agent = CodeAgent(
         image_generation_tool,
         quick_research,
         get_weather,
-        wordpress_auth,
         wordpress_post,
         blog_generator,
         visit_webpage,

tools/wordpress_auth.py

@@ -1,46 +0,0 @@
-from typing import Dict, Optional
-from smolagents.tools import Tool
-
-
-class WordPressAuthTool(Tool):
-    name = "wordpress_auth"
-    description = "Manages WordPress authentication credentials"
-    inputs = {
-        'action': {'type': 'string', 'description': 'Action to perform: set_credentials or get_credentials'},
-        'credentials': {
-            'type': 'object',
-            'description': 'WordPress credentials dictionary containing site_url, username, and app_password',
-            'nullable': True
-        }
-    }
-    output_type = "object"
-
-    def __init__(self):
-        super().__init__()
-        self._credentials: Optional[Dict] = None
-
-    def forward(self, action: str, credentials: Optional[Dict] = None) -> Dict:
-        """Manages WordPress credentials
-        Args:
-            action: Either 'set_credentials' or 'get_credentials'
-            credentials: Dict containing site_url, username, and app_password (only for set_credentials)
-        Returns:
-            Dict containing the current credentials or status message
-        """
-        if action == "set_credentials":
-            if not credentials:
-                return {"error": "Credentials required for set_credentials action"}
-
-            required_fields = ["site_url", "username", "app_password"]
-            if not all(field in credentials for field in required_fields):
-                return {"error": f"Missing required fields. Need: {required_fields}"}
-
-            self._credentials = credentials
-            return {"status": "Credentials stored successfully"}
-
-        elif action == "get_credentials":
-            if not self._credentials:
-                return {"error": "No credentials set. Please set credentials first."}
-            return self._credentials
-
-        return {"error": "Invalid action. Use 'set_credentials' or 'get_credentials'"}

tools/wordpress_post.py

@@ -18,9 +18,9 @@ class WordPressPostTool(Tool):
     }
     output_type = "object"
 
-    def __init__(self, auth_tool):
+    def __init__(self, wordpress_credentials):
         super().__init__()
-        self.auth_tool = auth_tool
+        self.credentials = wordpress_credentials
 
     def forward(self, title: str, content: str, status: str = "publish") -> Dict:
         """Publishes a post to WordPress
@@ -29,18 +29,14 @@ class WordPressPostTool(Tool):
             content: The content in HTML format
             status: Post status (publish, draft, private)
         Returns:
-            Dict containing status and post URL or error message
+            Dict containing the response from WordPress
         """
-        # Get WordPress credentials
-        creds = self.auth_tool.forward(action="get_credentials")
-        if "error" in creds:
-            return creds
+        if not self.credentials:
+            return {"error": "WordPress credentials not configured"}
 
-        url = f"{creds['site_url']}/wp-json/wp/v2/posts"
-
-        headers = {
-            "Content-Type": "application/json"
-        }
+        api_url = f"{self.credentials['site_url']}/wp-json/wp/v2/posts"
+        auth = HTTPBasicAuth(
+            self.credentials['username'], self.credentials['app_password'])
 
         data = {
             "title": title,
@@ -49,20 +45,8 @@ class WordPressPostTool(Tool):
         }
 
         try:
-            response = requests.post(
-                url,
-                json=data,
-                headers=headers,
-                auth=HTTPBasicAuth(creds['username'], creds['app_password'])
-            )
+            response = requests.post(api_url, json=data, auth=auth)
             response.raise_for_status()
-            return {
-                "status": "success",
-                "message": "Post published successfully!",
-                "url": response.json().get('link')
-            }
+            return {"status": "success", "post_id": response.json().get("id")}
         except requests.exceptions.RequestException as e:
-            return {
-                "status": "error",
-                "message": f"Error publishing post: {str(e)}"
-            }
+            return {"error": f"Failed to publish post: {str(e)}"}