Method record_success

src/storage/psql_manager.py:989–1030

成功调用后的条件写入

(
        self,
        filename: str,
        model_name: Optional[str] = None,
        mode: str = "geminicli"
    )

            return False

    async def record_success(
        self,
        filename: str,
        model_name: Optional[str] = None,
        mode: str = "geminicli"
    ) -> None:
        """成功调用后的条件写入"""
        self._ensure_initialized()
        filename = os.path.basename(filename)

        try:
            table_name = self._get_table_name(mode)
            async with self._pool.acquire() as conn:
                await conn.execute(f"""
                    UPDATE {table_name}
                    SET last_success = EXTRACT(EPOCH FROM NOW()),
                        error_codes = '[]',
                        error_messages = '{{}}',
                        updated_at = EXTRACT(EPOCH FROM NOW())
                    WHERE filename = $1
                      AND (error_codes IS NOT NULL AND error_codes != '[]' AND error_codes != '')
                """, filename)

                if model_name:
                    row = await conn.fetchrow(
                        f"SELECT model_cooldowns FROM {table_name} WHERE filename = $1", filename
                    )
                    if row:
                        cooldowns = json.loads(row["model_cooldowns"] or "{}")
                        if model_name in cooldowns:
                            cooldowns.pop(model_name)
                            await conn.execute(
                                f"""
                                UPDATE {table_name}
                                SET model_cooldowns = $1, updated_at = EXTRACT(EPOCH FROM NOW())
                                WHERE filename = $2
                                """,
                                json.dumps(cooldowns), filename
                            )

        except Exception as e:
            log.error(f"Error recording success for {filename}: {e}")