feat: Add welcome message and agent tools to populate form fields (#33)

Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

Author: 3 people

agentic_connector_builder_webapp/agentic_connector_builder_webapp.py

@@ -69,7 +69,12 @@ class ConnectorBuilderState(rx.State):
       email: email_address
 """
 
-    chat_messages: list[dict[str, str]] = []
+    chat_messages: list[dict[str, str]] = [
+        {
+            "role": "assistant",
+            "content": "Welcome! 👋\n\nWhat connector do you want to build today?",
+        }
+    ]
     chat_input: str = ""
     current_streaming_message: str = ""
     chat_loading: bool = False
@@ -200,6 +205,8 @@ async def send_message(self):
             documentation_urls=self.documentation_urls,
             functional_requirements=self.functional_requirements,
             test_list=self.test_list,
+            set_source_api_name=self.set_source_api_name,
+            set_connector_name=self.set_connector_name,
         )
 
         recent_messages = self.chat_messages[:-1][-HISTORY_MAX_MESSAGES:]

agentic_connector_builder_webapp/chat_agent.py

@@ -83,6 +83,20 @@ class SessionDeps:
     documentation_urls: str
     functional_requirements: str
     test_list: str
+from typing import Annotated, Any, Callable
+
+@dataclass
+class SessionDeps:
+    """Dependencies containing the current webapp session state."""
+    
+    yaml_content: str
+    connector_name: str
+    source_api_name: str
+    documentation_urls: str
+    functional_requirements: str
+    test_list: str
+    set_source_api_name: Callable[[str], Any] | None = None
+    set_connector_name: Callable[[str], Any] | None = None
 
 
 mcp_server = MCPServerStdio(
@@ -357,4 +371,64 @@ def replace_manifest_lines(
         except Exception as e:
             return f"Error replacing lines in manifest: {str(e)}"
 
+    @agent.tool
+    def set_api_name(
+        ctx: RunContext[SessionDeps],
+        api_name: Annotated[
+            str, Field(description="The name of the API (e.g., 'JSONPlaceholder API')")
+        ],
+    ) -> str:
+        """Set the Source API Name in the requirements form.
+
+        Use this tool when the user tells you what API they want to build a connector for.
+        This will populate the 'Source API name' field in the Define Requirements tab.
+
+        Args:
+            ctx: Runtime context with session dependencies
+            api_name: The name of the API
+
+        Returns:
+            A confirmation message indicating success.
+        """
+        try:
+            if ctx.deps.set_source_api_name:
+                ctx.deps.set_source_api_name(api_name)
+                return f"Successfully set the Source API Name to '{api_name}' in the requirements form."
+            else:
+                return "Error: Unable to update Source API Name (callback not available)"
+        except Exception as e:
+            return f"Error setting API name: {str(e)}"
+
+    @agent.tool
+    def set_connector_name(
+        ctx: RunContext[SessionDeps],
+        connector_name: Annotated[
+            str,
+            Field(
+                description="The connector name in the format 'source-{name}' (e.g., 'source-jsonplaceholder')"
+            ),
+        ],
+    ) -> str:
+        """Set the Connector Name in the requirements form.
+
+        Use this tool to populate the 'Connector name' field in the Define Requirements tab.
+        The connector name should follow the format 'source-{name}' where {name} is derived
+        from the API name (e.g., 'JSONPlaceholder API' -> 'source-jsonplaceholder').
+
+        Args:
+            ctx: Runtime context with session dependencies
+            connector_name: The connector name in source-{name} format
+
+        Returns:
+            A confirmation message indicating success.
+        """
+        try:
+            if ctx.deps.set_connector_name:
+                ctx.deps.set_connector_name(connector_name)
+                return f"Successfully set the Connector Name to '{connector_name}' in the requirements form."
+            else:
+                return "Error: Unable to update Connector Name (callback not available)"
+        except Exception as e:
+            return f"Error setting connector name: {str(e)}"
+
     return agent