跳转至

加密会话

EncryptedSession 可为任何会话实现提供透明加密，并通过自动使旧条目过期来保护对话数据。

功能

  • 透明加密：使用 Fernet 加密封装任何会话
  • 每会话密钥：使用 HKDF 密钥派生，为每个会话生成唯一的加密密钥
  • 自动过期：TTL 到期后，会静默跳过旧条目
  • 即插即用的替代方案：适用于任何现有会话实现

安装

加密会话需要 encrypt extra：

pip install 'openai-agents[encrypt]'

快速入门

此示例使用内存中的 SQLiteSession。内置会话不需要单独的数据库驱动程序。

import asyncio
from agents import Agent, Runner, SQLiteSession
from agents.extensions.memory import EncryptedSession

async def main():
    agent = Agent("Assistant")

    underlying_session = SQLiteSession("user-123")
    try:
        session = EncryptedSession(
            session_id="user-123",
            underlying_session=underlying_session,
            encryption_key="your-secret-key-here",
            ttl=600  # 10 minutes
        )

        result = await Runner.run(agent, "Hello", session=session)
        print(result.final_output)
    finally:
        underlying_session.close()

if __name__ == "__main__":
    asyncio.run(main())

配置

加密密钥

加密密钥可以是 Fernet 密钥，也可以是任意字符串：

from agents.extensions.memory import EncryptedSession

# Using a Fernet key (base64-encoded)
session = EncryptedSession(
    session_id="user-123",
    underlying_session=underlying_session,
    encryption_key="your-fernet-key-here",
    ttl=600
)

# Using a raw string (will be derived to a key)
session = EncryptedSession(
    session_id="user-123", 
    underlying_session=underlying_session,
    encryption_key="my-secret-password",
    ttl=600
)

TTL（存活时间）

设置加密条目的有效时长：

# Items expire after 1 hour
session = EncryptedSession(
    session_id="user-123",
    underlying_session=underlying_session,
    encryption_key="secret",
    ttl=3600  # 1 hour in seconds
)

# Items expire after 1 day
session = EncryptedSession(
    session_id="user-123",
    underlying_session=underlying_session,
    encryption_key="secret", 
    ttl=86400  # 24 hours in seconds
)

不同会话类型的用法

SQLite 会话

from agents import SQLiteSession
from agents.extensions.memory import EncryptedSession

# Create encrypted SQLite session
underlying = SQLiteSession("user-123", "conversations.db")

session = EncryptedSession(
    session_id="user-123",
    underlying_session=underlying,
    encryption_key="secret-key"
)

SQLAlchemy 会话

对于下面的 PostgreSQL 示例，请安装 encrypt 和 sqlalchemy extras。sqlalchemy extra 包含 postgresql+asyncpg:// URL 使用的 asyncpg 驱动程序。

pip install 'openai-agents[encrypt,sqlalchemy]'

from agents.extensions.memory import EncryptedSession, SQLAlchemySession

# Create encrypted SQLAlchemy session
underlying = SQLAlchemySession.from_url(
    "user-123",
    url="postgresql+asyncpg://user:pass@localhost/db",
    create_tables=True
)

session = EncryptedSession(
    session_id="user-123",
    underlying_session=underlying,
    encryption_key="secret-key"
)

应用程序负责释放由 SQLAlchemySession.from_url() 创建的引擎。在不再需要使用该引擎的所有 SQLAlchemySession 实例后，请在应用程序的清理流程中调用 await underlying.engine.dispose()，即使运行失败也应如此。

高级会话功能

将 EncryptedSession 与 AdvancedSQLiteSession 等高级会话实现配合使用时，请注意：

  • 由于消息内容已加密，find_turns_by_content() 等方法无法有效工作
  • 基于内容的搜索会对加密数据执行操作，因此效果有限

密钥派生

EncryptedSession 使用 HKDF（基于 HMAC 的密钥派生函数）为每个会话派生唯一的加密密钥：

  • 主密钥：您提供的加密密钥
  • 会话盐值：会话 ID
  • 信息字符串："agents.session-store.hkdf.v1"
  • 输出：32 字节的 Fernet 密钥

这可以确保： - 每个会话都有唯一的加密密钥 - 没有主密钥便无法派生密钥 - 无法跨不同会话解密会话数据

自动过期

当条目超过 TTL 时，检索期间会自动跳过这些条目：

# Items older than TTL are silently ignored
items = await session.get_items()  # Only returns non-expired items

# Expired items don't affect session behavior
result = await Runner.run(agent, "Continue conversation", session=session)

API 参考