|
import gradio as gr |
|
import openai |
|
from openai import OpenAI, OpenAIError |
|
|
|
|
|
|
|
def respond(message, history, system_message): |
|
messages = [{"role": "system", "content": system_message}] |
|
|
|
for val in history: |
|
if val[0]: |
|
messages.append({"role": "user", "content": val[0]}) |
|
if val[1]: |
|
messages.append({"role": "assistant", "content": val[1]}) |
|
|
|
messages.append({"role": "user", "content": message}) |
|
client = OpenAI( |
|
api_key="sk-Eqk19o6qqgk3i09tUkriFcESxqsQNsvdF18AaHaoxeHIv0oc", |
|
base_url="https://api.moonshot.cn/v1", |
|
) |
|
try: |
|
response = client.chat.completions.create( |
|
model="moonshot-v1-128k", |
|
messages=messages, |
|
temperature=0.3 |
|
|
|
) |
|
|
|
final_response = "" |
|
for part in response.choices[0].message: |
|
|
|
if part[0] == 'content': |
|
final_response += part[1] |
|
|
|
|
|
return final_response |
|
except OpenAIError as e: |
|
print(f"An error occurred while communicating with OpenAI: {e}") |
|
return None |
|
|
|
|
|
demo = gr.ChatInterface( |
|
respond, |
|
additional_inputs=[ |
|
gr.Textbox(value="You are a friendly Chatbot.", label="System message"), |
|
|
|
], |
|
) |
|
|
|
if __name__ == "__main__": |
|
demo.launch() |
|
|