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

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

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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