DrishtiSharma commited on
Commit
374d623
·
verified ·
1 Parent(s): 389a78f

Update agents/base_agent.py

Browse files
Files changed (1) hide show
  1. agents/base_agent.py +1 -1
agents/base_agent.py CHANGED
@@ -12,7 +12,7 @@ class BaseAgent(ABC):
12
 
13
  async def get_completion(self, prompt: str) -> str:
14
  try:
15
- response = client.chat.completions.create(groq_api_key=os.getenv("GROQ_API_KEY"), messages=[{'role': 'user', 'content': prompt}], model=self.model_name)
16
  return response.choices[0].message.content
17
  except Exception as e:
18
  raise Exception(f"Error getting completion: {str(e)}")
 
12
 
13
  async def get_completion(self, prompt: str) -> str:
14
  try:
15
+ response = client.chat.completions.create(messages=[{'role': 'user', 'content': prompt}], model=self.model_name)
16
  return response.choices[0].message.content
17
  except Exception as e:
18
  raise Exception(f"Error getting completion: {str(e)}")