jarvisx17 commited on
Commit
df160a9
1 Parent(s): 4c63f51

Update chat.py

Browse files
Files changed (1) hide show
  1. chat.py +4 -1
chat.py CHANGED
@@ -4,6 +4,9 @@ from langchain_groq import ChatGroq
4
  from langchain.chat_models import ChatOpenAI
5
  from langchain_core.prompts.prompt import PromptTemplate
6
  from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory
 
 
 
7
 
8
  template = f"""Role: You are a super friendly, enthusiastic, and empathetic female friend who chats to teenage girls.
9
 
@@ -38,7 +41,7 @@ PROMPT = PromptTemplate(
38
  )
39
 
40
  # Initialize the ChatGroq LLM
41
- llm = ChatOpenAI(model="gpt-3.5-turbo", openai_api_key="OPENAIKEY", temperature=0)
42
  # llm = ChatGroq(temperature=0,groq_api_key="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", model_name="llama3-70b-8192")
43
  #model=llama3-8b-8192
44
 
 
4
  from langchain.chat_models import ChatOpenAI
5
  from langchain_core.prompts.prompt import PromptTemplate
6
  from langchain_mongodb.chat_message_histories import MongoDBChatMessageHistory
7
+ import os
8
+
9
+ openai_key = os.environ['OPENAIKEY']
10
 
11
  template = f"""Role: You are a super friendly, enthusiastic, and empathetic female friend who chats to teenage girls.
12
 
 
41
  )
42
 
43
  # Initialize the ChatGroq LLM
44
+ llm = ChatOpenAI(model="gpt-3.5-turbo", openai_api_key=openai_key, temperature=0)
45
  # llm = ChatGroq(temperature=0,groq_api_key="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", model_name="llama3-70b-8192")
46
  #model=llama3-8b-8192
47