Spaces:
Sleeping
Sleeping
File size: 778 Bytes
558803b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from langchain_openai import ChatOpenAI
from langchain.chains import ConversationChain
from langchain.memory import ConversationBufferMemory
def get_chat_response(prompt, memory,openai_api_key):
model = ChatOpenAI(model="gpt-3.5-turbo", openai_api_key = openai_api_key, base_url= "https://api.aigc369.com/v1")
chain = ConversationChain(llm=model, memory=memory)
response = chain.invoke({"input":prompt})
return response["response"]
# memory = ConversationBufferMemory(return_messages=True)
# print(get_chat_response("北京有谁在呀？",memory, openai_api_key="sk-YWPVrZ0KM3B3BFVk40Ee827a2dE34124Af34F881Bf4d79B6"))
# print(get_chat_response("我的上一个问题是什么？",memory, openai_api_key="sk-YWPVrZ0KM3B3BFVk40Ee827a2dE34124Af34F881Bf4d79B6")) |