|
|
@@ -6,6 +6,7 @@ from anon import PluginManager, Plugin
|
|
|
from anon.event import MessageEvent
|
|
|
from anon.event.message import GroupMessage
|
|
|
from anon.logger import logger
|
|
|
+from anon.storage import Storage
|
|
|
|
|
|
|
|
|
class GlmClient:
|
|
|
@@ -78,13 +79,25 @@ class GlmClient:
|
|
|
|
|
|
class GlmPlugin(Plugin):
|
|
|
_client: GlmClient
|
|
|
+ _group_shared_cache: bool
|
|
|
|
|
|
async def on_load(self):
|
|
|
self._client = GlmClient()
|
|
|
+ self.storage = Storage('glm')
|
|
|
+ if self.storage['system'] is not None:
|
|
|
+ logger.info('GlmPlugin: load cache')
|
|
|
+ self._client.system = self.storage['system']
|
|
|
+ self._group_shared_cache = self.storage['group_shared_cache']
|
|
|
+ logger.info(f'GlmPlugin: system prompt: {self._client.system}')
|
|
|
+ else:
|
|
|
+ self.storage['group_shared_cache'] = False
|
|
|
+ self._group_shared_cache = False
|
|
|
+ self.storage['system'] = self._client.system
|
|
|
+ logger.info('GlmPlugin: init cache')
|
|
|
|
|
|
async def on_event(self, event: MessageEvent):
|
|
|
rid = event.sender.user_id
|
|
|
- if isinstance(event, GroupMessage):
|
|
|
+ if isinstance(event, GroupMessage) and self._group_shared_cache:
|
|
|
rid = event.gid
|
|
|
|
|
|
if event.raw.startswith('glm'):
|
|
|
@@ -106,7 +119,8 @@ class GlmPlugin(Plugin):
|
|
|
' - curlen [int]: set max cut length\n'
|
|
|
' - s/sys [str]: set system prompt\n'
|
|
|
' - g/sys: get system prompt\n'
|
|
|
- ' - clear: clear cache'
|
|
|
+ ' - clear: clear cache\n'
|
|
|
+ ' - sw: switch group shared cache'
|
|
|
)
|
|
|
if cmd == 'cutlen':
|
|
|
self._client.max_cut_length = int(args[0])
|
|
|
@@ -114,12 +128,18 @@ class GlmPlugin(Plugin):
|
|
|
if cmd == 's/sys':
|
|
|
self._client.system = ' '.join(args)
|
|
|
self._client.cache.clear()
|
|
|
+ self.storage['system'] = self._client.system
|
|
|
await event.reply(f'system prompt set to: {" ".join(args)}')
|
|
|
if cmd == 'g/sys':
|
|
|
await event.reply(f'system prompt: {self._client.system}')
|
|
|
if cmd == 'clear':
|
|
|
self._client.cache.clear()
|
|
|
await event.reply('cache cleared')
|
|
|
+ if cmd == 'sw':
|
|
|
+ self._group_shared_cache = not self._group_shared_cache
|
|
|
+ self.storage['group_shared_cache'] = self._group_shared_cache
|
|
|
+ await event.reply(
|
|
|
+ f'group_shared_cache set to {self._group_shared_cache}')
|
|
|
|
|
|
|
|
|
PluginManager().register_plugin(GlmPlugin([MessageEvent]))
|
