Skip to content

Commit ddbc05e

Browse files
AmbratolmAmbratolm
authored andcommitted
Fixed AI response char limit prob
1 parent 96cb31f commit ddbc05e

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

db/Item.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -238,7 +238,7 @@ class ItemStack(Model):
238238
id="potion",
239239
name="Potion",
240240
description="A basic potion that restores health.",
241-
emoji="<:potion:1350653172087263363>",
241+
emoji="<:potion:1350710440950366218>",
242242
alt_emoji="🧪",
243243
icon_url="https://cdn.discordapp.com/attachments/1348859490203734057/1350655931947417672/potion.png?ex=67d787dc&is=67d6365c&hm=87bec8eb5769212f444e990d27a245f014dc865eb4fb23b97fa718f2212fe214&",
244244
type=ItemType.CONSUMABLE,

utils/ai.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ class ActAi(BaseModel):
2222
api_key: NonEmptyStr
2323
instructions: NonEmptyStr | list[NonEmptyStr] | None = None
2424
model_name: str = Field(alias="model", default="gemini-2.0-flash")
25+
response_char_limit: int = 4000
2526

2627
_client: Client | None = None
2728
_config: GenerateContentConfig | None = None
@@ -52,7 +53,10 @@ async def prompt(self, text: str, file: ActFile | None = None) -> str | None:
5253
Part.from_bytes(data=file.data, mime_type=file.mime_type or "")
5354
)
5455
response = await chat.send_message(message, self._config)
55-
return response.text if response else None
56+
response_text = response.text if response else None
57+
if response_text and len(response_text) > self.response_char_limit:
58+
response_text = response_text[: (self.response_char_limit - 3)] + "..."
59+
return response_text
5660

5761
# ----------------------------------------------------------------------------------------------------
5862

0 commit comments

Comments
 (0)