llm_knowledge_base / chatgpt.py
allinaigc's picture
Upload 35 files
b2e325f verified
# -*- coding: utf-8 -*-
import requests
import streamlit as st
import openai
from openai import OpenAI
import os
from dotenv import load_dotenv
import numpy as np
import pandas as pd
import csv
import tempfile
from tempfile import NamedTemporaryFile
import pathlib
from pathlib import Path
import re
from re import sub
import matplotlib.pyplot as plt
from itertools import product
from tqdm import tqdm_notebook, tqdm, trange
import time
from time import sleep
import pretty_errors
import seaborn as sns
from matplotlib.pyplot import style
from rich import print
import warnings
warnings.filterwarnings('ignore')
load_dotenv()
### 设置openai的API key
os.environ["OPENAI_API_KEY"] = os.environ['user_token']
openai.api_key = os.environ['user_token']
##NOTE: 需要注意OpenAI的版本
def chatgpt(user_prompt, sys_prompt="你是一个咨询顾问,我需要你用简体中文来总结内容。", openai_model="gpt-3.5-turbo-16k"):
# print('chatgpt mode starts!')
try:
openai_client = OpenAI()
chatgpt_response = openai_client.chat.completions.create(
model=openai_model,
messages=[
{"role": "system", "content": sys_prompt},
{"role": "user", "content": user_prompt},
],
stream=False,
)
full_response = chatgpt_response.choices[0].message.content
print(full_response)
except Exception as e:
print(e)
full_response = "Sorry, I don't have an answer for that yet."
# return chatgpt_response, full_response
return full_response
# user_prompt = "where is Shanghai?"
# system_prompt = "You are a helpful AI assistant."
# full_response = chatgpt(user_prompt=user_prompt, sys_prompt=system_prompt)
# print(full_response)