Add categories to all the blocks
parent
1e05d6a8e9
commit
406206f5d0
|
@ -175,7 +175,7 @@ class InputBlock(ObjectLookupBase[Any]):
|
||||||
|
|
||||||
class OutputBlock(ObjectLookupBase[Any]):
|
class OutputBlock(ObjectLookupBase[Any]):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
super().__init__(categories={ BlockCategory.OUTPUT, BlockCategory.BASIC})
|
super().__init__(categories={BlockCategory.OUTPUT, BlockCategory.BASIC})
|
||||||
|
|
||||||
def block_id(self) -> str:
|
def block_id(self) -> str:
|
||||||
return "363ae599-353e-4804-937e-b2ee3cef3da4"
|
return "363ae599-353e-4804-937e-b2ee3cef3da4"
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
from typing import Any
|
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
|
from autogpt_server.data.model import SchemaField
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -4,7 +4,7 @@ import aiohttp
|
||||||
import discord
|
import discord
|
||||||
from pydantic import Field
|
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
|
from autogpt_server.data.model import BlockSecret, SecretField
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
from typing import Any, List, Tuple
|
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
|
from autogpt_server.data.model import SchemaField
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -2,7 +2,7 @@ import operator
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
from typing import Any
|
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
|
from autogpt_server.data.model import SchemaField
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -2,7 +2,7 @@ from typing import List
|
||||||
|
|
||||||
import requests
|
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
|
from autogpt_server.data.model import BlockSecret, SchemaField, SecretField
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -5,7 +5,7 @@ from typing import Any
|
||||||
import feedparser
|
import feedparser
|
||||||
import pydantic
|
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
|
from autogpt_server.data.model import SchemaField
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -3,7 +3,7 @@ from urllib.parse import parse_qs, urlparse
|
||||||
from youtube_transcript_api import YouTubeTranscriptApi
|
from youtube_transcript_api import YouTubeTranscriptApi
|
||||||
from youtube_transcript_api.formatters import TextFormatter
|
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
|
from autogpt_server.data.model import SchemaField
|
||||||
|
|
||||||
|
|
||||||
|
@ -27,7 +27,6 @@ class YouTubeTranscriberBlock(Block):
|
||||||
input_schema=YouTubeTranscriberBlock.Input,
|
input_schema=YouTubeTranscriberBlock.Input,
|
||||||
output_schema=YouTubeTranscriberBlock.Output,
|
output_schema=YouTubeTranscriberBlock.Output,
|
||||||
categories={BlockCategory.SOCIAL},
|
categories={BlockCategory.SOCIAL},
|
||||||
|
|
||||||
test_input={"youtube_url": "https://www.youtube.com/watch?v=dQw4w9WgXcQ"},
|
test_input={"youtube_url": "https://www.youtube.com/watch?v=dQw4w9WgXcQ"},
|
||||||
test_output=[
|
test_output=[
|
||||||
("video_id", "dQw4w9WgXcQ"),
|
("video_id", "dQw4w9WgXcQ"),
|
||||||
|
|
Loading…
Reference in New Issue