From 81b345bb56852681f368385a1dfeeac69e12f1a5 Mon Sep 17 00:00:00 2001
From: Maplemx <Maplemx@gmail.com>
Date: Thu, 24 Oct 2024 12:21:03 +0800
Subject: [PATCH] feat: ResponseGenerator

---
 .../agent_component/ResponseGenerator.py      | 77 +++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100644 Agently/plugins/agent_component/ResponseGenerator.py

diff --git a/Agently/plugins/agent_component/ResponseGenerator.py b/Agently/plugins/agent_component/ResponseGenerator.py
new file mode 100644
index 0000000..f3e59f2
--- /dev/null
+++ b/Agently/plugins/agent_component/ResponseGenerator.py
@@ -0,0 +1,77 @@
+import threading
+import asyncio
+import queue
+from .utils import ComponentABC
+
+class ResponseGenerator(ComponentABC):
+    def __init__(self, agent):
+        self.agent = agent
+        self.data_queue = queue.Queue()
+    
+    def put_data_to_generator(self, event, data):
+        self.data_queue.put((event, data))
+
+    def get_complete_generator(self):
+        thread = threading.Thread(target=self.agent.start)
+        thread.daemon = True
+        thread.start()
+        while True:
+            try:
+                item = self.data_queue.get_nowait()
+                if item == (None, None):
+                    break
+                yield item
+            except:
+                continue
+        thread.join()
+    
+    def get_realtime_generator(self):
+        self.agent.settings.set("use_realtime", True)
+        thread = threading.Thread(target=self.agent.start)
+        thread.daemon = True
+        thread.start()
+        while True:
+            try:
+                item = self.data_queue.get_nowait()
+                if item == (None, None):
+                    break
+                if item[0] == "realtime":
+                    yield item[1]
+            except:
+                continue
+        thread.join()
+    
+    def get_generator(self):
+        thread = threading.Thread(target=self.agent.start)
+        thread.daemon = True
+        thread.start()
+        while True:
+            try:
+                item = self.data_queue.get_nowait()
+                if item == (None, None):
+                    break
+                if not item[0].endswith(("_origin")):
+                    yield item
+            except:
+                continue
+        thread.join()
+
+    def _suffix(self, event, data):
+        if event != "response:finally":
+            self.put_data_to_generator(event, data)
+        else:
+            self.put_data_to_generator(None, None)
+    
+    def export(self):
+        return {
+            "suffix": self._suffix,
+            "alias": { 
+                "put_data_to_generator": { "func": self.put_data_to_generator },
+                "get_generator": { "func": self.get_generator, "return_value": True },
+                "get_realtime_generator": { "func": self.get_realtime_generator, "return_value": True },
+                "get_complete_generator": { "func": self.get_complete_generator, "return_value": True },
+            },
+        }
+
+def export():
+    return ("ResponseGenerator", ResponseGenerator)
\ No newline at end of file