jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

fix: Pass the OpenAI API Key to Crew() (#4555)

Browse source
This commit is contained in:
Eric Hare 2024-11-12 16:05:27 -08:00 • committed by GitHub
commit 335b649093
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 14 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

15
src/backend/base/langflow/components/crewai/hierarchical_crew.py
View file

@ -1,7 +1,9 @@
import os
from crewai import Crew, Process
from langflow.base.agents.crewai.crew import BaseCrewComponent
from langflow.io import HandleInput
from langflow.io import HandleInput, SecretStrInput
class HierarchicalCrewComponent(BaseCrewComponent):
@ -18,10 +20,21 @@ class HierarchicalCrewComponent(BaseCrewComponent):
HandleInput(name="tasks", display_name="Tasks", input_types=["HierarchicalTask"], is_list=True),
HandleInput(name="manager_llm", display_name="Manager LLM", input_types=["LanguageModel"], required=False),
HandleInput(name="manager_agent", display_name="Manager Agent", input_types=["Agent"], required=False),
SecretStrInput(
name="openai_api_key",
display_name="OpenAI API Key",
info="The OpenAI API Key to use for the OpenAI model.",
value="OPENAI_API_KEY",
),
]
def build_crew(self) -> Crew:
tasks, agents = self.get_tasks_and_agents()
# Set the OpenAI API Key
if self.openai_api_key:
os.environ["OPENAI_API_KEY"] = self.openai_api_key
return Crew(
agents=agents,
tasks=tasks,