From 406206f5d0ff8092725f5a4c782cb0dbb0812c62 Mon Sep 17 00:00:00 2001 From: SwiftyOS Date: Fri, 16 Aug 2024 10:11:30 +0200 Subject: [PATCH] Add categories to all the blocks --- rnd/autogpt_server/autogpt_server/blocks/basic.py | 2 +- rnd/autogpt_server/autogpt_server/blocks/branching.py | 2 +- rnd/autogpt_server/autogpt_server/blocks/discord.py | 2 +- rnd/autogpt_server/autogpt_server/blocks/iteration.py | 2 +- rnd/autogpt_server/autogpt_server/blocks/maths.py | 2 +- rnd/autogpt_server/autogpt_server/blocks/medium.py | 2 +- rnd/autogpt_server/autogpt_server/blocks/rss.py | 2 +- rnd/autogpt_server/autogpt_server/blocks/youtube.py | 3 +-- 8 files changed, 8 insertions(+), 9 deletions(-) diff --git a/rnd/autogpt_server/autogpt_server/blocks/basic.py b/rnd/autogpt_server/autogpt_server/blocks/basic.py index 332f202f8e..412b3063b1 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/basic.py +++ b/rnd/autogpt_server/autogpt_server/blocks/basic.py @@ -175,7 +175,7 @@ class InputBlock(ObjectLookupBase[Any]): class OutputBlock(ObjectLookupBase[Any]): def __init__(self): - super().__init__(categories={ BlockCategory.OUTPUT, BlockCategory.BASIC}) + super().__init__(categories={BlockCategory.OUTPUT, BlockCategory.BASIC}) def block_id(self) -> str: return "363ae599-353e-4804-937e-b2ee3cef3da4" diff --git a/rnd/autogpt_server/autogpt_server/blocks/branching.py b/rnd/autogpt_server/autogpt_server/blocks/branching.py index 87dfea3cdd..4081e6ef45 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/branching.py +++ b/rnd/autogpt_server/autogpt_server/blocks/branching.py @@ -1,7 +1,7 @@ from enum import Enum from typing import Any -from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory +from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema from autogpt_server.data.model import SchemaField diff --git a/rnd/autogpt_server/autogpt_server/blocks/discord.py b/rnd/autogpt_server/autogpt_server/blocks/discord.py index 8f27d948ff..f8b12bbc11 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/discord.py +++ b/rnd/autogpt_server/autogpt_server/blocks/discord.py @@ -4,7 +4,7 @@ import aiohttp import discord from pydantic import Field -from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory +from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema from autogpt_server.data.model import BlockSecret, SecretField diff --git a/rnd/autogpt_server/autogpt_server/blocks/iteration.py b/rnd/autogpt_server/autogpt_server/blocks/iteration.py index f38fb8e531..de77f46f16 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/iteration.py +++ b/rnd/autogpt_server/autogpt_server/blocks/iteration.py @@ -1,6 +1,6 @@ from typing import Any, List, Tuple -from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory +from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema from autogpt_server.data.model import SchemaField diff --git a/rnd/autogpt_server/autogpt_server/blocks/maths.py b/rnd/autogpt_server/autogpt_server/blocks/maths.py index a0ceb0db35..a7d8249bd8 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/maths.py +++ b/rnd/autogpt_server/autogpt_server/blocks/maths.py @@ -2,7 +2,7 @@ import operator from enum import Enum from typing import Any -from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory +from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema from autogpt_server.data.model import SchemaField diff --git a/rnd/autogpt_server/autogpt_server/blocks/medium.py b/rnd/autogpt_server/autogpt_server/blocks/medium.py index cc14dcc442..b1d91af3ea 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/medium.py +++ b/rnd/autogpt_server/autogpt_server/blocks/medium.py @@ -2,7 +2,7 @@ from typing import List import requests -from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory +from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema from autogpt_server.data.model import BlockSecret, SchemaField, SecretField diff --git a/rnd/autogpt_server/autogpt_server/blocks/rss.py b/rnd/autogpt_server/autogpt_server/blocks/rss.py index b5d8fe4db3..cdaada8740 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/rss.py +++ b/rnd/autogpt_server/autogpt_server/blocks/rss.py @@ -5,7 +5,7 @@ from typing import Any import feedparser import pydantic -from autogpt_server.data.block import Block, BlockOutput, BlockSchema, BlockCategory +from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema from autogpt_server.data.model import SchemaField diff --git a/rnd/autogpt_server/autogpt_server/blocks/youtube.py b/rnd/autogpt_server/autogpt_server/blocks/youtube.py index e2958be01e..344ac76100 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/youtube.py +++ b/rnd/autogpt_server/autogpt_server/blocks/youtube.py @@ -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, BlockCategory +from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema from autogpt_server.data.model import SchemaField @@ -27,7 +27,6 @@ class YouTubeTranscriberBlock(Block): 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"),