Upload 2 files

LLMAPI.py

@@ -0,0 +1,21 @@
+from openai import OpenAI
+ 
+
+def get_LLM_response(user_input,systen_prompt):
+    client = OpenAI(
+        api_key = "sk-HWm8ZBWVs9DStt3MM1aVWTtelJndUJmoR7rdhaV72Sf9meZG", # key替换成你的API key
+        base_url = "https://api.moonshot.cn/v1",
+    )
+
+    completion = client.chat.completions.create(
+        model = "moonshot-v1-8k",
+        messages = [
+            {"role": "system", "content": systen_prompt},
+            {"role": "user", "content": user_input}
+        ],
+        temperature = 0.3
+    )
+ 
+    return completion.choices[0].message.content    
+        
+

LLMStore_v1.py

@@ -0,0 +1,63 @@
+import gradio as gr
+from LLMAPI import get_LLM_response # 返回LLM的回复
+
+# 使用会话状态来存储LLM名称列表, 思考这里为啥不用全局变量?
+
+# 存储所有创建的LLM
+llm_store = []
+llm_list = []
+
+# 创建LLM的功能
+def create_llm(name, description, system_prompt):
+    llm = {
+        "name": name,
+        "description": description,
+        "system_prompt": system_prompt
+    }
+    llm_store.append(llm)
+    llm_list.append(name)
+    return f"LLM '{name}' 创建成功！", gr.Dropdown(choices=llm_list, interactive=True)
+
+# 根据选择的LLM获取其详细信息
+def get_llm_info(selected_llm_name):
+    for llm in llm_store:
+        if llm["name"] == selected_llm_name:
+            return llm["name"], llm["description"]
+
+# 对话功能
+def chat_with_llm(selected_llm_name, user_input):
+    for llm in llm_store:
+        if llm["name"] == selected_llm_name:
+            system_prompt = llm["system_prompt"]
+            # 这里可以调用你的LLM模型进行对话
+            response = get_LLM_response(user_input, system_prompt)
+            return response
+
+# 创建Gradio界面
+with gr.Blocks() as demo:
+     
+    with gr.Tab("对话"):
+        gr.Markdown("## LLM对话")
+        llm_dropdown = gr.Dropdown(label="选择LLM", choices=[], interactive=True)
+        llm_name_display = gr.Textbox(label="LLM名称", interactive=False)
+        llm_description_display = gr.Textbox(label="LLM描述", interactive=False)
+        user_input = gr.Textbox(label="你的输入")
+        chat_button = gr.Button("发送")
+        chat_output = gr.Textbox(label="对话结果")
+
+        llm_dropdown.change(get_llm_info, inputs=llm_dropdown, outputs=[llm_name_display, llm_description_display])
+        chat_button.click(chat_with_llm, inputs=[llm_dropdown, user_input], outputs=chat_output)
+
+
+    with gr.Tab("创建LLM"):
+        gr.Markdown("## 创建一个自定义的LLM")
+        name_input = gr.Textbox(label="名称")
+        description_input = gr.Textbox(label="描述")
+        system_prompt_input = gr.Textbox(label="预制提示词/System Prompt")
+        create_button = gr.Button("创建")
+        create_output = gr.Textbox(label="创建结果")
+
+        create_button.click(create_llm, inputs=[name_input, description_input, system_prompt_input], 
+                            outputs=[create_output, llm_dropdown])
+
+demo.launch(share=True) # share=True to make the app accessible to others