Add categories to all the blocks

pull/7802/head
SwiftyOS 2024-08-16 10:11:30 +02:00
parent 1e05d6a8e9
commit 406206f5d0
8 changed files with 8 additions and 9 deletions

View File

@ -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"

View File

@ -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

View File

@ -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

View File

@ -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

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, BlockCategory
from autogpt_server.data.block import Block, BlockCategory, BlockOutput, BlockSchema
from autogpt_server.data.model import SchemaField

View File

@ -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

View File

@ -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

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, 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"),