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 crewai import Crew, Process
from langflow.base.agents.crewai.crew import BaseCrewComponent from langflow.base.agents.crewai.crew import BaseCrewComponent
from langflow.io import HandleInput from langflow.io import HandleInput, SecretStrInput
class HierarchicalCrewComponent(BaseCrewComponent): class HierarchicalCrewComponent(BaseCrewComponent):
@ -18,10 +20,21 @@ class HierarchicalCrewComponent(BaseCrewComponent):
HandleInput(name="tasks", display_name="Tasks", input_types=["HierarchicalTask"], is_list=True), 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_llm", display_name="Manager LLM", input_types=["LanguageModel"], required=False),
HandleInput(name="manager_agent", display_name="Manager Agent", input_types=["Agent"], 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: def build_crew(self) -> Crew:
tasks, agents = self.get_tasks_and_agents() 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( return Crew(
agents=agents, agents=agents,
tasks=tasks, tasks=tasks,