dftmsg

app.py

@@ -1,17 +1,31 @@
+import streamlit as st
 from openai import OpenAI
 
+# 设置 API 参数
 base_url = "https://internlm-chat.intern-ai.org.cn/puyu/api/v1/"
-model="internlm2.5-latest"
+api_key = "sk-请填写准确的 token！"
+model = "internlm2.5-latest"
 
+# 创建 OpenAI 客户端
 client = OpenAI(
-    api_key=api_key , 
+    api_key=api_key,
     base_url=base_url,
 )
 
-chat_rsp = client.chat.completions.create(
-    model=model,
-    messages=[{"role": "user", "content": "xtuner是什么？"}],
-)
+# 设置页面标题
+st.title("Chat with InternLM")
+
+# 创建一个文本输入框供用户输入问题
+user_input = st.text_input("请输入你的问题:")
+
+# 按钮用于提交问题
+if st.button("发送"):
+    # 调用 OpenAI API 获取回复
+    chat_rsp = client.chat.completions.create(
+        model=model,
+        messages=[{"role": "user", "content": user_input}],
+    )
 
-for choice in chat_rsp.choices:
-    print(choice.message.content)
+    # 显示回复
+    for choice in chat_rsp.choices:
+        st.write(f"回复: {choice.message.content}")

b.py

@@ -1,22 +0,0 @@
-from openai import OpenAI
-
-base_url = "https://internlm-chat.intern-ai.org.cn/puyu/api/v1/"
-api_key = "sk-请填写准确的 token！"
-model="internlm2.5-latest"
-
-# base_url = "https://api.siliconflow.cn/v1"
-# api_key = "sk-请填写准确的 token！"
-# model="internlm/internlm2_5-7b-chat"
-
-client = OpenAI(
-    api_key=api_key , 
-    base_url=base_url,
-)
-
-chat_rsp = client.chat.completions.create(
-    model=model,
-    messages=[{"role": "user", "content": "xtuner是什么？"}],
-)
-
-for choice in chat_rsp.choices:
-    print(choice.message.content)

requirements.txt

@@ -1,5 +0,0 @@
-llama-index==0.11.20
-llama-index-llms-replicate==0.3.0
-llama-index-llms-openai-like==0.2.0
-llama-index-embeddings-huggingface==0.3.1
-llama-index-embeddings-instructor==0.2.1