refactor(app): pass thread to start()

Author: idiotWu

npi/app/discord/app.py

@@ -8,6 +8,7 @@
 from npi.config import config
 from npi.error.auth import UnauthorizedError
 from .schema import *
+from ...core.thread import Thread
 
 client = discord.Client(intents=discord.Intents.default())
 
@@ -47,8 +48,8 @@ def __init__(self, llm=None):
         self.client = discord.Client(intents=discord.Intents.default())
         self._access_token = cred.access_token
 
-    async def start(self):
-        await super().start()
+    async def start(self, thread: Thread = None):
+        await super().start(thread)
         await self.client.login(self._access_token)
 
     async def dispose(self):

npi/browser_app/twitter/app.py

@@ -45,6 +45,8 @@
     'home': 'https://twitter.com/home'
 }
 
+from ...core.thread import Thread
+
 
 class ImageFilterConverter(MarkdownConverter):
     def process_text(self, el):
@@ -117,9 +119,9 @@ def __init__(self, llm=None, headless: bool = True):
 
         self.register(Navigator(playwright=self.playwright))
 
-    async def start(self):
+    async def start(self, thread: Thread = None):
         if not self._started:
-            await super().start()
+            await super().start(thread)
             await self._login()
 
     async def _login(self):

npi/core/app.py

@@ -150,7 +150,7 @@ def __init__(
         self.fn_map = {}
         _register_tools(self)
 
-    async def start(self):
+    async def start(self, thread: Thread = None):
         """Start the app"""
         self._started = True
 

npi/core/browser_app.py

@@ -43,10 +43,10 @@ def __init__(
         self.use_screenshot = use_screenshot
         self.playwright = playwright or PlaywrightContext(headless)
 
-    async def start(self):
+    async def start(self, thread: Thread = None):
         """Start the Browser App"""
         if not self._started:
-            await super().start()
+            await super().start(thread)
             await self.playwright.start()
 
     async def dispose(self):

npi/server/app.py

@@ -161,7 +161,7 @@ async def run(self, thread: Thread):
         app = None
         try:
             app = self.get_app(thread.app_type)
-            await app.start()
+            await app.start(thread)
             result = await app.chat(thread.instruction, thread)
             thread.finish(result)
         except UnauthorizedError as e: