Skip to content

Commit

Permalink
✨ feat: Update llmkira/task/schema.py
Browse files Browse the repository at this point in the history
Updated the llmkira/task/schema.py file to include additional imports and make some changes to the code.
  • Loading branch information
sudoskys committed Dec 3, 2023
1 parent 097c9f6 commit ff8735f
Showing 1 changed file with 13 additions and 12 deletions.
25 changes: 13 additions & 12 deletions llmkira/task/schema.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,13 +11,14 @@
import khl
import orjson
import shortuuid
from llmkira.schema import RawMessage
from llmkira.sdk.endpoint.schema import LlmResult
from llmkira.sdk.schema import File, Function, ToolMessage, FunctionMessage, TaskBatch
from loguru import logger
from pydantic import model_validator, ConfigDict, Field, BaseModel
from telebot import types

from llmkira.schema import RawMessage
from llmkira.sdk.endpoint.schema import LlmResult
from llmkira.sdk.schema import File, Function, ToolMessage, FunctionMessage, TaskBatch

if TYPE_CHECKING:
from llmkira.sender.slack.schema import SlackMessageEvent

Expand Down Expand Up @@ -151,7 +152,7 @@ def check(self):
@classmethod
def from_root(cls, release_chain, function_enable, platform: str = "default", **kwargs):
return cls(
sign_as=(0, "root", platform),
sign_as=(0, "root", platform, str(shortuuid.uuid()).upper()[:5]),
release_chain=release_chain,
function_enable=function_enable,
**kwargs
Expand Down Expand Up @@ -418,7 +419,7 @@ def _convert(_message: types.Message) -> Optional[RawMessage]:
消息标准化
"""
if not _message:
raise ValueError(f"Message is empty")
raise ValueError("Message is empty")
if isinstance(_message, types.Message):
user_id = _message.from_user.id
chat_id = _message.chat.id
Expand All @@ -429,7 +430,7 @@ def _convert(_message: types.Message) -> Optional[RawMessage]:
return RawMessage(
user_id=user_id,
chat_id=chat_id,
text=text if text else f"(empty message)",
text=text if text else "(empty message)",
created_at=str(created_at)
)

Expand Down Expand Up @@ -555,7 +556,7 @@ def _convert(_message: hikari.Message) -> Optional[RawMessage]:
消息标准化
"""
if not _message:
raise ValueError(f"Message is empty")
raise ValueError("Message is empty")
if isinstance(_message, hikari.Message):
user_id = message.author.id
chat_id = message.guild_id if message.guild_id else message.channel_id
Expand All @@ -568,7 +569,7 @@ def _convert(_message: hikari.Message) -> Optional[RawMessage]:
user_id=user_id,
chat_id=chat_id,
thread_id=thread_id,
text=text if text else f"(empty message)",
text=text if text else "(empty message)",
created_at=str(created_at)
)

Expand Down Expand Up @@ -638,7 +639,7 @@ def _convert(_message: khl.Message) -> Optional[RawMessage]:
消息标准化
"""
if not _message:
raise ValueError(f"Message is empty")
raise ValueError("Message is empty")
if isinstance(_message, khl.Message):
user_id = message.author_id
chat_id = message.ctx.guild.id if message.ctx.guild else message.ctx.channel.id
Expand All @@ -651,7 +652,7 @@ def _convert(_message: khl.Message) -> Optional[RawMessage]:
user_id=user_id,
chat_id=chat_id,
thread_id=thread_id,
text=text if text else f"(empty message)",
text=text if text else "(empty message)",
created_at=str(created_at)
)

Expand Down Expand Up @@ -721,7 +722,7 @@ def _convert(_message: "SlackMessageEvent") -> Optional[RawMessage]:
消息标准化
"""
if not _message:
raise ValueError(f"Message is empty")
raise ValueError("Message is empty")
if _message.__repr_name__() == "SlackMessageEvent":
user_id = message.user
chat_id = message.channel
Expand All @@ -734,7 +735,7 @@ def _convert(_message: "SlackMessageEvent") -> Optional[RawMessage]:
user_id=user_id,
chat_id=chat_id,
thread_id=thread_id,
text=text if text else f"(empty message)",
text=text if text else "(empty message)",
created_at=str(created_at)
)

Expand Down

0 comments on commit ff8735f

Please sign in to comment.