Manoj21k commited on
Commit
ff4ea1e
1 Parent(s): 41e41b6

Update src/main.py

Browse files
Files changed (1) hide show
  1. 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='gsk')
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='gsk...',
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