Skip to content

Commit 040f32a

Browse files
committed
Add a method named full_messages_for_ask to ChatAIStream.
1 parent d4658bf commit 040f32a

File tree

5 files changed

+14
-8
lines changed

5 files changed

+14
-8
lines changed

README.md

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -170,6 +170,9 @@ Message broker between YouTube chat stream and ChatGPT.
170170
- Internal process will be terminated soon after.
171171
- No arguments required, nothing returns.
172172

173+
### full_messages_for_ask()
174+
- Indicate whether the queue which spools messages to send ChatAI is full or not.
175+
173176
And other [threading.Thread](https://docs.python.org/3/library/threading.html) public pethods are available.
174177

175178
## Callbacks

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,2 @@
1-
streamchat-agent==1.0.1
2-
chatai-agent==1.0.0
1+
streamchat-agent==1.0.2
2+
chatai-agent==1.1.0

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ def _requires_from_file(filename):
1212

1313
setup(
1414
name="chatai-stream",
15-
version="0.0.3",
15+
version="1.0.0",
1616
license="MIT",
1717
description="ChatGPT reacts YouTube chat messages.",
1818
author="General Yadoc",

src/ChatAIStream.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,31 +15,34 @@ class params():
1515
ai_params: aiParams
1616

1717
class ChatAIStream(threading.Thread):
18-
def my_pre_filter_cb(self, c):
18+
def __my_pre_filter_cb(self, c):
1919
prefiltered_c = c
2020
if prefiltered_c and self.pre_filter_cb:
2121
prefiltered_c = self.pre_filter_cb(prefiltered_c)
2222
prefiltered_c.message = re.sub(r':[^:]+:', ".", prefiltered_c.message)
2323
prefiltered_c.message = re.sub(r'^[\.]+', "", prefiltered_c.message)
2424
return None if prefiltered_c.message == "" else prefiltered_c
2525

26-
def ask_stream_message_to_ai(self, c):
26+
def __ask_stream_message_to_ai(self, c):
2727
if self.get_stream_message_cb:
2828
self.get_stream_message_cb(c)
2929
if self.ai_agent:
3030
self.ai_agent.put_message(ca.userMessage(message=c.message, extern=c))
3131

3232
def __init__( self, params):
3333
self.get_stream_message_cb=params.stream_params.get_item_cb
34-
params.stream_params.get_item_cb=self.ask_stream_message_to_ai
34+
params.stream_params.get_item_cb=self.__ask_stream_message_to_ai
3535
self.pre_filter_cb=params.stream_params.pre_filter_cb
36-
params.stream_params.pre_filter_cb=self.my_pre_filter_cb
36+
params.stream_params.pre_filter_cb=self.__my_pre_filter_cb
3737

3838
self.ai_agent = ca.ChatAIAgent( params.ai_params )
3939
self.stream_agent = sca.StreamChatAgent( params.stream_params )
4040

4141
super(ChatAIStream, self).__init__(daemon=True)
4242

43+
def full_messages_for_ask(self):
44+
return self.ai_agent.full_messages()
45+
4346
def run(self):
4447
self.stream_agent.start()
4548
self.ai_agent.start()

src/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
from .ChatAIStream import *
22

3-
__version__ = '0.0.3'
3+
__version__ = '1.0.0'

0 commit comments

Comments
 (0)