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

added categories

Browse Source
This commit is contained in:
SwiftyOS
2024-08-14 11:34:10 +02:00
parent 8643f70cf2
commit 8d241acd6d
10 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
rnd/autogpt_server/autogpt_server/blocks/basic.py
View File

@@ -167,7 +167,7 @@ class ObjectLookupBlock(ObjectLookupBase[Any]):
class InputBlock(ObjectLookupBase[Any]):
def __init__(self):
super().__init__(categories={BlockCategory.BASIC, BlockCategory.INPUT})
super().__init__(categories={BlockCategory.INPUT, BlockCategory.BASIC})
def block_id(self) -> str:
return "c0a8e994-ebf1-4a9c-a4d8-89d09c86741b"
@@ -175,7 +175,7 @@ class InputBlock(ObjectLookupBase[Any]):
class OutputBlock(ObjectLookupBase[Any]):
def __init__(self):
super().__init__(categories={BlockCategory.BASIC, BlockCategory.OUTPUT})
super().__init__(categories={ BlockCategory.OUTPUT, BlockCategory.BASIC})
def block_id(self) -> str:
return "363ae599-353e-4804-937e-b2ee3cef3da4"

3
rnd/autogpt_server/autogpt_server/blocks/branching.py
View File

@@ -1,7 +1,7 @@
from enum import Enum
from typing import Any
from autogpt_server.data.block import Block, BlockOutput, BlockSchema
from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory
from autogpt_server.data.model import SchemaField
@@ -56,6 +56,7 @@ class ConditionBlock(Block):
input_schema=ConditionBlock.Input,
output_schema=ConditionBlock.Output,
description="Handles conditional logic based on comparison operators",
categories={BlockCategory.LOGIC},
test_input={
"value1": 10,
"operator": ComparisonOperator.GREATER_THAN.value,

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

@@ -4,7 +4,7 @@ import aiohttp
import discord
from pydantic import Field
from autogpt_server.data.block import Block, BlockOutput, BlockSchema
from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory
from autogpt_server.data.model import BlockSecret, SecretField
@@ -31,6 +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
categories={BlockCategory.SOCIAL},
test_input={"discord_bot_token": "test_token", "continuous_read": False},
test_output=[
(
@@ -147,6 +148,7 @@ class DiscordMessageSenderBlock(Block):
id="h1i2j3k4-5l6m-7n8o-9p0q-r1s2t3u4v5w6", # Unique ID for the node
input_schema=DiscordMessageSenderBlock.Input, # Assign input schema
output_schema=DiscordMessageSenderBlock.Output, # Assign output schema
categories={BlockCategory.SOCIAL},
test_input={
"discord_bot_token": "YOUR_DISCORD_BOT_TOKEN",
"channel_name": "general",

2
rnd/autogpt_server/autogpt_server/blocks/email_block.py
View File

@@ -45,7 +45,7 @@ class SendEmailBlock(Block):
super().__init__(
id="a1234567-89ab-cdef-0123-456789abcdef",
description="This block sends an email using the provided SMTP credentials.",
categories={BlockCategory.TEXT},
categories={BlockCategory.OUTPUT},
input_schema=SendEmailBlock.Input,
output_schema=SendEmailBlock.Output,
test_input={

2
rnd/autogpt_server/autogpt_server/blocks/http.py
View File

@@ -32,7 +32,7 @@ class HttpRequestBlock(Block):
super().__init__(
id="6595ae1f-b924-42cb-9a41-551a0611c4b4",
description="This block makes an HTTP request to the given URL.",
categories={BlockCategory.BASIC},
categories={BlockCategory.INPUT},
input_schema=HttpRequestBlock.Input,
output_schema=HttpRequestBlock.Output,
)

3
rnd/autogpt_server/autogpt_server/blocks/iteration.py
View File

@@ -1,6 +1,6 @@
from typing import Any, List, Tuple
from autogpt_server.data.block import Block, BlockOutput, BlockSchema
from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory
from autogpt_server.data.model import SchemaField
@@ -21,6 +21,7 @@ class ForEachBlock(Block):
id="f8e7d6c5-b4a3-2c1d-0e9f-8g7h6i5j4k3l",
input_schema=ForEachBlock.Input,
output_schema=ForEachBlock.Output,
categories={BlockCategory.LOGIC},
test_input={"items": [1, "two", {"three": 3}, [4, 5]]},
test_output=[
("item", (0, 1)),

4
rnd/autogpt_server/autogpt_server/blocks/maths.py
View File

@@ -2,7 +2,7 @@ import operator
from enum import Enum
from typing import Any
from autogpt_server.data.block import Block, BlockOutput, BlockSchema
from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory
from autogpt_server.data.model import SchemaField
@@ -39,6 +39,7 @@ class MathsBlock(Block):
id="b1ab9b19-67a6-406d-abf5-2dba76d00c79",
input_schema=MathsBlock.Input,
output_schema=MathsBlock.Output,
categories={BlockCategory.LOGIC},
test_input={
"operation": Operation.ADD.value,
"a": 10.0,
@@ -97,6 +98,7 @@ class CounterBlock(Block):
id="3c9c2f42-b0c3-435f-ba35-05f7a25c772a",
input_schema=CounterBlock.Input,
output_schema=CounterBlock.Output,
categories={BlockCategory.LOGIC},
test_input={"collection": [1, 2, 3, 4, 5]},
test_output=[
("count", 5),

3
rnd/autogpt_server/autogpt_server/blocks/medium.py
View File

@@ -2,7 +2,7 @@ from typing import List
import requests
from autogpt_server.data.block import Block, BlockOutput, BlockSchema
from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory
from autogpt_server.data.model import BlockSecret, SchemaField, SecretField
@@ -70,6 +70,7 @@ class CreateMediumPostBlock(Block):
id="3f7b2dcb-4a78-4e3f-b0f1-88132e1b89df",
input_schema=CreateMediumPostBlock.Input,
output_schema=CreateMediumPostBlock.Output,
categories={BlockCategory.SOCIAL},
test_input={
"author_id": "1234567890abcdef",
"title": "Test Post",

3
rnd/autogpt_server/autogpt_server/blocks/rss.py
View File

@@ -5,7 +5,7 @@ from typing import Any
import feedparser
import pydantic
from autogpt_server.data.block import Block, BlockOutput, BlockSchema
from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory
from autogpt_server.data.model import SchemaField
@@ -46,6 +46,7 @@ class RSSReaderBlock(Block):
id="c6731acb-4105-4zp1-bc9b-03d0036h370g",
input_schema=RSSReaderBlock.Input,
output_schema=RSSReaderBlock.Output,
categories={BlockCategory.OUTPUT},
test_input={
"rss_url": "https://example.com/rss",
"time_period": 10_000_000,

4
rnd/autogpt_server/autogpt_server/blocks/youtube.py
View File

@@ -3,7 +3,7 @@ from urllib.parse import parse_qs, urlparse
from youtube_transcript_api import YouTubeTranscriptApi
from youtube_transcript_api.formatters import TextFormatter
from autogpt_server.data.block import Block, BlockOutput, BlockSchema
from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory
from autogpt_server.data.model import SchemaField
@@ -26,6 +26,8 @@ class YouTubeTranscriberBlock(Block):
id="f3a8f7e1-4b1d-4e5f-9f2a-7c3d5a2e6b4c",
input_schema=YouTubeTranscriberBlock.Input,
output_schema=YouTubeTranscriberBlock.Output,
categories={BlockCategory.SOCIAL},
test_input={"youtube_url": "https://www.youtube.com/watch?v=dQw4w9WgXcQ"},
test_output=[
("video_id", "dQw4w9WgXcQ"),