Spaces:
Sleeping
Sleeping
Update src/main.py
Browse files- src/main.py +3 -2
src/main.py
CHANGED
|
@@ -7,9 +7,10 @@ from langchain_experimental.agents import create_csv_agent
|
|
| 7 |
from langchain_groq import ChatGroq
|
| 8 |
from langchain.memory import ChatMessageHistory
|
| 9 |
from groq import Groq
|
|
|
|
| 10 |
|
| 11 |
# Initialize Groq client and model
|
| 12 |
-
client = Groq(api_key='
|
| 13 |
MODEL = 'llama3-70b-8192'
|
| 14 |
|
| 15 |
# Initialize chat history
|
|
@@ -20,7 +21,7 @@ history.add_ai_message("whats up?")
|
|
| 20 |
# Initialize language model
|
| 21 |
llm = ChatGroq(
|
| 22 |
temperature=0,
|
| 23 |
-
groq_api_key='
|
| 24 |
model_name='llama3-70b-8192'
|
| 25 |
)
|
| 26 |
|
|
|
|
| 7 |
from langchain_groq import ChatGroq
|
| 8 |
from langchain.memory import ChatMessageHistory
|
| 9 |
from groq import Groq
|
| 10 |
+
import os
|
| 11 |
|
| 12 |
# Initialize Groq client and model
|
| 13 |
+
client = Groq(api_key=os.getenv('Groq_API'))
|
| 14 |
MODEL = 'llama3-70b-8192'
|
| 15 |
|
| 16 |
# Initialize chat history
|
|
|
|
| 21 |
# Initialize language model
|
| 22 |
llm = ChatGroq(
|
| 23 |
temperature=0,
|
| 24 |
+
groq_api_key=os.getenv('Groq_API'),
|
| 25 |
model_name='llama3-70b-8192'
|
| 26 |
)
|
| 27 |
|