使用 Cohere command-r 和 Streamlit 创建具有上下文检索功能的聊天机器人

graph td
    A[用户界面 - Streamlit] --> B[文件上传]
    A --> C[聊天输入]
    B --> D[文件处理器]
    C --> E[聊天管理器]
    D --> F[Cohere客户端]
    E --> F
    F --> G[AI响应生成]
    G --> A

def process_file(self, uploaded_file):
    if uploaded_file.type == "application/pdf":
        return self.extract_text_from_pdf(uploaded_file)
    else:
        # 可扩展，支持更多文件类型
        return uploaded_file.read().decode()

def build_prompt(self, user_input, context=None):
    context_str = f"{context}\n\n" if context else ""
    return (
        f"{context_str}"
        f"问题: {user_input}\n"
        f"请直接给出答案，除非被要求使用现有指标和历史数据详细说明。"
    )

def chat(self, user_input, context=None):
    # 保持对话历史
    self.conversation_history.append({"role": "user", "content": user_input})

    # 限制历史记录长度，防止上下文溢出
    if len(self.conversation_history) > 10:
        self.conversation_history = self.conversation_history[-10:]

cohere==5.13.11
streamlit==1.41.1
pypdf2==3.0.1

# 创建虚拟环境
python3 -m venv chatish_env

# 激活环境
source chatish_env/bin/activate

# 安装依赖项
pip install -r requirements.txt

# 运行应用
streamlit run app.py