Create tool for generating automations in Studio (#438)

* Create tool for generating automations in Studio

This commit creates a tool to use CrewAI Enterprise API to generate
crews using CrewAI Studio.

* Replace CREWAI_BASE_URL with CREWAI_PLUS_URL

* Add missing /crewai_plus in URL

src/crewai_tools/__init__.py

@@ -12,16 +12,16 @@ from .tools import (
     ApifyActorsTool,
     ArxivPaperTool,
     BraveSearchTool,
-    BrightDataWebUnlockerTool,
-    BrightDataSearchTool,
     BrightDataDatasetTool,
+    BrightDataSearchTool,
+    BrightDataWebUnlockerTool,
     BrowserbaseLoadTool,
     CodeDocsSearchTool,
     CodeInterpreterTool,
     ComposioTool,
-    ContextualAIQueryTool,
     ContextualAICreateAgentTool,
     ContextualAIParseTool,
+    ContextualAIQueryTool,
     ContextualAIRerankTool,
     CouchbaseFTSVectorSearchTool,
     CrewaiEnterpriseTools,
@@ -38,6 +38,7 @@ from .tools import (
     FirecrawlCrawlWebsiteTool,
     FirecrawlScrapeWebsiteTool,
     FirecrawlSearchTool,
+    GenerateCrewaiAutomationTool,
     GithubSearchTool,
     HyperbrowserLoadTool,
     InvokeCrewAIAutomationTool,

src/crewai_tools/tools/__init__.py

@@ -2,13 +2,20 @@ from .ai_mind_tool.ai_mind_tool import AIMindTool
 from .apify_actors_tool.apify_actors_tool import ApifyActorsTool
 from .arxiv_paper_tool.arxiv_paper_tool import ArxivPaperTool
 from .brave_search_tool.brave_search_tool import BraveSearchTool
+from .brightdata_tool import (
+    BrightDataDatasetTool,
+    BrightDataSearchTool,
+    BrightDataWebUnlockerTool,
+)
 from .browserbase_load_tool.browserbase_load_tool import BrowserbaseLoadTool
 from .code_docs_search_tool.code_docs_search_tool import CodeDocsSearchTool
 from .code_interpreter_tool.code_interpreter_tool import CodeInterpreterTool
 from .composio_tool.composio_tool import ComposioTool
-from .contextualai_query_tool.contextual_query_tool import ContextualAIQueryTool
-from .contextualai_create_agent_tool.contextual_create_agent_tool import ContextualAICreateAgentTool
+from .contextualai_create_agent_tool.contextual_create_agent_tool import (
+    ContextualAICreateAgentTool,
+)
 from .contextualai_parse_tool.contextual_parse_tool import ContextualAIParseTool
+from .contextualai_query_tool.contextual_query_tool import ContextualAIQueryTool
 from .contextualai_rerank_tool.contextual_rerank_tool import ContextualAIRerankTool
 from .couchbase_tool.couchbase_tool import CouchbaseFTSVectorSearchTool
 from .crewai_enterprise_tools.crewai_enterprise_tools import CrewaiEnterpriseTools
@@ -29,9 +36,14 @@ from .firecrawl_scrape_website_tool.firecrawl_scrape_website_tool import (
     FirecrawlScrapeWebsiteTool,
 )
 from .firecrawl_search_tool.firecrawl_search_tool import FirecrawlSearchTool
+from .generate_crewai_automation_tool.generate_crewai_automation_tool import (
+    GenerateCrewaiAutomationTool,
+)
 from .github_search_tool.github_search_tool import GithubSearchTool
 from .hyperbrowser_load_tool.hyperbrowser_load_tool import HyperbrowserLoadTool
-from .invoke_crewai_automation_tool.invoke_crewai_automation_tool import InvokeCrewAIAutomationTool
+from .invoke_crewai_automation_tool.invoke_crewai_automation_tool import (
+    InvokeCrewAIAutomationTool,
+)
 from .json_search_tool.json_search_tool import JSONSearchTool
 from .linkup.linkup_search_tool import LinkupSearchTool
 from .llamaindex_tool.llamaindex_tool import LlamaIndexTool
@@ -108,9 +120,4 @@ from .youtube_channel_search_tool.youtube_channel_search_tool import (
     YoutubeChannelSearchTool,
 )
 from .youtube_video_search_tool.youtube_video_search_tool import YoutubeVideoSearchTool
-from .brightdata_tool import (
-    BrightDataDatasetTool,
-    BrightDataSearchTool,
-    BrightDataWebUnlockerTool
-)
 from .zapier_action_tool.zapier_action_tool import ZapierActionTools

src/crewai_tools/tools/generate_crewai_automation_tool/README.md

@@ -0,0 +1,50 @@
+# GenerateCrewaiAutomationTool
+
+## Description
+
+The GenerateCrewaiAutomationTool integrates with CrewAI Studio API to generate complete CrewAI automations from natural language descriptions. It translates high-level requirements into functional CrewAI implementations and returns direct links to Studio projects.
+
+## Environment Variables
+
+Set your CrewAI Personal Access Token (CrewAI Enterprise > Settings > Account > Personal Access Token):
+
+```bash
+export CREWAI_PERSONAL_ACCESS_TOKEN="your_personal_access_token_here"
+export CREWAI_PLUS_URL="https://app.crewai.com"  # optional
+```
+
+## Example
+
+```python
+from crewai_tools import GenerateCrewaiAutomationTool
+from crewai import Agent, Task, Crew
+
+# Initialize tool
+tool = GenerateCrewaiAutomationTool()
+
+# Generate automation
+result = tool.run(
+    prompt="Generate a CrewAI automation that scrapes websites and stores data in a database",
+    organization_id="org_123"  # optional but recommended
+)
+
+print(result)
+# Output: Generated CrewAI Studio project URL: https://studio.crewai.com/project/abc123
+
+# Use with agent
+agent = Agent(
+    role="Automation Architect",
+    goal="Generate CrewAI automations",
+    backstory="Expert at creating automated workflows",
+    tools=[tool]
+)
+
+task = Task(
+    description="Create a lead qualification automation",
+    agent=agent,
+    expected_output="Studio project URL"
+)
+
+crew = Crew(agents=[agent], tasks=[task])
+result = crew.kickoff()
+```

src/crewai_tools/tools/generate_crewai_automation_tool/generate_crewai_automation_tool.py

@@ -0,0 +1,70 @@
+import os
+from typing import List, Optional, Type
+
+import requests
+from crewai.tools import BaseTool, EnvVar
+from pydantic import BaseModel, Field
+
+
+class GenerateCrewaiAutomationToolSchema(BaseModel):
+    prompt: str = Field(
+        description="The prompt to generate the CrewAI automation, e.g. 'Generate a CrewAI automation that will scrape the website and store the data in a database.'"
+    )
+    organization_id: Optional[str] = Field(
+        default=None,
+        description="The identifier for the CrewAI Enterprise organization. If not specified, a default organization will be used.",
+    )
+
+
+class GenerateCrewaiAutomationTool(BaseTool):
+    name: str = "Generate CrewAI Automation"
+    description: str = (
+        "A tool that leverages CrewAI Studio's capabilities to automatically generate complete CrewAI "
+        "automations based on natural language descriptions. It translates high-level requirements into "
+        "functional CrewAI implementations."
+    )
+    args_schema: Type[BaseModel] = GenerateCrewaiAutomationToolSchema
+    crewai_enterprise_url: str = Field(
+        default_factory=lambda: os.getenv("CREWAI_PLUS_URL", "https://app.crewai.com"),
+        description="The base URL of CrewAI Enterprise. If not provided, it will be loaded from the environment variable CREWAI_PLUS_URL with default https://app.crewai.com.",
+    )
+    personal_access_token: Optional[str] = Field(
+        default_factory=lambda: os.getenv("CREWAI_PERSONAL_ACCESS_TOKEN"),
+        description="The user's Personal Access Token to access CrewAI Enterprise API. If not provided, it will be loaded from the environment variable CREWAI_PERSONAL_ACCESS_TOKEN.",
+    )
+    env_vars: List[EnvVar] = [
+        EnvVar(
+            name="CREWAI_PERSONAL_ACCESS_TOKEN",
+            description="Personal Access Token for CrewAI Enterprise API",
+            required=True,
+        ),
+        EnvVar(
+            name="CREWAI_PLUS_URL",
+            description="Base URL for CrewAI Enterprise API",
+            required=False,
+        ),
+    ]
+
+    def _run(self, **kwargs) -> str:
+        input_data = GenerateCrewaiAutomationToolSchema(**kwargs)
+        response = requests.post(
+            f"{self.crewai_enterprise_url}/crewai_plus/api/v1/studio",
+            headers=self._get_headers(input_data.organization_id),
+            json={"prompt": input_data.prompt},
+        )
+
+        response.raise_for_status()
+        studio_project_url = response.json().get("url")
+        return f"Generated CrewAI Studio project URL: {studio_project_url}"
+
+    def _get_headers(self, organization_id: Optional[str] = None) -> dict:
+        headers = {
+            "Authorization": f"Bearer {self.personal_access_token}",
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+        }
+
+        if organization_id:
+            headers["X-Crewai-Organization-Id"] = organization_id
+
+        return headers