Merge pull request #83 from trypear/feat/creator-mode-rebased

Creator mode

Author: nang-dev

.vscodeignore

@@ -29,6 +29,9 @@ jest.*
 .roomodes
 .rooignore
 .roo/**
+.pearai-agent/**
+.pearai-agent-ignore
+.pearai-agent-modes
 benchmark/**
 cline_docs/**
 e2e/**

package-lock.json

@@ -384,6 +384,7 @@
 		"@google/genai": "^0.9.0",
 		"@mistralai/mistralai": "^1.3.6",
 		"@modelcontextprotocol/sdk": "^1.7.0",
+		"@pearai/core": "file:./../pearai-submodule/core",
 		"@types/clone-deep": "^4.0.4",
 		"@types/pdf-parse": "^1.1.4",
 		"@types/tmp": "^0.2.6",

package.json

@@ -2,3 +2,4 @@ export { handleUri } from "./handleUri"
 export { registerCommands } from "./registerCommands"
 export { registerCodeActions } from "./registerCodeActions"
 export { registerTerminalActions } from "./registerTerminalActions"
+export { registerPearListener } from "./registerPearListener"

src/activate/index.ts

@@ -0,0 +1,115 @@
+import * as vscode from "vscode"
+import { ClineProvider } from "../core/webview/ClineProvider"
+import { assert } from "../utils/util"
+import { PEARAI_CREATOR_MODE_WEBAPP_MANAGER_SLUG } from "../shared/modes"
+
+export const getPearaiExtension = async () => {
+	const pearAiExtension = vscode.extensions.getExtension("pearai.pearai")
+
+	assert(!!pearAiExtension, "PearAI extension not found")
+
+	if (!pearAiExtension.isActive) {
+		await pearAiExtension.activate()
+	}
+
+	return pearAiExtension
+}
+
+// TODO: TYPES
+export const getpearAIExports = async () => {
+	const pearAiExtension = await getPearaiExtension()
+
+	assert(!!pearAiExtension.exports, "⚠️⚠️ Error, no PearAI Exports could be found :( ⚠️⚠️");
+
+	return pearAiExtension.exports;
+}
+ 
+// TODO: SHOULD HAVE TYPE SYNCED WITH THE PEARAI SUBMODULE!
+type CreatorModeState = "OVERLAY_CLOSED" | "OVERLAY_OPEN" | "OVERLAY_CLOSED_CREATOR_ACTIVE"
+
+export const registerPearListener = async (provider: ClineProvider) => {
+	// Getting the pear ai extension instance
+	const exports = await getpearAIExports()
+
+	exports.pearAPI.creatorMode.onDidRequestExecutePlan(async (msg: any) => {
+		console.dir(`onDidRequestNewTask triggered with: ${JSON.stringify(msg)}`)
+
+		let canContinue = false;
+
+		while(!canContinue) {
+			await new Promise((resolve) => setTimeout(resolve, 10));
+			canContinue = provider.viewLaunched && provider.isViewLaunched;
+		}
+
+
+		// Get the sidebar provider
+		// Focus the sidebar first
+		await vscode.commands.executeCommand("pearai-roo-cline.SidebarProvider.focus")
+
+		// Wait for the view to be ready using a helper function
+		await ensureViewIsReady(provider)
+		// Wait a brief moment for UI to update
+		await new Promise((resolve) => setTimeout(resolve, 3000))
+
+		// * This does actually work but the UI update does not happen. This method calls this.postStateToWebview() so not sure what is going on - James
+		if(msg.newProjectType === "WEBAPP") {
+			// Only switch to the creator manager if we're creating a new project
+			// TODO: later when we need to make a different type of project, we need to change this
+			await provider.handleModeSwitch(PEARAI_CREATOR_MODE_WEBAPP_MANAGER_SLUG);
+		}
+
+		// Clicl the chat btn
+		await provider.postMessageToWebview({ type: "action", action: "chatButtonClicked" })
+
+		const creatorModeConfig = {
+			creatorMode: true,
+			newProjectType: msg.newProjectType,
+			newProjectPath: msg.newProjectPath,
+		}
+
+
+		// Initialize with task
+		await provider.initClineWithTask(msg.plan, undefined, undefined, undefined, creatorModeConfig);
+	});
+	// If there's a creator event in the cache after the extensions were refreshed, we need to get it!
+	exports.pearAPI.creatorMode.triggerCachedCreatorEvent(true);
+
+	exports.pearAPI.creatorMode.onDidChangeCreatorModeState(async (state: CreatorModeState) => {
+		// Get the sidebar provider
+		const sidebarProvider = ClineProvider.getVisibleInstance();
+		
+		if (sidebarProvider) {
+			// Send a message to the webview that will trigger a window event
+			sidebarProvider.postMessageToWebview({
+				type: "creatorModeUpdate",
+				text: state,
+			});
+		}
+	});
+
+}
+
+// TODO: decide if this is needed
+// Helper function to ensure the webview is ready
+async function ensureViewIsReady(provider: ClineProvider): Promise<void> {
+	// If the view is already launched, we're good to go
+	if (provider.viewLaunched) {
+		return
+	}
+
+	// Otherwise, we need to wait for it to initialize
+	return new Promise<void>((resolve) => {
+		// Set up a one-time listener for when the view is ready
+		const disposable = provider.on("clineCreated", () => {
+			// Clean up the listener
+			disposable.dispose()
+			resolve()
+		})
+
+		// Set a timeout just in case
+		setTimeout(() => {
+			disposable.dispose()
+			resolve()
+		}, 5000)
+	})
+}

src/activate/registerPearListener.ts

@@ -105,9 +105,18 @@ export class AnthropicHandler extends BaseProvider implements SingleCompletionHa
 							case "claude-3-opus-20240229":
 							case "claude-3-haiku-20240307":
 								betas.push("prompt-caching-2024-07-31")
+								// Include prompt_key if newProjectType is set
 								return {
-									headers: { "anthropic-beta": betas.join(",") },
-									authorization: `Bearer ${this.options.apiKey}`,
+									headers: {
+										"anthropic-beta": betas.join(","),
+										prompt_key: this.options.creatorModeConfig?.newProjectType
+											? String(this.options.creatorModeConfig.newProjectType)
+											: undefined,
+										project_path: this.options.creatorModeConfig?.newProjectPath
+											? String(this.options.creatorModeConfig.newProjectPath)
+											: undefined,
+										authorization: `Bearer ${this.options.apiKey}`,
+									},
 								}
 							default:
 								return undefined

src/api/providers/anthropic.ts

@@ -116,28 +116,33 @@ export class PearAIHandler extends BaseProvider implements SingleCompletionHandl
 	}
 
 	async *createMessage(systemPrompt: string, messages: any[]): AsyncGenerator<any> {
-		const generator = this.handler.createMessage(systemPrompt, messages)
-		let warningMsg = ""
+		try {
+			const generator = this.handler.createMessage(systemPrompt, messages)
+			let warningMsg = ""
 
-		for await (const chunk of generator) {
-			console.dir(chunk)
-			if (chunk.type === "text" && chunk.metadata?.ui_only) {
-				warningMsg += chunk.metadata?.content
-				continue
+			for await (const chunk of generator) {
+				console.dir(chunk)
+				if (chunk.type === "text" && chunk.metadata?.ui_only) {
+					warningMsg += chunk.metadata?.content ?? ""
+					continue
+				}
+				yield chunk
 			}
-			yield chunk
-		}
 
-		if (warningMsg) {
-			if (warningMsg.includes("pay-as-you-go")) {
-				vscode.window.showInformationMessage(warningMsg, "View Pay-As-You-Go").then((selection) => {
-					if (selection === "View Pay-As-You-Go") {
-						vscode.env.openExternal(vscode.Uri.parse("https://trypear.ai/pay-as-you-go"))
-					}
-				})
-			} else {
-				vscode.window.showInformationMessage(warningMsg)
+			if (warningMsg) {
+				if (warningMsg.includes("pay-as-you-go")) {
+					vscode.window.showInformationMessage(warningMsg, "View Pay-As-You-Go").then((selection) => {
+						if (selection === "View Pay-As-You-Go") {
+							vscode.env.openExternal(vscode.Uri.parse("https://trypear.ai/pay-as-you-go"))
+						}
+					})
+				} else {
+					vscode.window.showInformationMessage(warningMsg)
+				}
 			}
+		} catch (e) {
+			const errorMessage = e instanceof Error ? e.message : String(e)
+			vscode.window.showWarningMessage(`Notice: ${errorMessage}`)
 		}
 	}