python使用openai与sqlite实现智能体聊天长记忆存储示例代码-python代码-BFW代码库

python使用openai与sqlite实现智能体聊天长记忆存储示例代码

代码语言：python

所属分类：其他

代码描述：python使用openai与sqlite实现智能体聊天长记忆存储示例代码

代码标签： python 使用 openai sqlite 智能体聊天长记忆存储代码

下面为部分代码预览，完整代码请点击下载或在bfwstudio webide中打开

#!/usr/local/python3/bin/python3
# -*- coding: utf-8 -*
import sqlite3
import json
import datetime
import numpy as np
from openai import OpenAI
from typing import List, Dict, Any

# 配置 OpenAI API
# 请将此处替换为你的实际 API Key
api_key = "sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
client = OpenAI(api_key=api_key)

# 嵌入模型和对话模型配置
EMBEDDING_MODEL = "text-embedding-3-small"
CHAT_MODEL = "gpt-4o-mini" # 或者 gpt-3.5-turbo

class MemoryAgent:
    def __init__(self, agent_name: str, db_path: str = "memory.db"):
        self.agent_name = agent_name
        self.db_path = db_path
        self._init_db()

    def _init_db(self):
        """初始化 SQLite 数据库结构"""
        conn = sqlite3.connect(self.db_path)
        cur = conn.cursor()
        
        # 创建记忆事件表（情景记忆）
        cur.execute("""
            CREATE TABLE IF NOT EXISTS memory_events (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                .........完整代码请登录后点击上方下载按钮下载查看