Run poe check

integrations-service/integrations/models/__init__.py

@@ -1,7 +1,8 @@
-from .dalle_image_generator import DalleImageGeneratorArguments, DalleImageGeneratorSetup
+from .dalle_image_generator import (
+    DalleImageGeneratorArguments,
+    DalleImageGeneratorSetup,
+)
 from .duckduckgo_search import DuckDuckGoSearchExecutionArguments
-from .weather import WeatherExecutionArguments, WeatherExecutionSetup
-from .wikipedia import WikipediaExecutionArguments
 from .hacker_news import HackerNewsExecutionArguments
 
 # TODO: Move these models somewhere else
@@ -12,3 +13,5 @@
     IntegrationExecutionRequest,
     IntegrationExecutionResponse,
 )
+from .weather import WeatherExecutionArguments, WeatherExecutionSetup
+from .wikipedia import WikipediaExecutionArguments

integrations-service/integrations/models/dalle_image_generator.py

@@ -1,7 +1,9 @@
 from pydantic import BaseModel, Field
 
+
 class DalleImageGeneratorSetup(BaseModel):
     api_key: str = Field(str, description="The API key for DALL-E")
 
+
 class DalleImageGeneratorArguments(BaseModel):
     prompt: str = Field(str, description="The image generation prompt")

integrations-service/integrations/models/hacker_news.py

@@ -2,6 +2,4 @@
 
 
 class HackerNewsExecutionArguments(BaseModel):
-    url: str = Field(
-        ..., description="The URL of the Hacker News thread to fetch"
-    )
+    url: str = Field(..., description="The URL of the Hacker News thread to fetch")

integrations-service/integrations/models/models.py

@@ -2,11 +2,14 @@
 
 from pydantic import BaseModel
 
-from .dalle_image_generator import DalleImageGeneratorArguments, DalleImageGeneratorSetup
+from .dalle_image_generator import (
+    DalleImageGeneratorArguments,
+    DalleImageGeneratorSetup,
+)
 from .duckduckgo_search import DuckDuckGoSearchExecutionArguments
-from .wikipedia import WikipediaExecutionArguments
-from .weather import WeatherExecutionSetup, WeatherExecutionArguments
 from .hacker_news import HackerNewsExecutionArguments
+from .weather import WeatherExecutionArguments, WeatherExecutionSetup
+from .wikipedia import WikipediaExecutionArguments
 
 ExecuteIntegrationArguments = Union[
     WikipediaExecutionArguments,

integrations-service/integrations/routers/execution/execute.py

@@ -8,7 +8,9 @@
 @router.post("/execute/{provider}", tags=["execution"])
 async def execute(
     provider: str = Path(..., description="The integration provider"),
-    request: IntegrationExecutionRequest = Body(..., description="The integration execution request")
+    request: IntegrationExecutionRequest = Body(
+        ..., description="The integration execution request"
+    ),
 ) -> IntegrationExecutionResponse:
     try:
         result = await execute_integration(provider, request.setup, request.arguments)

integrations-service/integrations/utils/execute_integration.py

@@ -1,12 +1,15 @@
 from ..models import ExecuteIntegrationArguments, ExecuteIntegrationSetup
 from .integrations.dalle_image_generator import dalle_image_generator
 from .integrations.duckduckgo_search import duckduckgo_search
-from .integrations.wikipedia import wikipedia
-from .integrations.weather import weather
 from .integrations.hacker_news import hacker_news
+from .integrations.weather import weather
+from .integrations.wikipedia import wikipedia
+
 
 async def execute_integration(
-    provider: str, setup: ExecuteIntegrationSetup | None, arguments: ExecuteIntegrationArguments
+    provider: str,
+    setup: ExecuteIntegrationSetup | None,
+    arguments: ExecuteIntegrationArguments,
 ) -> str:
     match provider:
         case "duckduckgo_search":

integrations-service/integrations/utils/integrations/gmail/send_mail.py

@@ -3,4 +3,4 @@ async def send_mail(arguments: dict) -> str:
     Dummy integration for sending an email to a specified recipient with a given subject and message.
     """
 
-    return "Mail sent"
+    return "Mail sent"