yumoqing/llmage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2025-08-25 11:15:31 +08:00
parent c073da5c5a
commit 2fc96f7954
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llmage/init.py
View File

@ -5,6 +5,7 @@ from llmage.llmclient import (
get_llms_by_catelog, get_llms_by_catelog,
) )
from llmage.messages import ( from llmage.messages import (
BaseMessages,
SessionMessages, SessionMessages,
default_sysmessage, default_sysmessage,
default_usrmessage, default_usrmessage,
@ -22,4 +23,5 @@ def load_llmage():
env.default_usrmessage = default_usrmessage env.default_usrmessage = default_usrmessage
env.default_llmmessage = default_llmmessage env.default_llmmessage = default_llmmessage
env.SessageMessages = SessionMessages env.SessageMessages = SessionMessages
env.BaseMessages = BaseMessages

6
llmage/messages.py
View File

@ -44,14 +44,14 @@ class BaseMessages:
return await self.append_messages(self.llm_message, **kw) return await self.append_messages(self.llm_message, **kw)
async def get_messages(self): async def get_messages(self):
pass return []
async def set_messages(self, msgs): async def set_messages(self, msgs):
pass pass
class SessionMessages(BaseMessages): class SessionMessages(BaseMessages):
async def get_messages(self): async def get_messages(self):
env = self.request['run_ns'] env = self.request._run_ns
s = await env.get_session() s = await env.get_session()
userid = await env.get_user() userid = await env.get_user()
mk = f'{self.llmid}_{userid}_msgs' mk = f'{self.llmid}_{userid}_msgs'
@ -61,7 +61,7 @@ class SessionMessages(BaseMessages):
return msgs return msgs
async def set_messages(self, msgs): async def set_messages(self, msgs):
env = self.request['run_ns'] env = self.request._run_ns
s = await env.get_session() s = await env.get_session()
userid = await env.get_user() userid = await env.get_user()
mk = f'{self.llmid}_{userid}_msgs' mk = f'{self.llmid}_{userid}_msgs'