Update agents/base_agent.py
Browse files- agents/base_agent.py +1 -1
agents/base_agent.py
CHANGED
|
@@ -7,7 +7,7 @@ client = Groq()
|
|
| 7 |
|
| 8 |
|
| 9 |
class BaseAgent(ABC):
|
| 10 |
-
def __init__(self, model_name: str = "llama3
|
| 11 |
self.model_name = model_name
|
| 12 |
|
| 13 |
async def get_completion(self, prompt: str) -> str:
|
|
|
|
| 7 |
|
| 8 |
|
| 9 |
class BaseAgent(ABC):
|
| 10 |
+
def __init__(self, model_name: str = "llama3-8b-8192"):
|
| 11 |
self.model_name = model_name
|
| 12 |
|
| 13 |
async def get_completion(self, prompt: str) -> str:
|