Cmuroc27 commited on
Commit
9e6b745
·
1 Parent(s): 35c3ec3

para las decisions

Browse files
Files changed (1) hide show
  1. agents.py +3 -1
agents.py CHANGED
@@ -1,4 +1,5 @@
1
  import asyncio
 
2
  from llama_index.core.agent.workflow import AgentWorkflow
3
  from llama_index.llms.openai import OpenAI
4
  from tools import (image_analyzer_tool, youtube_transcript_tool, calculator_tool, read_document_tool, search_tool)
@@ -6,10 +7,11 @@ from llama_index.core.workflow import (Workflow, Context, step, StartEvent, Stop
6
  from llama_index.core.agent import ReActAgent
7
  from dotenv import load_dotenv
8
  load_dotenv()
9
-
10
  # Alfred React Agent
11
  class AlfredAdvancedWorkflow(AgentWorkflow):
12
  def __init__(self, llm:OpenAI):
 
13
  self.llm = OpenAI(model = "gpt-4o-mini", temperature=0.7, api_key=OPEN_AI)
14
  self.root_agent = ReActAgent(
15
  name="alfred_router",
 
1
  import asyncio
2
+ import os
3
  from llama_index.core.agent.workflow import AgentWorkflow
4
  from llama_index.llms.openai import OpenAI
5
  from tools import (image_analyzer_tool, youtube_transcript_tool, calculator_tool, read_document_tool, search_tool)
 
7
  from llama_index.core.agent import ReActAgent
8
  from dotenv import load_dotenv
9
  load_dotenv()
10
+ OPEN_AI = os.getenv("OPENAI_API_KEY")
11
  # Alfred React Agent
12
  class AlfredAdvancedWorkflow(AgentWorkflow):
13
  def __init__(self, llm:OpenAI):
14
+ OPEN_AI = os.getenv("OPENAI_API_KEY")
15
  self.llm = OpenAI(model = "gpt-4o-mini", temperature=0.7, api_key=OPEN_AI)
16
  self.root_agent = ReActAgent(
17
  name="alfred_router",