mirror of
https://github.com/hwchase17/langchain
synced 2024-11-18 09:25:54 +00:00
87e502c6bc
Co-authored-by: jacoblee93 <jacoblee93@gmail.com> Co-authored-by: Harrison Chase <hw.chase.17@gmail.com>
45 lines
1.6 KiB
Plaintext
45 lines
1.6 KiB
Plaintext
```python
|
|
from langchain.prompts import (
|
|
ChatPromptTemplate,
|
|
MessagesPlaceholder,
|
|
SystemMessagePromptTemplate,
|
|
HumanMessagePromptTemplate
|
|
)
|
|
from langchain.chains import ConversationChain
|
|
from langchain.chat_models import ChatOpenAI
|
|
from langchain.memory import ConversationBufferMemory
|
|
|
|
prompt = ChatPromptTemplate.from_messages([
|
|
SystemMessagePromptTemplate.from_template(
|
|
"The following is a friendly conversation between a human and an AI. The AI is talkative and "
|
|
"provides lots of specific details from its context. If the AI does not know the answer to a "
|
|
"question, it truthfully says it does not know."
|
|
),
|
|
MessagesPlaceholder(variable_name="history"),
|
|
HumanMessagePromptTemplate.from_template("{input}")
|
|
])
|
|
|
|
llm = ChatOpenAI(temperature=0)
|
|
memory = ConversationBufferMemory(return_messages=True)
|
|
conversation = ConversationChain(memory=memory, prompt=prompt, llm=llm)
|
|
|
|
conversation.predict(input="Hi there!")
|
|
```
|
|
```pycon
|
|
Hello! How can I assist you today?
|
|
```
|
|
|
|
```python
|
|
conversation.predict(input="I'm doing well! Just having a conversation with an AI.")
|
|
```
|
|
```pycon
|
|
That sounds like fun! I'm happy to chat with you. Is there anything specific you'd like to talk about?
|
|
```
|
|
|
|
```python
|
|
conversation.predict(input="Tell me about yourself.")
|
|
```
|
|
```pycon
|
|
Sure! I am an AI language model created by OpenAI. I was trained on a large dataset of text from the internet, which allows me to understand and generate human-like language. I can answer questions, provide information, and even have conversations like this one. Is there anything else you'd like to know about me?
|
|
```
|