Spaces:
Running
Running
Update chat.py
Browse files
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=
|
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 |
|