Update .env.sample, remove image_url from context of image generation

video-db · Oct 23, 2024 · 6930f32 · 6930f32
1 parent 9150c03
commit 6930f32
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/backend/.env.sample b/backend/.env.sample
@@ -1,9 +1,16 @@
+# VideoDB Integration
+VIDEO_DB_API_KEY=
 
-# OpenAI config
+# LLM Integrations
 OPENAI_API_KEY=
 
-# VideoDB config
-VIDEO_DB_API_KEY=
-# only for dev
-VIDEO_DB_BASE_URL=
+# Tools
+REPLICATE_API_TOKEN=
+
+# Brandkit Agent
+INTRO_VIDEO_ID=
+OUTRO_VIDEO_ID=
+BRAND_IMAGE_ID=
 
+# Profanity Remover Agent
+BEEP_AUDIO_ID=
diff --git a/backend/spielberg/agents/image_generation.py b/backend/spielberg/agents/image_generation.py
@@ -55,5 +55,5 @@ def run(self, prompt: str, *args, **kwargs) -> AgentResponse:
         return AgentResponse(
             status=AgentStatus.SUCCESS,
             message=f"Agent {self.name} completed successfully.",
-            data={"image_url": image_url},
+            data={},
         )