Update my_memory_logic.py
Browse files- my_memory_logic.py +2 -1
my_memory_logic.py
CHANGED
@@ -33,7 +33,8 @@ restatement_prompt = ChatPromptTemplate.from_messages([
|
|
33 |
# 4) Initialize the ChatGroq LLM
|
34 |
# Ensure you have your GROQ_API_KEY set in the environment
|
35 |
restatement_llm = ChatGroq(
|
36 |
-
model="llama3-70b-8192",
|
|
|
37 |
groq_api_key=os.environ["GROQ_API_KEY"]
|
38 |
)
|
39 |
|
|
|
33 |
# 4) Initialize the ChatGroq LLM
|
34 |
# Ensure you have your GROQ_API_KEY set in the environment
|
35 |
restatement_llm = ChatGroq(
|
36 |
+
model="llama3-70b-8192",
|
37 |
+
# model="mixtral-8x7b-32768"# or whichever model
|
38 |
groq_api_key=os.environ["GROQ_API_KEY"]
|
39 |
)
|
40 |
|