Spaces:
Sleeping
Sleeping
File size: 1,859 Bytes
f4b691b 22cee06 f4b691b 2a15010 f4b691b 2a15010 f4b691b |
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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 |
# You can find this code for Chainlit python streaming here (https://docs.chainlit.io/concepts/streaming/python)
# OpenAI Chat completion
import os
import sys
from openai import AsyncOpenAI # importing openai for API usage
import chainlit as cl # importing chainlit for our app
from chainlit.prompt import Prompt, PromptMessage # importing prompt tools
from chainlit.playground.providers import ChatOpenAI # importing ChatOpenAI tools
from dotenv import load_dotenv
load_dotenv()
# Add path to the root of the repo to the system path
sys.path.append(".")
from rag import _build_vector_db, retrieval_augmented_qa_pipeline, RetrievalAugmentedQAPipeline
# ChatOpenAI Templates
system_template = """You are a helpful assistant who always speaks in a pleasant tone!
"""
user_template = """{input}
Think through your response step by step.
"""
@cl.on_chat_start # marks a function that will be executed at the start of a user session
async def start_chat():
settings = {
"model": "gpt-3.5-turbo",
"temperature": 0,
"max_tokens": 500,
"top_p": 1,
"frequency_penalty": 0,
"presence_penalty": 0,
}
cl.user_session.set("settings", settings)
@cl.on_message # marks a function that should be run each time the chatbot receives a message from a user
async def main(message: cl.Message):
settings = cl.user_session.get("settings")
# client = AsyncOpenAI()
client = ChatOpenAI()
print(message.content)
vector_db = _build_vector_db()
pipeline = RetrievalAugmentedQAPipeline(
llm=client,
vector_db_retriever=vector_db
)
response = pipeline.run_pipeline(message.content)
msg = cl.Message(content="")
# Update the prompt object with the completion
msg.prompt = response
# Send and close the message stream
await msg.send()
|