forked from NuQloud/nc-talk-ai
127 lines
4.1 KiB
PHP
127 lines
4.1 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace OCA\EducAI\Db;
|
|
|
|
use JsonSerializable;
|
|
use OCP\AppFramework\Db\Entity;
|
|
|
|
/**
|
|
* Entity for queued LLM requests waiting due to rate limits.
|
|
*
|
|
* @method int getId()
|
|
* @method void setId(int $id)
|
|
* @method int getBotId()
|
|
* @method void setBotId(int $botId)
|
|
* @method string getRoomToken()
|
|
* @method void setRoomToken(string $roomToken)
|
|
* @method string getUserId()
|
|
* @method void setUserId(string $userId)
|
|
* @method string getMessage()
|
|
* @method void setMessage(string $message)
|
|
* @method ?string getOriginalMessage()
|
|
* @method void setOriginalMessage(?string $originalMessage)
|
|
* @method ?int getReplyToMessageId()
|
|
* @method void setReplyToMessageId(?int $replyToMessageId)
|
|
* @method ?int getThreadRootMessageId()
|
|
* @method void setThreadRootMessageId(?int $threadRootMessageId)
|
|
* @method ?int getTalkMessageId()
|
|
* @method void setTalkMessageId(?int $talkMessageId)
|
|
* @method string getStatus()
|
|
* @method void setStatus(string $status)
|
|
* @method ?string getResult()
|
|
* @method void setResult(?string $result)
|
|
* @method ?string getError()
|
|
* @method void setError(?string $error)
|
|
* @method int getAttempts()
|
|
* @method void setAttempts(int $attempts)
|
|
* @method int getPriority()
|
|
* @method void setPriority(int $priority)
|
|
* @method int getCreatedAt()
|
|
* @method void setCreatedAt(int $createdAt)
|
|
* @method ?int getProcessedAt()
|
|
* @method void setProcessedAt(?int $processedAt)
|
|
*/
|
|
class QueuedRequest extends Entity implements JsonSerializable {
|
|
public const STATUS_PENDING = 'pending';
|
|
public const STATUS_PROCESSING = 'processing';
|
|
public const STATUS_COMPLETED = 'completed';
|
|
public const STATUS_FAILED = 'failed';
|
|
|
|
protected int $botId = 0;
|
|
protected string $roomToken = '';
|
|
protected string $userId = '';
|
|
protected string $message = '';
|
|
protected ?string $originalMessage = null;
|
|
protected ?int $replyToMessageId = null;
|
|
protected ?int $threadRootMessageId = null;
|
|
protected ?int $talkMessageId = null;
|
|
protected string $status = self::STATUS_PENDING;
|
|
protected ?string $result = null;
|
|
protected ?string $error = null;
|
|
protected int $attempts = 0;
|
|
protected int $priority = 100;
|
|
protected int $createdAt = 0;
|
|
protected ?int $processedAt = null;
|
|
|
|
public function __construct() {
|
|
$this->addType('id', 'integer');
|
|
$this->addType('botId', 'integer');
|
|
$this->addType('replyToMessageId', 'integer');
|
|
$this->addType('threadRootMessageId', 'integer');
|
|
$this->addType('talkMessageId', 'integer');
|
|
$this->addType('attempts', 'integer');
|
|
$this->addType('priority', 'integer');
|
|
$this->addType('createdAt', 'integer');
|
|
$this->addType('processedAt', 'integer');
|
|
}
|
|
|
|
/**
|
|
* Check if request can be retried
|
|
*/
|
|
public function canRetry(int $maxAttempts = 3): bool {
|
|
return $this->status === self::STATUS_FAILED && $this->attempts < $maxAttempts;
|
|
}
|
|
|
|
/**
|
|
* Check if request is stale (older than given seconds)
|
|
*/
|
|
public function isStale(int $maxAgeSeconds = 3600): bool {
|
|
return (time() - $this->createdAt) > $maxAgeSeconds;
|
|
}
|
|
|
|
/**
|
|
* Increment attempt counter
|
|
*/
|
|
public function incrementAttempts(): void {
|
|
$this->attempts++;
|
|
}
|
|
|
|
public function jsonSerialize(): array {
|
|
return [
|
|
'id' => $this->id,
|
|
'bot_id' => $this->botId,
|
|
'room_token' => $this->roomToken,
|
|
'user_id' => $this->userId,
|
|
'message' => $this->message,
|
|
'original_message' => $this->originalMessage,
|
|
'reply_to_message_id' => $this->replyToMessageId,
|
|
'thread_root_message_id' => $this->threadRootMessageId,
|
|
'talk_message_id' => $this->talkMessageId,
|
|
'status' => $this->status,
|
|
'result' => $this->result,
|
|
'error' => $this->error,
|
|
'attempts' => $this->attempts,
|
|
'priority' => $this->priority,
|
|
'created_at' => $this->createdAt,
|
|
'processed_at' => $this->processedAt,
|
|
'can_retry' => $this->canRetry(),
|
|
'is_stale' => $this->isStale(),
|
|
];
|
|
}
|
|
}
|
|
|
|
|
|
|