e8ae46b286
- Add end_user_controller with unauthenticated endpoint for creating end users - Implement get_or_create_end_user logic to handle duplicate end users by other_id - Register end_user_controller router in main controller initialization - Add list_memory_configs endpoint to retrieve all workspace memory configurations - Update MemoryWriteRequest and MemoryReadRequest to make config_id required field - Refactor memory API endpoints to parse request body directly instead of using Body parameter - Add CreateEndUserRequest and CreateEndUserResponse schemas for end user creation - Add ListConfigsResponse schema for configs listing endpoint - Remove unused config_id and llm_model_id parameters from Neo4j write operation - Update .gitignore to exclude redbear-mem-metrics and pitch-deck directories
116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
"""Memory 服务接口 - 基于 API Key 认证"""
|
|
|
|
from app.core.api_key_auth import require_api_key
|
|
from app.core.logging_config import get_business_logger
|
|
from app.core.response_utils import success
|
|
from app.db import get_db
|
|
from app.schemas.api_key_schema import ApiKeyAuth
|
|
from app.schemas.memory_api_schema import (
|
|
ListConfigsResponse,
|
|
MemoryReadRequest,
|
|
MemoryReadResponse,
|
|
MemoryWriteRequest,
|
|
MemoryWriteResponse,
|
|
)
|
|
from app.services.memory_api_service import MemoryAPIService
|
|
from fastapi import APIRouter, Body, Depends, Request
|
|
from sqlalchemy.orm import Session
|
|
|
|
router = APIRouter(prefix="/memory", tags=["V1 - Memory API"])
|
|
logger = get_business_logger()
|
|
|
|
|
|
@router.get("")
|
|
async def get_memory_info():
|
|
"""获取记忆服务信息（占位）"""
|
|
return success(data={}, msg="Memory API - Coming Soon")
|
|
|
|
|
|
@router.post("/write_api_service")
|
|
@require_api_key(scopes=["memory"])
|
|
async def write_memory_api_service(
|
|
request: Request,
|
|
api_key_auth: ApiKeyAuth = None,
|
|
db: Session = Depends(get_db),
|
|
message: str = Body(..., description="Message content"),
|
|
):
|
|
"""
|
|
Write memory to storage.
|
|
|
|
Stores memory content for the specified end user using the Memory API Service.
|
|
"""
|
|
body = await request.json()
|
|
payload = MemoryWriteRequest(**body)
|
|
logger.info(f"Memory write request - end_user_id: {payload.end_user_id}, workspace_id: {api_key_auth.workspace_id}")
|
|
|
|
memory_api_service = MemoryAPIService(db)
|
|
|
|
result = await memory_api_service.write_memory(
|
|
workspace_id=api_key_auth.workspace_id,
|
|
end_user_id=payload.end_user_id,
|
|
message=payload.message,
|
|
config_id=payload.config_id,
|
|
storage_type=payload.storage_type,
|
|
user_rag_memory_id=payload.user_rag_memory_id,
|
|
)
|
|
|
|
logger.info(f"Memory write successful for end_user: {payload.end_user_id}")
|
|
return success(data=MemoryWriteResponse(**result).model_dump(), msg="Memory written successfully")
|
|
|
|
|
|
@router.post("/read_api_service")
|
|
@require_api_key(scopes=["memory"])
|
|
async def read_memory_api_service(
|
|
request: Request,
|
|
api_key_auth: ApiKeyAuth = None,
|
|
db: Session = Depends(get_db),
|
|
message: str = Body(..., description="Query message"),
|
|
):
|
|
"""
|
|
Read memory from storage.
|
|
|
|
Queries and retrieves memories for the specified end user with context-aware responses.
|
|
"""
|
|
body = await request.json()
|
|
payload = MemoryReadRequest(**body)
|
|
logger.info(f"Memory read request - end_user_id: {payload.end_user_id}")
|
|
|
|
memory_api_service = MemoryAPIService(db)
|
|
|
|
result = await memory_api_service.read_memory(
|
|
workspace_id=api_key_auth.workspace_id,
|
|
end_user_id=payload.end_user_id,
|
|
message=payload.message,
|
|
search_switch=payload.search_switch,
|
|
config_id=payload.config_id,
|
|
storage_type=payload.storage_type,
|
|
user_rag_memory_id=payload.user_rag_memory_id,
|
|
)
|
|
|
|
logger.info(f"Memory read successful for end_user: {payload.end_user_id}")
|
|
return success(data=MemoryReadResponse(**result).model_dump(), msg="Memory read successfully")
|
|
|
|
|
|
@router.get("/configs")
|
|
@require_api_key(scopes=["memory"])
|
|
async def list_memory_configs(
|
|
request: Request,
|
|
api_key_auth: ApiKeyAuth = None,
|
|
db: Session = Depends(get_db),
|
|
):
|
|
"""
|
|
List all memory configs for the workspace.
|
|
|
|
Returns all available memory configurations associated with the authorized workspace.
|
|
"""
|
|
logger.info(f"List configs request - workspace_id: {api_key_auth.workspace_id}")
|
|
|
|
memory_api_service = MemoryAPIService(db)
|
|
|
|
result = memory_api_service.list_memory_configs(
|
|
workspace_id=api_key_auth.workspace_id,
|
|
)
|
|
|
|
logger.info(f"Listed {result['total']} configs for workspace: {api_key_auth.workspace_id}")
|
|
return success(data=ListConfigsResponse(**result).model_dump(), msg="Configs listed successfully")
|