github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-09 21:27:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

add more info to appFinished event for extension

Browse Source
This commit is contained in:
LeonOstrez
2024-06-25 16:59:13 +01:00
parent b78ffacb47
commit 3fb38cdbf3
5 changed files with 84 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
core/agents/task_completer.py
View File

@@ -38,8 +38,16 @@ class TaskCompleter(BaseAgent):
if current_task_index1 == len(tasks):
if source == "app":
await self.ui.send_app_finished()
await self.ui.send_app_finished(
app_id=str(self.state_manager.project.id),
app_name=self.state_manager.project.name,
folder_name=self.state_manager.project.folder_name,
)
elif source == "feature":
await self.ui.send_feature_finished()
await self.ui.send_feature_finished(
app_id=str(self.state_manager.project.id),
app_name=self.state_manager.project.name,
folder_name=self.state_manager.project.folder_name,
)
return AgentResponse.done(self)

22
core/ui/base.py
View File

@@ -121,15 +121,33 @@ class UIBase:
"""
raise NotImplementedError()
async def send_app_finished(self):
async def send_app_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
"""
Send the app finished message.
:param app_id: App ID.
:param app_name: App name.
:param folder_name: Folder name.
"""
raise NotImplementedError()
async def send_feature_finished(self):
async def send_feature_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
"""
Send the feature finished message.
:param app_id: App ID.
:param app_name: App name.
:param folder_name: Folder name.
"""
raise NotImplementedError()

16
core/ui/console.py
View File

@@ -33,14 +33,24 @@ class PlainConsoleUI(UIBase):
else:
print(message)
async def send_key_expired(self, message: Optional[str]):
async def send_key_expired(self, message: Optional[str] = None):
if message:
await self.send_message(message)
async def send_app_finished(self):
async def send_app_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
pass
async def send_feature_finished(self):
async def send_feature_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
pass
async def ask_question(

32
core/ui/ipc_client.py
View File

@@ -199,11 +199,35 @@ class IPCClientUI(UIBase):
async def send_key_expired(self, message: Optional[str] = None):
await self._send(MessageType.KEY_EXPIRED)
async def send_app_finished(self):
await self._send(MessageType.APP_FINISHED)
async def send_app_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
await self._send(
MessageType.APP_FINISHED,
content={
"app_id": app_id,
"app_name": app_name,
"folder_name": folder_name,
},
)
async def send_feature_finished(self):
await self._send(MessageType.FEATURE_FINISHED)
async def send_feature_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
await self._send(
MessageType.FEATURE_FINISHED,
content={
"app_id": app_id,
"app_name": app_name,
"folder_name": folder_name,
},
)
async def ask_question(
self,

16
core/ui/virtual.py
View File

@@ -34,13 +34,23 @@ class VirtualUI(UIBase):
else:
print(message)
async def send_key_expired(self, message: Optional[str]):
async def send_key_expired(self, message: Optional[str] = None):
pass
async def send_app_finished(self):
async def send_app_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
pass
async def send_feature_finished(self):
async def send_feature_finished(
self,
app_id: Optional[str] = None,
app_name: Optional[str] = None,
folder_name: Optional[str] = None,
):
pass
async def ask_question(