【deepseek+LangChain】实现postgre数据库分析问答系统

任务介绍

本地有一个postgre数据记录中国近20年人口明细数据和出生率、死亡率明细。

要求使用LangChain0.3和deepseek api连接本地postgresql数据库，并执行数据分析，整个数据分析系统具备记忆功能，输入exit退出。

详细代码

实现ai大模型本地postgresql数据库查询分析的python代码实现如下：

import re
from langchain_openai import ChatOpenAI
from langchain_community.utilities import SQLDatabase
from langchain_core.messages import SystemMessage, HumanMessage, AIMessage
from langchain.chains import create_sql_query_chain
from langchain_core.prompts import PromptTemplate  # 新增关键导入

class DeepSeekDBAnalyzer:
    def __init__(self):
        # 初始化DeepSeek模型
        self.llm = ChatOpenAI(
            base_url="https://api.deepseek.com",
            api_key="your_deepseek_api_key",
            model="deepseek-chat",
            temperature=0
        )
        
        # 连接PostgreSQL数据库
        self.db = SQLDatabase.from_uri(
            "postgresql+psycopg2://user:password@192.168.124.13:5432/my_db",
            include_tables=['total_population_year', 'birth_rate_year'],
        )
        
        # 创建自定义提示模板（关键修复点）
        custom_prompt = PromptTemplate(
            input_variables=[</