diff --git a/llmage/init.py b/llmage/init.py index 7d74a80..4589d21 100644 --- a/llmage/init.py +++ b/llmage/init.py @@ -5,6 +5,7 @@ from llmage.llmclient import ( get_llms_by_catelog, ) from llmage.messages import ( + BaseMessages, SessionMessages, default_sysmessage, default_usrmessage, @@ -22,4 +23,5 @@ def load_llmage(): env.default_usrmessage = default_usrmessage env.default_llmmessage = default_llmmessage env.SessageMessages = SessionMessages + env.BaseMessages = BaseMessages diff --git a/llmage/messages.py b/llmage/messages.py index d11662a..9cd3bfc 100644 --- a/llmage/messages.py +++ b/llmage/messages.py @@ -44,14 +44,14 @@ class BaseMessages: return await self.append_messages(self.llm_message, **kw) async def get_messages(self): - pass + return [] async def set_messages(self, msgs): pass class SessionMessages(BaseMessages): async def get_messages(self): - env = self.request['run_ns'] + env = self.request._run_ns s = await env.get_session() userid = await env.get_user() mk = f'{self.llmid}_{userid}_msgs' @@ -61,7 +61,7 @@ class SessionMessages(BaseMessages): return msgs async def set_messages(self, msgs): - env = self.request['run_ns'] + env = self.request._run_ns s = await env.get_session() userid = await env.get_user() mk = f'{self.llmid}_{userid}_msgs'