add workflow

.github/workflows/pipeline.yaml

@@ -0,0 +1,52 @@
+name: 'Build and Deploy - Bots'
+
+on:
+  workflow_dispatch:
+  push:
+    branches: [main]
+
+  pull_request:
+    branches: [main]
+
+env:
+  REGISTRY: ghcr.io
+  IMAGE_NAME: ${{ github.repository }}
+  ENV_STAGE: stage
+
+jobs:
+  build-and-push-to-ghcr:
+    name: Build
+    runs-on: ubuntu-20.04
+    permissions:
+      contents: read
+      packages: write
+    outputs:
+      tag: ${{ steps.prep.outputs.tag }}
+    steps:
+      - uses: actions/checkout@v2
+
+      - id: prep
+        run: |
+          TAG=$(echo $GITHUB_SHA | head -c7)
+          IMAGE="${{ env.REGISTRY }}/edenartlab/bots"
+          echo ::set-output name=tagged_image::${IMAGE}:${TAG}
+          echo ::set-output name=tag::${TAG}
+          echo ::set-output name=image::${IMAGE}
+
+      - name: Log in to GHCR
+        if: github.event_name != 'pull_request'
+        uses: docker/login-action@v1
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Build and Push
+        uses: docker/build-push-action@v2
+        with:
+          context: .
+          file: ./Dockerfile
+          push: true
+          tags:
+            ${{ steps.prep.outputs.tagged_image }},${{ steps.prep.outputs.image
+            }}:latest

src/bots/banny/.env.example

@@ -0,0 +1,9 @@
+DISCORD_TOKEN=
+MONGO_URI=
+MONGO_DB_NAME=
+EDEN_API_URL=
+EDEN_API_KEY=
+EDEN_API_SECRET=
+ALLOWED_GUILDS=
+ALLOWED_GUILDS_TEST=
+ALLOWED_CHANNELS=

src/bots/eden/.env.example

@@ -0,0 +1,10 @@
+DISCORD_TOKEN=
+MONGO_URI=
+MONGO_DB_NAME=
+EDEN_API_URL=
+EDEN_API_KEY=
+EDEN_API_SECRET=
+ALLOWED_GUILDS=
+ALLOWED_GUILDS_TEST=
+ALLOWED_CHANNELS=
+OPENAI_API_KEY=

src/cogs/AssistantCog.py

@@ -83,7 +83,7 @@ async def on_message(self, message: discord.Message) -> None:
                 reply_message = await message.reply(reply)
 
                 # check if there is a config
-                config = response["attachment"]
+                config = response["config"]
                 if not config:
                     return
 
@@ -97,12 +97,12 @@ async def on_message(self, message: discord.Message) -> None:
                     text_input = mode
 
                 if "init_image_data" in config:
-                    config["init_image_data"] = attachment_lookup_url[
+                    config["init_image_data"] = attachment_lookup_url.get(
                         config["init_image_data"]
-                    ]
+                    )
                 if "interpolation_init_images" in config:
                     config["interpolation_init_images"] = [
-                        attachment_lookup_url[img]
+                        attachment_lookup_url.get(img)
                         for img in config["interpolation_init_images"]
                     ]
 
@@ -129,7 +129,7 @@ async def on_message(self, message: discord.Message) -> None:
                     is_video_request=is_video_request,
                 )
                 await generation_loop(
-                    generation_loop_input, eden_credentials=self.sign_in_credentials
+                    generation_loop_input, eden_credentials=self.eden_credentials
                 )
 
     def message_preprocessor(self, message: discord.Message) -> str: