UKURIKIYEYEZU commited on
Commit
2e344b9
·
verified ·
1 Parent(s): 8b31831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -1,9 +1,12 @@
1
  import os
2
- from google.colab import userdata
3
 
4
- groq_api_key= userdata.get('groq2')
 
 
 
5
  from langchain_groq import ChatGroq
6
- llm =ChatGroq(model="llama-3.3-70b-versatile", api_key=groq_api_key)
 
7
 
8
  from langchain.prompts import ChatPromptTemplate, PromptTemplate
9
  from langchain.output_parsers import ResponseSchema, StructuredOutputParser
 
1
  import os
 
2
 
3
+ # Get the secret key from the environment
4
+ groq_api_key = os.environ.get('groq2')
5
+
6
+ ## LLM used for RAG
7
  from langchain_groq import ChatGroq
8
+
9
+ llm = ChatGroq(model="llama-3.1-70b-versatile",api_key=groq_api_key )
10
 
11
  from langchain.prompts import ChatPromptTemplate, PromptTemplate
12
  from langchain.output_parsers import ResponseSchema, StructuredOutputParser