vijaya01 commited on
Commit
89f13cf
·
verified ·
1 Parent(s): a854a3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -5,7 +5,8 @@ import inspect
5
  import pandas as pd
6
 
7
  from crewai import Crew, Agent, Task
8
- from langchain.llms import Together
 
9
  import os
10
 
11
  # (Keep Constants as is)
@@ -21,12 +22,15 @@ class BasicAgent:
21
  def __init__(self):
22
  print("✅ Initializing CrewAI agent...")
23
 
24
- self.llm = Together(
25
- model="togethercomputer/llama-2-70b-chat", # or any supported model
 
 
26
  temperature=0.3,
27
- together_api_key=os.getenv("TOGETHER_API_KEY")
28
  )
29
 
 
30
  self.agent = Agent(
31
  role="QA Specialist",
32
  goal="Answer benchmark questions accurately with deep reasoning and knowledge",
 
5
  import pandas as pd
6
 
7
  from crewai import Crew, Agent, Task
8
+ # from langchain.llms import Together
9
+ from langchain.chat_models import ChatOpenAI
10
  import os
11
 
12
  # (Keep Constants as is)
 
22
  def __init__(self):
23
  print("✅ Initializing CrewAI agent...")
24
 
25
+
26
+
27
+ self.llm = ChatOpenAI(
28
+ model_name="gpt-4",
29
  temperature=0.3,
30
+ openai_api_key=os.getenv("OPENAI_API_KEY")
31
  )
32
 
33
+
34
  self.agent = Agent(
35
  role="QA Specialist",
36
  goal="Answer benchmark questions accurately with deep reasoning and knowledge",