Get a value from the cache. Args: id: Unique identifier for the cached item field: The field to retrieve fro

Execute a raw SQL query and return results. Args: query: SQL query string params: Parameters for the query

Execute a SQL query and return results as a list of dictionaries. Args: query: SQL query to execute params:

Flush pending traces to Langfuse (手动刷新).

Convert ORM entity to domain model

Get the application configuration (cached).

Set a value in the cache. Args: id: Unique identifier for the cached item field: The field to set in the cac

转换为领域模型

Count total number of entities. Returns: Total count

更新用户

删除用户

Execute a SQL query with improved error handling and connection management. Args: query: SQL query to execute

创建用户

Create ORM entity from domain model

Run complete question answering pipeline with streaming Args: question: User's natural language question pro

Close all connections in the pool and cleanup resources.

Establish a connection to the database. Args: connection_info: Database connection parameters Raises:

Close Qdrant connection

Delete a model

Get a value from Redis

根据 ID 获取用户

Run an Alembic command with proper environment setup.

Get a connection from the pool or create a new one with validation.

Save a chat history record. Args: history_data: Dictionary containing history record attributes Returns:

Remove expired cache entries

Ensure connection pool exists

Extract connection parameters from ConnectionInfo object or dict.

Find a database session in function arguments or keyword arguments. Args: args: Function positional arguments kwargs: Functi

Generate simple title from question

Complete analysis pipeline for a chat request. Args: request: FastAPI request object dto: Chat data transfer

Get a raw database connection and cursor with automatic cleanup. Args: dict_cursor: Use dictionary cursor if True, regul

Constructs an absolute file path by joining the directory of the current file with the provided relative path. Args: path (str): The rela

从领域模型创建

Generate text from prompt (async) Args: prompt: Input prompt temperature: Sampling temperature (0.0-2.0)

Get entity by ID or raise NotFoundError. Args: id_value: Primary key value Returns: Entity

Classify user's question intent

Execute SQL with retry logic. Args: query_id: Query ID (usually ChatSession ID or specific Query ID) initial

Test connection to a datasource using the optimized ConnectionManager. Args: db_type: Database type connecti

Remove or mask sensitive information from connection_info. Args: connection_info: Original connection info f

Get entity by ID synchronously.

Close the MySQL database connection. Raises: DatabaseError: If connection cannot be closed properly

Close the PostgreSQL database connection. Raises: DatabaseError: If connection cannot be closed properly

Establish a connection to the MySQL database. Args: connection_info: MySQL connection parameters Raises:

Establish a connection to the PostgreSQL database. Args: connection_info: PostgreSQL connection parameters Rais

Create a new instance of the appropriate database adapter. Args: db_type: Type of database Returns: DatabaseAdapter ins

Delete entity by ID. Args: id_value: Primary key value Returns: True if deleted, False if not found

删除会话

Generate text from prompt with streaming (async) Args: prompt: Input prompt temperature: Sampling temperatur

Get status information about database adapters and connections. Returns: Dictionary with adapter status information

Get model by ID with columns

Get chat session by ID. Args: id: Chat session ID Returns: Chat session if found, None otherwise

Get the global rate limiter instance. Args: redis_client: Optional Redis client to use Returns: Singleton EnhancedRa

Initialize logger with configured settings.

Search for similar vectors

Convert configuration to JSON-serializable dict.

Update model fields

Get entity by ID asynchronously.

Check rate limit using in-memory storage (fallback). Uses deque to store timestamps with sliding window.

Clean up expired connections. This method is called periodically and closes connections that have been idle for too long.

Close a specific connection. Args: conn_id: Connection ID to close

Configure connection with optimization settings.

生成 Token 对

Ensure history collection exists

Generate embedding for text

Extract client IP from request, properly handling proxies

Generate simple fallback answer

Get fallback table configuration

Generate answer summarization prompt

Generate reasoning prompt

Determine if this call should be sampled (根据采样率).

Count total number of entities synchronously.

Create new entity synchronously.

Delete entity by ID synchronously.

Get all entities with pagination synchronously.

Update existing entity synchronously.

Check database health and connection status. Returns: Dictionary with health check results

Check if request should be rate limited. Args: request: FastAPI request user_id: Optional authenticated user

Initialize migrations.

Establish a connection to the DuckDB database. Args: connection_info: DuckDB connection parameters Raises:

Create a new model

Create the database if it doesn't exist.

Create a new query history record. Args: query_data: Query history data dictionary Returns: New Que

Execute SQL query using the optimized ConnectionManager with retries. Args: datasource_id: ID of the datasource to query

Generate a unique ID for the cache. Returns: str: A unique identifier

Generate SQL query from natural language question. Args: request: FastAPI request object id: ChatSession ID

* Get a datasource by ID * @param id - Datasource ID * @returns Datasource details

* Get a paginated list of datasources * @param params - Pagination parameters and filters * @returns List of datasources

Get all entities with pagination. Args: skip: Number of records to skip limit: Maximum number of records to

Get a database session directly without using a context manager. IMPORTANT: The caller is responsible for committing/rolling back an

根据邮箱获取用户

Get entity by ID. Automatically uses sync or async implementation based on session type. Args: id_value: Primar