feat(server, builder): OAuth support + API key management + GitHub blocks

docs/content/server/new_blocks.md

@@ -84,7 +84,7 @@ Follow these steps to create and test a new block:
 5. **Implement the `run` method with error handling:**, this should contain the main logic of the block:
 
    ```python
-   def run(self, input_data: Input) -> BlockOutput:
+   def run(self, input_data: Input, **kwargs) -> BlockOutput:
        try:
            topic = input_data.topic
            url = f"https://en.wikipedia.org/api/rest_v1/page/summary/{topic}"

rnd/autogpt_libs/autogpt_libs/supabase_integration_credentials_store/types.py

@@ -27,20 +27,29 @@ class OAuth2Credentials(_BaseCredentials):
     scopes: list[str]
     metadata: dict[str, Any] = Field(default_factory=dict)
 
+    def bearer(self) -> str:
+        return f"Bearer {self.access_token.get_secret_value()}"
+
 
 class APIKeyCredentials(_BaseCredentials):
     type: Literal["api_key"] = "api_key"
     api_key: SecretStr
     expires_at: Optional[int]
     """Unix timestamp (seconds) indicating when the API key expires (if at all)"""
 
+    def bearer(self) -> str:
+        return f"Bearer {self.api_key.get_secret_value()}"
+
 
 Credentials = Annotated[
     OAuth2Credentials | APIKeyCredentials,
     Field(discriminator="type"),
 ]
 
 
+CredentialsType = Literal["api_key", "oauth2"]
+
+
 class OAuthState(BaseModel):
     token: str
     provider: str

rnd/autogpt_server/autogpt_server/blocks/basic.py

@@ -57,7 +57,7 @@ def __init__(self):
             static_output=True,
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         yield "output", input_data.data or input_data.input
 
 
@@ -79,13 +79,12 @@ def __init__(self):
             test_output=("status", "printed"),
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         print(">>>>> Print: ", input_data.text)
         yield "status", "printed"
 
 
 class FindInDictionaryBlock(Block):
-
     class Input(BlockSchema):
         input: Any = Field(description="Dictionary to lookup from")
         key: str | int = Field(description="Key to lookup in the dictionary")
@@ -119,7 +118,7 @@ def __init__(self):
             categories={BlockCategory.BASIC},
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         obj = input_data.input
         key = input_data.key
 
@@ -201,7 +200,7 @@ def __init__(self):
             ui_type=BlockUIType.INPUT,
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         yield "result", input_data.value
 
 
@@ -284,7 +283,7 @@ def __init__(self):
             ui_type=BlockUIType.OUTPUT,
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         """
         Attempts to format the recorded_value using the fmt_string if provided.
         If formatting fails or no fmt_string is given, returns the original recorded_value.
@@ -344,7 +343,7 @@ def __init__(self):
             ],
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         try:
             # If no dictionary is provided, create a new one
             if input_data.dictionary is None:
@@ -415,7 +414,7 @@ def __init__(self):
             ],
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         try:
             # If no list is provided, create a new one
             if input_data.list is None:
@@ -456,5 +455,5 @@ def __init__(self):
             ui_type=BlockUIType.NOTE,
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         yield "output", input_data.text

rnd/autogpt_server/autogpt_server/blocks/block.py

@@ -31,7 +31,7 @@ def __init__(self):
             disabled=True,
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         code = input_data.code
 
         if search := re.search(r"class (\w+)\(Block\):", code):

rnd/autogpt_server/autogpt_server/blocks/branching.py

@@ -70,7 +70,7 @@ def __init__(self):
             ],
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         value1 = input_data.value1
         operator = input_data.operator
         value2 = input_data.value2

rnd/autogpt_server/autogpt_server/blocks/csv.py

@@ -32,7 +32,7 @@ def __init__(self):
             ],
         )
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         import csv
         from io import StringIO
 

rnd/autogpt_server/autogpt_server/blocks/discord.py

@@ -81,14 +81,14 @@ async def on_message(message):
 
         await client.start(token)
 
-    def run(self, input_data: "ReadDiscordMessagesBlock.Input") -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         while True:
             for output_name, output_value in self.__run(input_data):
                 yield output_name, output_value
             if not input_data.continuous_read:
                 break
 
-    def __run(self, input_data: "ReadDiscordMessagesBlock.Input") -> BlockOutput:
+    def __run(self, input_data: Input) -> BlockOutput:
         try:
             loop = asyncio.get_event_loop()
             future = self.run_bot(input_data.discord_bot_token.get_secret_value())
@@ -187,7 +187,7 @@ def chunk_message(self, message: str, limit: int = 2000) -> list:
         """Splits a message into chunks not exceeding the Discord limit."""
         return [message[i : i + limit] for i in range(0, len(message), limit)]
 
-    def run(self, input_data: "SendDiscordMessageBlock.Input") -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         try:
             loop = asyncio.get_event_loop()
             future = self.send_message(

rnd/autogpt_server/autogpt_server/blocks/email_block.py

@@ -88,7 +88,7 @@ def send_email(
         except Exception as e:
             return f"Failed to send email: {str(e)}"
 
-    def run(self, input_data: Input) -> BlockOutput:
+    def run(self, input_data: Input, **kwargs) -> BlockOutput:
         status = self.send_email(
             input_data.creds,
             input_data.to_email,