랭체인을 이용한 챗봇-01

간단하게 작성한 chatbot 입니다.
필요한 모듈(패키지)는 설치해 주시고 코드를 복붙하고 main.py로 저장합니다.
커맨드라인에서 ">python main.py"실행하고 Ctrl + (마우스로 URL 클릭) 하면 브라우저가 새로 열리며 대화창이 뜹니다.
그리고 대화를 시작하면 됩니다.

 

import gradio as gr

from langchain_community.chat_models import ChatOpenAI

api_key = "OpenAI API Key"    # 발급받은 OpenAI API Key 값 입력
chat_model = ChatOpenAI(
    model="gpt-3.5-turbo", openai_api_key=api_key)

with gr.Blocks() as demo:
    chatbot = gr.Chatbot()
    msg = gr.Textbox()
    clear = gr.ClearButton([msg, chatbot])

    def respond(message, chat_history):
        bot_message = chat_model.predict(message)
        chat_history.append((message, bot_message))
        return "", chat_history

    msg.submit(respond, [msg, chatbot], [msg, chatbot])

if __name__ == "__main__":
    demo.launch()

 

결과 01

 

 

결과 02

 

이렇게 대화할 수 있습니다. 랭체인을 이용했고 언어모델은 gpt-3.5-turbo를 사용했습니다.