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:
|