Update agents/base_agent.py
Browse files- 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(
|
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)}")
|