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