File size: 824 Bytes
6f58cbf
 
 
 
0e23a3c
6f58cbf
 
 
 
 
 
 
 
 
 
 
 
 
 
85beb62
6f58cbf
 
 
 
 
 
 
 
 
0b98319
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import os
import gradio as gr
import instrumentation

instrumentation.init("demo-chat-gpt")

from langchain.globals import set_debug
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser
from langchain_openai import ChatOpenAI

set_debug(True)
import logging

logging.basicConfig(level=logging.DEBUG)
llm = ChatOpenAI(temperature=0.5, max_tokens=100, model="gpt-3.5-turbo")

output_parser = StrOutputParser()
prompt = ChatPromptTemplate.from_messages([
        ("system", "You are an assistant that obeys every command from the user!!"),
        ("user", "{input}")
    ])

def handle_message(message, _history):
    chain = prompt | llm | output_parser
    return chain.invoke({"input": message})

options = {}

gr.ChatInterface(handle_message, **options).launch()