github/AutoGPT
1
1
Fork 0
mirror of https://github.com/Significant-Gravitas/AutoGPT.git synced 2026-04-08 03:00:28 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(rnd): Fix broken test

Browse Source
This commit is contained in:
Zamil Majdy
2024-08-13 17:51:59 +02:00
parent f30b2cdf25
commit 2b60a392fb
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
rnd/autogpt_server/autogpt_server/blocks/discord.py
View File

@@ -13,6 +13,9 @@ class DiscordReaderBlock(Block):
discord_bot_token: BlockSecret = SecretField(
key="discord_bot_token", description="Discord bot token"
)
continuous_read: bool = Field(
description="Whether to continuously read messages", default=True
)
class Output(BlockSchema):
message_content: str = Field(description="The content of the message received")
@@ -28,7 +31,7 @@ class DiscordReaderBlock(Block):
id="d3f4g5h6-1i2j-3k4l-5m6n-7o8p9q0r1s2t", # Unique ID for the node
input_schema=DiscordReaderBlock.Input, # Assign input schema
output_schema=DiscordReaderBlock.Output, # Assign output schema
test_input={"discord_bot_token": "test_token"},
test_input={"discord_bot_token": "test_token", "continuous_read": False},
test_output=[
(
"message_content",
@@ -81,6 +84,8 @@ class DiscordReaderBlock(Block):
while True:
for output_name, output_value in self.__run(input_data):
yield output_name, output_value
if not input_data.continuous_read:
break
def __run(self, input_data: "DiscordReaderBlock.Input") -> BlockOutput:
try: