adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1000 from SuanmoSuanyangTechnology/fix/memory_search

Browse Source 
fix(api): correct import paths in memory_read and celery task command
This commit is contained in:
Ke Sun
2026-04-24 19:11:23 +08:00
committed by GitHub
parent 6cb48664b7 422af69904
commit ed5f98a746
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/app/controllers/service/memory_api_controller.py
View File

@@ -3,6 +3,7 @@
from fastapi import APIRouter, Body, Depends, Query, Request from fastapi import APIRouter, Body, Depends, Query, Request
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from app.celery_task_scheduler import scheduler
from app.core.api_key_auth import require_api_key from app.core.api_key_auth import require_api_key
from app.core.logging_config import get_business_logger from app.core.logging_config import get_business_logger
from app.core.quota_stub import check_end_user_quota from app.core.quota_stub import check_end_user_quota
@@ -18,7 +19,6 @@ from app.schemas.memory_api_schema import (
MemoryWriteSyncResponse, MemoryWriteSyncResponse,
) )
from app.services.memory_api_service import MemoryAPIService from app.services.memory_api_service import MemoryAPIService
from celery_task_scheduler import scheduler
router = APIRouter(prefix="/memory", tags=["V1 - Memory API"]) router = APIRouter(prefix="/memory", tags=["V1 - Memory API"])
logger = get_business_logger() logger = get_business_logger()

2
api/app/core/memory/agent/langgraph_graph/routing/write_router.py
View File

@@ -1,6 +1,7 @@
import json import json
import os import os
from app.celery_task_scheduler import scheduler
from app.core.logging_config import get_agent_logger from app.core.logging_config import get_agent_logger
from app.core.memory.agent.langgraph_graph.tools.write_tool import format_parsing, messages_parse from app.core.memory.agent.langgraph_graph.tools.write_tool import format_parsing, messages_parse
from app.core.memory.agent.models.write_aggregate_model import WriteAggregateModel from app.core.memory.agent.models.write_aggregate_model import WriteAggregateModel
@@ -13,7 +14,6 @@ from app.db import get_db_context
from app.repositories.memory_short_repository import LongTermMemoryRepository from app.repositories.memory_short_repository import LongTermMemoryRepository
from app.schemas.memory_agent_schema import AgentMemory_Long_Term from app.schemas.memory_agent_schema import AgentMemory_Long_Term
from app.utils.config_utils import resolve_config_id from app.utils.config_utils import resolve_config_id
from celery_task_scheduler import scheduler
logger = get_agent_logger(__name__) logger = get_agent_logger(__name__)
template_root = os.path.join(PROJECT_ROOT_, 'memory', 'agent', 'utils', 'prompt') template_root = os.path.join(PROJECT_ROOT_, 'memory', 'agent', 'utils', 'prompt')

2
api/app/core/workflow/nodes/memory/node.py
View File

@@ -1,6 +1,7 @@
import re import re
from typing import Any from typing import Any
from app.celery_task_scheduler import scheduler
from app.core.memory.enums import SearchStrategy from app.core.memory.enums import SearchStrategy
from app.core.memory.memory_service import MemoryService from app.core.memory.memory_service import MemoryService
from app.core.workflow.engine.state_manager import WorkflowState from app.core.workflow.engine.state_manager import WorkflowState
@@ -11,7 +12,6 @@ from app.core.workflow.variable.base_variable import VariableType
from app.core.workflow.variable.variable_objects import FileVariable, ArrayVariable from app.core.workflow.variable.variable_objects import FileVariable, ArrayVariable
from app.db import get_db_read from app.db import get_db_read
from app.schemas import FileInput from app.schemas import FileInput
from celery_task_scheduler import scheduler
class MemoryReadNode(BaseNode): class MemoryReadNode(BaseNode):