ryefoxlime commited on
Commit
19a426a
·
verified ·
1 Parent(s): f66699b

Updated to have open router gemini models

Browse files
Files changed (1) hide show
  1. src/research_agent/crew.py +1 -1
src/research_agent/crew.py CHANGED
@@ -11,7 +11,7 @@ sys.path.append('..')
11
  @CrewBase
12
  class MarketUseCaseCrew:
13
  def llm(self):
14
- return LLM(model="gemini/gemini-1.5-flash-002", api_key=os.getenv("GOOGLE_API_KEY"))
15
 
16
  @agent
17
  def researcher(self) -> Agent:
 
11
  @CrewBase
12
  class MarketUseCaseCrew:
13
  def llm(self):
14
+ return LLM(model="openrouter/google/gemini-2.0-flash-001", base_url="https://openrouter.ai/api/v1", api_key=os.getenv("OPENROUTER_API_KEY"))
15
 
16
  @agent
17
  def researcher(self) -> Agent: